Educational Codeforces Round 81 (Rated for Div. 2) D. Same GCDs （容斥）

题目链接

题面：

题意：
给定两个整数 $a$，$m$。
询问满足 $gcd(a,m)=gcd(a+x,m)$ 且 $0\le x<m$ 的不同$x$ 的个数。

题解：
我们设 $g=gcd(a,m)$，那么我们有 $gcd(k1\ast g+x,k2\ast g)=g$，即 $x=k3\ast g$ 且 $0\le k3\ast g<m$ ，$0\le k3<m/g=k2$

$gcd(k1\ast g+k3\ast g,k2\ast g)=g$ ，有 $gcd(k1+k3,k2)=1$，$k1>=1$。

$gcd(k1+k3,k2)=1$ 等价于 $gcd((k1+k3)modk2,k2)=1$，又 $k3\in [0,k2]$。

实际上就是求 $[1,k2]$ 中有多少个数与 $k2$ 互质。

很明显欧拉函数求解即可。

我们还可以用容斥来求。

我们将 $k2$ 的因子从小到大排序，排序后，其因子的关系形成一张 $DAG$。

我们设 $cnt[i]$ 为 $[1,k2]$ 中的数 与$k2$ 的 $gcd$ 为 $k2$ 第 $i$ 个因子的数的个数。

初始化时，我们将 $cnt[i]$ 初始化为$[1,k2]$ 中的数 与$k2$ 的 $gcd$ 大于等于 $k2$ 第 $i$ 个因子的数的个数。

那么 $cnt[1]$ 即为所求 $gcd=1$

那么很容易得到一个 $O(sum(fac{)}^2)$ 的容斥算法。

代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<queue>
#include<bitset>
#include<map>
#include<unordered_map>
#include<unordered_set>
#include<set>
#include<ctime>
#define ui unsigned int
#define ll long long
#define llu unsigned ll
#define ld long double
#define pr make_pair
#define pb push_back
#define lc (cnt<<1)
#define rc (cnt<<1|1)
#define len(x)  (t[(x)].r-t[(x)].l+1)
#define tmid ((l+r)>>1)
#define fhead(x) for(int i=head[(x)];i;i=nt[i])
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)>(y)?(y):(x))
using namespace std;

const int inf=0x3f3f3f3f;
const ll lnf=0x3f3f3f3f3f3f3f3f;
const double dnf=1e18;
const double alpha=0.75;
const int mod=998244353;
const double eps=1e-8;
const double pi=acos(-1.0);
const int hp=13331;
const int maxn=100100;
const int maxm=100100;
const int maxp=100100;
const int up=1100;

ll fac[maxn],cnt[maxn],tot=0;

ll gcd(ll a,ll b)
{
    if(b==0) return a;
    return gcd(b,a%b);
}

void gao(ll n)
{
    tot=0;
    for(ll i=1;i*i<=n;i++)
    {
        if(n%i) continue;
        fac[++tot]=i;
        if(i*i!=n) fac[++tot]=n/i;
    }
}

int main(void)
{
    int tt;
    scanf("%d",&tt);
    while(tt--)
    {
        ll a,m;
        scanf("%lld%lld",&a,&m);
        ll g=gcd(a,m);
        a/=g,m/=g;
        gao(m);
        sort(fac+1,fac+tot+1);
        for(int i=1;i<=tot;i++)
            cnt[i]=m/fac[i];
        for(int i=tot;i>=1;i--)
        {
            for(int j=i+1;j<=tot;j++)
                if(fac[j]%fac[i]==0)
                    cnt[i]-=cnt[j];
        }
        printf("%lld\n",cnt[1]);
    }
    return 0;
}