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;

}

你可能感兴趣的:(HDU)