鬼畜的多项式

写点做多项式笔记以及遇到的各种蛋疼的东西....(不懂可以Q我辣我十分愿意!

(picks博客已经成为中国多项式入门到精通的经典教程辣!详见:http://picks.logdown.com/archives

多项式乘法:

裸的fft= =复数搞搞

数论变换的多项式乘法:

对于一些要对系数取模而取的模十分鬼畜即满足存在$k$,使得$2^k | (P-1) 且 2^k > n$时那么可以用一个$P$的原根代替复数根,即用$g^{\frac{P-1}{m}}$代替fft里的单位根$\omega ^{ \frac{2 \pi i}{m} }$。其逆为$n^{-1} g^{-\frac{P-1}{m}}$。容易证明这是正确的。(可以照着算导对单位根的所需要性质然后来证明)

多项式求逆:

这个求逆指的是对一个$mod$特定次数的$x$求逆,例如求$A(x)$在模$x^n$下的逆,即求一个$B(x)$使得$A(x)B(x) \equiv 1 \pmod{x^n}$,这个用倍增的思想容易得到

$$ B_{n}(x) \equiv B_{ \left \lceil \frac{n}{2} \right \rceil } \left( 2 - A(x)B_{ \left \lceil \frac{n}{2} \right \rceil } \right) \pmod{x^n} $$

$B_{n}(x)$指的是在模$x^n$下的$A$的逆

这里一定注意在码的时候,右式得到的次数界应该大于等于$n+n$(在这里大于的部分被模掉了!)的啊...不要以为是$n+\left \lceil \frac{n}{2} \right \rceil +1$啊555,在这里wa了好多发啊QAQ

多项式除法+取模:

指的是给出$A(x), B(x)$求$A(x) = B(x)C(x) + D(x)$,其中满足$deg(A)=deg(B)+deg(C), deg(D)<deg(B)$

这里需要用点技巧orz如果能想到翻转多项式那么这题就好做了orz令$F^{R}(x)$表示将多项式$F(x)$系数前后翻转后的多项式,即$F^{R}(x) = x^{deg(F)} F(\frac{1}{x})$。那么我们将$\frac{1}{x}$先带入$A(x) = B(x)C(x) + D(x)$并乘上一个$x^{deg(A)}$最后化简能得到就能得到

$$A^R(x) \equiv B^R(x)C^R(x) \pmod{x^{deg(A)-deg(B)}}$$

取模也相应解决了

多项式开根:

指的是在模$x^n$的意义下给出$A(x)$,求$\sqrt{A(x)}$。那么能开根的条件就是$A(x)$的最低次是2的倍数且系数能被开根。

如果想到倍增这个问题也是很好解决的...最终得到

$$A \equiv \left( 2^{-1} A_{ \left \lceil \frac{n}{2} \right \rceil } + 2^{-1} A A_{ \left \lceil \frac{n}{2} \right \rceil }^{-1} \right)^2 \pmod{x^n}$$

下标的意义同求逆那里。($A_{t}$表示$A_{t}^2 \equiv A \pmod{x^t}$

要注意的和求逆那里一样,右式次数界同样是大于等于$n+n$且求逆的时候是取模$x^n$下的逆,因为$A_{ \left \lceil \frac{n}{2} \right \rceil }^{-1}$的次数界可能为$n$。

对应的例题看picks博客辣= =

 

多项式求逆:【BZOJ】3456: 城市规划

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N=130050, fN=N<<2;

const ll mo=1004535809;

ll G[35], nG[35];

int rev[fN];

ll ipow(ll a, int b) { ll x=1; for(; b; b>>=1, (a*=a)%=mo) if(b&1) (x*=a)%=mo; return x; }

void fft(ll *a, int n, int f) {

	for(int i=0; i<n; ++i) if(i<rev[i]) swap(a[i], a[rev[i]]);

	int now=-1;

	for(int m=2; m<=n; m<<=1) {

		int mid=m>>1; ++now;

		ll wn=G[now]; if(f) wn=nG[now];

		for(int i=0; i<n; i+=m) {

			ll w=1;

			for(int j=0; j<mid; ++j) {

				ll u=a[i+j], v=a[i+j+mid]*w%mo;

				a[i+j]=(u+v)%mo;

				a[i+j+mid]=(u-v+mo)%mo;

				(w*=wn)%=mo;

			}

		}

	}

}

ll tmp[fN];

void getinv(ll *A, ll *B, int n) {

	if(n==1) { B[0]=ipow(A[0], mo-2); return; }

	getinv(A, B, (n+1)>>1);

	int len=1, bl=-1, nn=(n<<1)-1;

	for(; len<nn; len<<=1, ++bl);

	for(int i=1; i<len; ++i) rev[i]=(rev[i>>1]>>1)|(((ll)i&1)<<bl);

	for(int i=0; i<n; ++i) tmp[i]=A[i]; for(int i=n; i<len; ++i) tmp[i]=0;

	fft(tmp, len, 0); fft(B, len, 0);

	for(int i=0; i<len; ++i) B[i]=B[i]*((2-tmp[i]*B[i]%mo+mo)%mo)%mo;

	fft(B, len, 1); ll nN=ipow(len, mo-2);

	for(int i=0; i<n; ++i) (B[i]*=nN)%=mo; for(int i=n; i<len; ++i) B[i]=0;

}

ll ni[N], p[N], A[fN], B[fN], nA[fN];

int main() {

	int n;

	scanf("%d", &n);

	if(n<=2) { puts("1"); return 0; }



	int len=1, bl=-1, nn=((n+1)<<1)-1;

	for(; len<nn; len<<=1, ++bl);

	G[bl]=ipow(3, (mo-1)/len); nG[bl]=ipow(G[bl], mo-2);

	for(int i=bl-1; i>=0; --i) G[i]=G[i+1]*G[i+1]%mo, nG[i]=nG[i+1]*nG[i+1]%mo;

	ni[1]=1; p[1]=1; p[0]=1;

	for(int i=2; i<=n; ++i) ni[i]=((-(mo/i)*ni[mo%i])%mo+mo)%mo;

	for(int i=2; i<=n; ++i) p[i]=p[i-1]*ni[i]%mo;

	A[0]=1, B[0]=0;

	ll last=1, C=1;

	for(int i=1; i<=n; ++i) A[i]=last*p[i]%mo, B[i]=last*p[i-1]%mo, last=last*((C<<=1)%=mo)%mo;

	getinv(A, nA, n+1);

	for(int i=1; i<len; ++i) rev[i]=(rev[i>>1]>>1)|(((ll)i&1)<<bl);

	fft(nA, len, 0); fft(B, len, 0);

	for(int i=0; i<len; ++i) (B[i]*=nA[i])%=mo;

	fft(B, len, 1); ll nN=ipow(len, mo-2);

	for(int i=0; i<=n; ++i) (B[i]*=nN)%=mo;

	ll pp=1;

	for(int i=2; i<=n; ++i) (pp*=(i-1))%=mo, (B[i]*=pp)%=mo;

	printf("%lld\n", B[n]);

	return 0;

}

  

多项式除法+取模:

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

#define CC(a, b) memset(a, 0, sizeof(int)*(b))

const int N=130005, fN=N<<2, BN=30, mo=1004535809, n_G=3;

int G[BN], nG[BN], rev[fN];

int ipow(int a, int b) { int x=1; for(; b; b>>=1, a=(ll)a*a%mo) if(b&1) x=(ll)x*a%mo; return x; }

void P(int *a, int n) { for(int i=n-1; i>=0; --i) printf("%d ", a[i]); puts(""); }

void fft_init() {

	G[21]=ipow(n_G, (mo-1)/(1<<21)); nG[21]=ipow(G[21], mo-2);

	for(int i=20; i; --i) G[i]=(ll)G[i+1]*G[i+1]%mo, nG[i]=(ll)nG[i+1]*nG[i+1]%mo;

}

int getlen(int n) {

	int len=1, b=-1;

	for(; len<n; len<<=1, ++b);

	for(int i=0; i<len; ++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<b);

	return len;

}

void fft(int *a, int n, int f) {

	for(int i=0; i<n; ++i) if(i<rev[i]) swap(a[i], a[rev[i]]);

	for(int m=2, now=1; m<=n; m<<=1, ++now) {

		int mid=m>>1, wn=G[now], w, u, v; if(f) wn=nG[now];

		for(int i=0; i<n; i+=m) {

			w=1;

			for(int j=0; j<mid; ++j) {

				u=a[i+j], v=(ll)a[i+j+mid]*w%mo;

				a[i+j]=(u+v)%mo; a[i+j+mid]=(u-v+mo)%mo;

				w=(ll)w*wn%mo;

			}

		}

	}

}

void mul(int *a, int *b, int *c, int n, int m, int &k) {

	int len=getlen((n+m)-1), nlen=ipow(len, mo-2);

	static int x[fN], y[fN];

	for(int i=0; i<n; ++i) x[i]=a[i];

	for(int i=0; i<m; ++i) y[i]=b[i];

	fft(x, len, 0); fft(y, len, 0);

	for(int i=0; i<len; ++i) y[i]=(ll)y[i]*x[i]%mo;

	fft(y, len, 1);

	k=n+m-1;

	for(int i=0; i<k; ++i) c[i]=(ll)y[i]*nlen%mo;

	while(k && c[k-1]==0) --k;

	CC(x, len); CC(y, len);

}

void getinv(int *a, int *b, int n) {

	if(n==1) { b[0]=ipow(a[0], mo-2); return; }

	getinv(a, b, (n+1)>>1);

	static int c[N], d[N];

	memcpy(c, a, sizeof(int)*(n)); memcpy(d, b, sizeof(int)*((n+1)>>1));

	int len=getlen(n+n-1), nlen=ipow(len, mo-2);

	fft(c, len, 0); fft(d, len, 0);

	for(int i=0; i<len; ++i) d[i]=(ll)d[i]*(2-(ll)d[i]*c[i]%mo+mo)%mo;

	fft(d, len, 1);

	for(int i=0; i<n; ++i) b[i]=(ll)d[i]*nlen%mo;

	CC(c, len); CC(d, len);

}

void div(int *a, int *b, int *c, int n, int m, int &k) {

	static int ar[fN], br[fN], nbr[fN];

	if(n<m) { k=1; return; }

	for(int i=0; i<n; ++i) ar[i]=a[n-i-1]; 

	for(int i=0; i<m; ++i) br[i]=b[m-i-1]; 

	k=n-m+1;

	getinv(br, nbr, k);

	int len=getlen(n+n-1), nlen=ipow(len, mo-2);

	fft(ar, len, 0); fft(nbr, len, 0);

	for(int i=0; i<len; ++i) nbr[i]=(ll)ar[i]*nbr[i]%mo;

	fft(nbr, len, 1);

	for(int i=0; i<k; ++i) c[i]=(ll)nbr[k-i-1]*nlen%mo;

	while(k && c[k-1]==0) --k;

	CC(ar, len); CC(br, len); CC(nbr, len);

}

void getmod(int *a, int *b, int *r, int n, int m, int &k) {

	static int c[fN], t[fN], clen;

	div(a, b, c, n, m, clen);

	int len=getlen((clen+m)-1), nlen=ipow(len, mo-2);

	for(int i=0; i<m; ++i) t[i]=b[i];

	fft(t, len, 0); fft(c, len, 0);

	for(int i=0; i<len; ++i) c[i]=(ll)c[i]*t[i]%mo;

	fft(c, len, 1);

	for(int i=0; i<n; ++i) r[i]=(a[i]-(ll)c[i]*nlen%mo+mo)%mo;

	k=n; while(k && r[k-1]==0) --k;

	CC(c, len); CC(t, len); 

}

int main() {

	fft_init();



	return 0;

}

  

多项式开根:【CF】438E. The Child and Binary Tree

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N=(1e5+10)*4, mo=998244353;

int two, G[30], nG[30], rev[N];

int ipow(int a, int b) { int x=1; for(; b; b>>=1, a=(ll)a*a%mo) if(b&1) x=(ll)x*a%mo; return x; }

void fft_init() {

	two=ipow(2, mo-2); G[23]=ipow(3, (mo-1)/(1<<23)); nG[23]=ipow(G[23], mo-2);

	for(int i=22; i; --i) G[i]=(ll)G[i+1]*G[i+1]%mo, nG[i]=(ll)nG[i+1]*nG[i+1]%mo;

}

int getlen(int n) {

	int len=1, bl=-1;

	for(; len<n; len<<=1, ++bl);

	for(int i=1; i<len; ++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<bl);

	return len;

}

void fft(int *a, int n, int f) {

	for(int i=0; i<n; ++i) if(i<rev[i]) swap(a[i], a[rev[i]]);

	for(int m=2, now=1; m<=n; m<<=1, ++now) {

		int mid=m>>1, w=1, wn=G[now], u, v; if(f) wn=nG[now];

		for(int i=0; i<n; i+=m, w=1) 

			for(int j=0; j<mid; ++j) {

				u=a[i+j], v=(ll)a[i+j+mid]*w%mo;

				a[i+j]=(u+v)%mo; a[i+j+mid]=(u-v+mo)%mo; w=(ll)w*wn%mo;

			}

	}

}

void getinv(int *a, int *b, int n) {

	if(n==1) { b[0]=ipow(a[0], mo-2); return; }

	getinv(a, b, (n+1)>>1);

	static int c[N], d[N];

	memcpy(c, a, sizeof(int)*(n)); memcpy(d, b, sizeof(int)*((n+1)>>1));

	int len=getlen(n+n-1), nlen=ipow(len, mo-2);

	fft(c, len, 0); fft(d, len, 0);

	for(int i=0; i<len; ++i) d[i]=(ll)d[i]*(2-(ll)d[i]*c[i]%mo+mo)%mo;

	fft(d, len, 1);

	for(int i=0; i<n; ++i) b[i]=(ll)d[i]*nlen%mo;

	memset(c, 0, sizeof(int)*(len)); memset(d, 0, sizeof(int)*(len));

}

void getroot(int *a, int *b, int n) {

	if(n==1) { b[0]=sqrt(a[0]); return; }

	getroot(a, b, (n+1)>>1);

	static int c[N], d[N];

	memcpy(c, a, sizeof(int)*(n));

	getinv(b, d, n);

	int len=getlen(n+n-1), nlen=ipow(len, mo-2);

	fft(c, len, 0); fft(d, len, 0);

	for(int i=0; i<len; ++i) d[i]=(ll)c[i]*d[i]%mo;

	fft(d, len, 1);

	for(int i=0; i<n; ++i) b[i]=(ll)two*((b[i]+(ll)d[i]*nlen%mo)%mo)%mo;

	memset(d, 0, sizeof(int)*(len)); memset(c, 0, sizeof(int)*(len));

}

int a[N], b[N];

int main() {

	fft_init();

	int m, n; scanf("%d%d", &n, &m);

	for(int i=0; i<n; ++i) { int x; scanf("%d", &x); if(x<=m) a[x]=mo-4; }

	a[0]=1;

	getroot(a, b, m+1);

	b[0]=(b[0]+1)%mo;

	getinv(b, a, m+1);

	for(int i=1; i<=m; ++i) printf("%d\n", (a[i]<<1)%mo);

	return 0;

}

  

你可能感兴趣的:(鬼畜的多项式)