【BZOJ】【P3823】【定情信物】【题解】【线性筛逆元】

传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=3823

有点意思的题目

脑子里yy不出来4维的情况……

把前三维打个表

发现0维个数好像是 2^n

然后就是 f[i]=f[i-1]*(n-i+1)/2i 筛逆元算就完了

UPD:数据加强被叉了T_T

原因是p的倍数没有逆元……

所以记录一下当前数中有多少个p就可以了

Code:

#include<bits/stdc++.h>using namespace std;typedef long long LL;LL n,p;LL pw(LL x,LL k){    LL ans=1;    for(;k;k>>=1){        if(k&1)ans=(ans*x)%p;        x=(x*x)%p;    }return ans;}LL f,ans;int inv[int(2e7+5)];int cnt=0;int main(){    cin>>n>>p;    ans=f=pw(2,n);cnt=0;inv[1]=1;    for(int i=2;i<=min(2*n,p-1);i++)inv[i]=(LL)(p-p/i)*inv[p%i]%p;    for(int i=1;i<=n;i++){int tmp=n-i+1;while(tmp%p==0)cnt++,tmp/=p;f=(LL)f*tmp%p;tmp=2*i;while(tmp%p==0)cnt--,tmp/=p;f=(LL)f*inv[tmp%p]%p;        ans^=cnt?0:f;    }cout<<ans<<endl;    return 0;}