【容斥定理 && dfs】UESTC 1544 当咸鱼也要按照基本法

Problem Description

给你n条咸鱼。有m个操作。
每个操作给你一个数，这个第这个数的倍数的咸鱼的咸鱼值异或上1。（初始咸鱼值都是0）
问你最终有多少条咸鱼的咸鱼值是1。

思路：

容斥定理：有三个集合A,B,C要求三个集合的并 = A + B + C - AB - AC - BC + ABC
我们这里咸鱼的倍数看成一个集合。
我们要求的是覆盖奇数次的个数 = A + B + C - 2AB - 2AC - 2BC + 4ABC。
我们求出这个，就求出结果了，学习了一波圣昭巨巨的dfs。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int M = 20;
ll a[M], ans, n;
int m;
ll gcd(ll x, ll y)
{
    if(!y) return x;
    else return gcd(y, x%y);
}
//为了枚举所有情况
void dfs(ll nlcm, int x, int pos)//当前选择的所有数的lcm（不包括即将要选的a[pos]），当前选择了几个数，第x个数的下标
{
    if(nlcm > n) return;
    if(pos > m) return;
    ll lcm = nlcm*a[pos]/gcd(nlcm, a[pos]);//当前选择的所有数的lcm
    if(x&1) ans += n/lcm*(1LL<<(x-1));//如果是奇数个 +
    else ans -= n/lcm*(1LL<<(x-1));//如果是偶数个 -
    dfs(lcm, x+1, pos+1);//这个数选
    dfs(nlcm, x, pos+1);//这个数不选
}
int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
       ans = 0;
       cin >> n >> m;
       for(int i = 1; i <= m; i++)
            cin >> a[i];
        dfs(1, 1, 1);
        cout << ans << endl;
    }
    return 0;
}