hdu 4407容斥原理
比赛的时候,2点半就想出该怎么做了,打代码打到3点半,还在调试,最可恨的是比赛结束还没过,一直WA,比赛完以后再测了测,发现是一处小错误。。。
/* * A/win.cpp * Created on: 2012-9-8 * Author : ben */ #include <cstdio> #include <cstdlib> #include <cstring> #include <cmath> #include <ctime> #include <iostream> #include <algorithm> #include <queue> #include <set> #include <map> #include <stack> #include <string> #include <vector> #include <deque> #include <list> #include <functional> #include <numeric> #include <cctype> using namespace std; typedef long long LL; const int MAXM = 1010; const int MAXN = 400009; const int PRIME_LEN = 168; int primes[PRIME_LEN] = { 2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97, 101, 103, 107, 109, 113, 127, 131, 137, 139, 149, 151, 157, 163, 167, 173, 179, 181, 191, 193, 197, 199, 211, 223, 227, 229, 233, 239, 241, 251, 257, 263, 269, 271, 277, 281, 283, 293, 307, 311, 313, 317, 331, 337, 347, 349, 353, 359, 367, 373, 379, 383, 389, 397, 401, 409, 419, 421, 431, 433, 439, 443, 449, 457, 461, 463, 467, 479, 487, 491, 499, 503, 509, 521, 523, 541, 547, 557, 563, 569, 571, 577, 587, 593, 599, 601, 607, 613, 617, 619, 631, 641, 643, 647, 653, 659, 661, 673, 677, 683, 691, 701, 709, 719, 727, 733, 739, 743, 751, 757, 761, 769, 773, 787, 797, 809, 811, 821, 823, 827, 829, 839, 853, 857, 859, 863, 877, 881, 883, 887, 907, 911, 919, 929, 937, 941, 947, 953, 967, 971, 977, 983, 991, 997 }; int primeFactor[MAXM], primeFactorLen; typedef struct Change { int x; int y; } Change; Change changes[MAXM]; int CI; int last[MAXN]; LL gcd(LL a, LL b) { LL r; while(b) { r = a % b; a = b, b = r; } return a; } LL getlcm(LL a, LL b) { LL c = gcd(a, b); return a / c * b; } bool isCoPrime(int a, int b) { int r; while (b) { r = a % b; a = b, b = r; } return a == 1; } void findPrimeFactor(int p) { int temp = (int) sqrt(1.0 * p); primeFactorLen = 0; for (int i = 0; (i < PRIME_LEN) && (primes[i] <= temp); i++) { if (p % primes[i] == 0) { primeFactor[primeFactorLen++] = primes[i]; while (p % primes[i] == 0) { p /= primes[i]; } } } if (p > 1) { primeFactor[primeFactorLen++] = p; } } inline LL primeNSum(LL x, LL data) { return data * ((x / data) * (x / data + 1) / 2); } void dfs(int now, int count, LL lcm, LL &ret, int x) { lcm = getlcm(lcm, primeFactor[now]); if (count % 2 == 0) { ret += primeNSum(x, lcm); } else { ret -= primeNSum(x, lcm); } for (int i = now + 1; i < primeFactorLen; i++) { dfs(i, count + 1, lcm, ret, x); } } LL getsum(LL x) { if(x <= 0) { return 0; } LL ret = x * ( x + 1) / 2; for (int i = 0; i < primeFactorLen; i++) { dfs(i, 1, primeFactor[i], ret, x); } return ret; } void work(int x, int y, int p) { findPrimeFactor(p); LL sum = getsum(y) - getsum(x - 1); for (int i = 0; i < CI; i++) { if(last[changes[i].x] != i) { continue; } if (changes[i].x >= x && changes[i].x <= y) { int oldnum = changes[i].x; int newnum = changes[i].y; if (isCoPrime(oldnum, p)) { sum -= oldnum; } if (isCoPrime(newnum, p)) { sum += newnum; } } } printf("%I64d\n", sum); } int main() { #ifndef ONLINE_JUDGE freopen("data.in", "r", stdin); #endif int T, N, M, op, x, y, p; scanf("%d", &T); while(T--) { CI = 0; scanf("%d%d", &N, &M); for(int i = 0; i < M; i++) { scanf("%d", &op); if(op == 1) { scanf("%d%d%d", &x, &y, &p); work(x, y, p); } else { scanf("%d%d", &changes[CI].x, &changes[CI].y); last[changes[CI].x] = CI; CI++; } } } return 0; }