首先,这个题跟一般的容斥题不太一样。
题意:有m个石子围成一圈, 有n只青蛙从跳石子, 都从0号石子开始, 每只能越过xi个石子。问所有被至少踩过一次的石子的序号之和。
分析:多画几组数据可以看出:走的石头的编号就是步数跟石头数的最大共约数,那么第一想法就是容斥,最普通不过了,例如第一个样例,明显就是m以内2的倍数+3的倍数-2和3的倍数。然而失望的是n的范围太大,无法是直接这样算。换种思路,算出m的所有约数,因为上面计算过程中如果哪个数多做出了贡献,那么这个数肯定是m的约数,然后计算贡献即可,例如2贡献了1次,3又贡献了一次,所以6加起来一共贡献了2次,然后减去差值即可。
以前没见过这种容斥,这次学到了。
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e4+10;
typedef long long ll;
ll a[maxn];
int vis[maxn];
int main(){
int T;
scanf("%d",&T);
for(int cs=1;cs<=T;cs++){
int n;
ll m;
vector<ll> v;
scanf("%d%lld",&n,&m);
for(ll i=1;i*i<=m;i++){
if(m%i==0){
v.push_back(i);
if(i*i!=m) v.push_back(m/i);
}
}
sort(v.begin(),v.end());
memset(vis,0,sizeof(vis));
for(int i=0;i<n;i++){
scanf("%lld",&a[i]);
ll res=__gcd(a[i],m);
for(int j=0;j<v.size();j++){
if(v[j]%res==0){
vis[j]=1; //标记参与产生贡献的因子。
}
}
}
ll ans=0;
for(int i=0;i<v.size();i++){
if(vis[i]){
ll num=m/v[i];
ans+=(0+(num-1)*v[i])*num/2*vis[i];
for(int j=i+1;j<v.size();j++){
if(v[j]%v[i]==0){
vis[j]-=vis[i]; //因为计算v[i]的时候,v[j]的贡献计算了v[i]次,所以要减去。
}
}
}
}
printf("Case #%d: %lld\n",cs,ans);
}
}