题目大意:一串项链由n个戒指组成,对于每个戒指,一共有M个点,R种颜色,且旋转后相同的戒指是相同的,然后一串项链又由N个戒指组成,同时要满足相邻的两个戒指不能相同,这串项链上某个位置插入了一个特殊的东西,且如果特殊的东西插入的地方不同,即使戒指都是相同的,这两串项链也是不同的,求一共有多少不同的爱之项链。
思路:首先可以求出一共有多少种不同的戒指,又由于有那个特殊的东西,相当于这串项链即使旋转后相同,但特殊的东西插入的位置也肯定是不同的,因此即不考虑旋转,只考虑相邻位置不同的爱之项链的方案数。
令ans表示有多少种不同的戒指。
然后这样就可以运用容斥原理。对于第i个戒指和第i+1个戒指相同,可以看成第i个限制,然后第n个限制即第n个戒指和第1个限制不同,那么即要求满足所有限制的方案数。令Si为满足第i个限制的方案数的集合,那么即求S1~Sn的交,也就是其补集的并,那么就是总方案数减去所有不满足任意一个限制数加上所有不满足任意二个限制数......
然后这个式子是怎样的呢,首先总方案数显然就是ans^n,然后不满足任意一个限制数即有两个相邻的相同,可以看成一条边连接的两个点相同,就是C(n,1)*ans^(n-1),以此类推,然后可以得到:sigma(0<=i<=n,(-1)^i*C(n,i)*ans^(n-i)),然后发现这就是二项式定理,就可以得到(ans-1)^n, 但真的就是这样吗?当i=n时,就会有一个Bug,用公式算得的答案是(-1)^n,然而n个限制均不满足时的情况即所有颜色都一样,有ans种,因此还要加上(n&1?1-m:m-1)才行。
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<cmath> 6 using namespace std; 7 #define p 3214567 8 #define maxn 200020 9 10 int m,r,tot; 11 int prime[maxn],phi[maxn]; 12 bool isprime[maxn]; 13 long long n; 14 15 int power(int a,long long k){ 16 if (k==0) return 1; 17 if (k==1) return a%p; 18 int x=power(a,k/2),ans=1ll*x*x%p; 19 if (k&1) ans=1ll*a*ans%p; 20 return ans; 21 } 22 23 int fphi(int x){ 24 int ans=x; 25 for (int i=2;i*i<=x;i++) 26 if (x%i==0){ 27 ans=ans-ans/i; 28 while (x%i==0) x/=i; 29 } 30 if (x!=1) ans=ans-ans/x; 31 return ans; 32 } 33 34 int main(){ 35 scanf("%lld%d%d",&n,&m,&r);tot=0,memset(isprime,1,sizeof(isprime)),phi[1]=1; 36 for (int i=2;i<maxn;i++){ 37 if (isprime[i]) prime[++tot]=i,phi[i]=i-1; 38 for (int j=1;j<=tot && i*prime[j]<maxn;j++){ 39 isprime[i*prime[j]]=0; 40 if (i%prime[j]==0){ 41 phi[i*prime[j]]=phi[i]*prime[j]; 42 break; 43 } 44 phi[i*prime[j]]=phi[i]*(prime[j]-1); 45 } 46 } 47 int ans=0; 48 for (int i=1;i*i<=m;i++) 49 if (m%i==0){ 50 ans=(ans+1ll*power(r,i)*fphi(m/i)%p)%p; 51 if (i*i!=m) ans=(ans+1ll*power(r,m/i)*phi[i]%p)%p; 52 } 53 ans=1LL*ans*power(m,p-2)%p; 54 int t=(1ll*power(ans-1,n)+(n&1?1-ans:ans-1))%p; 55 printf("%d ",(t+p)%p); 56 return 0; 57 }