数论---荣斥定理

荣斥定理

在计数时，必须注意没有重复，没有遗漏，为了使重叠部分不被重复计算，人们研究出一种新的计数方法，即：先不考虑重叠部分情况，把包含某内容中的所有对象数目先计算出来，然后把重复计算的数目排斥出去，使得计数结果既没有遗漏，有没有重复。即荣斥定理，也叫荣斥原理。

公式

比如在下图中对于m=3的情况，右式可化为 = {[（A+B - A∩B）+C - B∩C] - C∩A }+ A∩B∩C

用韦恩图来表示如下图

则图中部分为[（1+2+4+5+3+6）+4+5+6+7-5-6-4-5）]+5 = 1+2+3+4+5+6+7

那么，有了这个结论，我们便可以解决很多数论问题

如果被计数的事物有A、B两类，那么，A类B类元素个数总和= 属于A类元素个数+ 属于B类元素个数—既是A类又是B类的元素个数。（A∪B = A+B - A∩B)

如果被计数的事物有A、B、C三类，那么，A类和B类和C类元素个数总和= A类元素个数+ B类元素个数+C类元素个数—既是A类又是B类的元素个数—既是A类又是C类的元素个数—既是B类又是C类的元素个数+既是A类又是B类而且是C类的元素个数。（A∪B∪C = A+B+C - A∩B - B∩C - C∩A + A∩B∩C）

例如，求100内是3或5的倍数（被3或5整除）

那么就可以求得100内3的倍数的个数n1，5的倍数的个数n2，以及3*5的倍数的个数n3

则n=n1+n2-n3

例2 求1001内与1001互质的数的个数

可以循环遍历gcd枚举求解，但这样复杂度要明显高很多

1001=7*11*13，所以可以先求得1001内与7、11、13不互质的数的个数

然后用总数减去这个数

例3 求[1,r]中与n互质的数的个数

int solve(int r, int n)
{
    vector<int>p;
    for (int i = 2; i * i <= n; ++i){
        if (n % i == 0){
            p.push_back(i);
            while (n % i == 0) n /= i;
        }
    }
    if (n > 1) p.push_back(n);
    int sum = 0;
    for (int S = 1; S < (1 << p.size()); ++S){
        int mult = 1, bits = 0;
        for (int i = 0; i < p.size(); ++i){
            if (S & (1 << i)){
                ++bits;
                mult *= p[i];
            }
        }
        int cur = r / mult;
        if (bits % 2 == 1) sum += cur;
        else sum -= cur;
    }
    return r - sum;
}

代码的时间复杂度为O(sqrt(n))