AtCoder Beginner Contest 137 F

AtCoder Beginner Contest 137 F

数论鬼题(虽然不算特别数论)

希望你在浏览这篇题解前已经知道了费马小定理

利用用费马小定理构造函数\(g(x)=(x-i)^{P-1}\)

\[x=i,g(x)=0
\]

\[x\ne i ,g(x)=1
\]

则我们可以构造

\[f(x)=\sum^{i=0}_{P-1}(-a_i*(x-i)^{P-1}+a_i)
\]

对于第\(i\)条式子当且仅当\(a_i=1 \ and \ x=i\)时取到\(1\)

代码写的比较奇怪

const int N=3100;

int P,a[N];

int po[N]={1},Inv[N]={1,1};
int b[N];
int C(int n,int m){
	if(n<0||m<0||n<m) return 0;
	return po[n]*Inv[m]%P*Inv[n-m]%P;
}

int fl=0;
int main(){
	P=rd();
	rep(i,1,P+1) po[i]=po[i-1]*i%P;
	rep(i,2,P-1) Inv[i]=(P-P/i)*Inv[P%i]%P;
	rep(i,2,P+1) Inv[i]=Inv[i]*Inv[i-1]%P;
	rep(i,0,P-1) {
		if(rd()) {
			fl=1;
			int t=1;
			drep(j,P-1,0) b[j]+=C(P-1,j)%P*t%P,t=t*(P-i)%P;
		} else b[0]++;
	}
	rep(i,0,P-1) {
		int x=(P-b[i])%P;
		x=(x%P+P)%P;
		printf("%d%c",x,(i==P-1)?'\n':' ');
	}
}