設$s[i]$為進行$i$次加密後所有奶牛數字的和,有$s[i]=(n-1)s[i-1]$。
設$c[i]$為某頭固定的奶牛進行$i$次加密後的數字,
若$i$為奇數,有:
\[c[i]=((1-n)^0+(1-n)^1+...+(1-n)^{T-1})s-c[0]=\frac{(1-(1-n)^T)s}{n}-c[0]\]
若$i$為偶數,有:
\[c[i]=-((1-n)^0+(1-n)^1+...+(1-n)^{T-1})s+c[0]=-\frac{(1-(1-n)^T)s}{n}+c[0]\]
預先算出$\frac{(1-(1-n)^T)s}{n}$的值後直接$O(n)$計算即可。
#include<cstdio> #define P 98765431 typedef long long ll; int n,T,i,c[50000],y;ll s,x; ll pow(ll a,ll b){ll t=1;for(;b;b>>=1,a=a*a%P)if(b&1)t=t*a%P;return t;} int main(){ for(scanf("%d%d",&n,&T),x=(1LL-pow(1-n,T))*pow(n,P-2)%P;i<n;i++)scanf("%d",&c[i]),s=(s+c[i])%P; for(x=x*s%P,i=0;i<n;i++)y=((x-c[i])%P+P)%P,printf("%d\n",T&1?y:P-y); return 0; }