51Nod-1439-互质对

ACM模版

描述

题解

思路十分清楚，我们只需要在每次操作后加上或者减去该元素与集合中元素互质的个数。

难点是，我们不可能暴力求出该数和集合中所有元素的互质个数，这里需要用到容斥的思路，我们可以考虑将每个数进行合数拆分，之所以这样是因为假如 $a、b$ 互质，那么 $a$ 约去一个质因子，$a / p、b$ 一定不再互质。那么我们就需要对每个数的质因子进行组合，但是存在一个问题便是增加或者减去的部分会有重复，比如说 $a$ 既是 $2$ 又是 $5$ 的倍数，那么在操作过程中，先增减所有是 $2$ 倍数的元素，再增减所有是 5 <script type="math/tex" id="MathJax-Element-359">5</script> 的倍数，……，这样一定会存在重复的情况，当然，很明显这是容斥的规律，所以我们可以根据不同的组合进行搜索，然后通过容斥控制增减。

代码

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <vector>

using namespace std;

const int MAXN = 5e5 + 5;

int n, q, tag;
long long ans;
int cnt[MAXN];
int vis[MAXN];
int a[MAXN];
int flag[MAXN];
vector<int> prime[MAXN];

void dfs(int pos, int mul, int sym, int flag)
{
    if (pos == prime[tag].size())
    {
        if (flag == -1)
        {
            cnt[mul]--;
        }
        ans += sym * cnt[mul];

        if (flag == 1)
        {
            cnt[mul]++;
        }

        return ;
    }

    dfs(pos + 1, mul, sym, flag);
    dfs(pos + 1, mul * prime[tag][pos], -sym, flag);
}

void init()
{
    for (int i = 2; i < MAXN; i++)
    {
        if (vis[i])
        {
            continue;
        }
        for (int j = i; j < MAXN; j = j + i)
        {
            vis[j] = 1;
            prime[j].push_back(i);
        }
    }
}

int main()
{
    init();

    cin >> n >> q;
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", a + i);
    }

    ans = 0;
    int x;
    for (int i = 1; i <= q; i++)
    {
        scanf("%d", &x);
        if (flag[x] == 0)
        {
            flag[x] = 1;
            tag = a[x];
            dfs(0, 1, 1, 1);
        }
        else
        {
            flag[x] = 0;
            tag = a[x];
            dfs(0, 1, -1, -1);
        }

        printf("%lld\n", ans);
    }

    return 0;
}