P2553 [AHOI2001]多项式乘法 （字符串处理，FFT，模拟好题）

P2553 [AHOI2001]多项式乘法

打模拟加FFT

​ 读入字符串后照例从头开始往后，如果找到一个数字，那么它肯定是 x，也就是系数，此时我们用读入优化时的方式将指针往后移，并把它存下来，然后此时指针指向的位置应该是 x 的最后一个字符，判断下个位置的字符是不是 'a' ，如果是的话意味着我们刚才存的 x 是这一项的系数，而我们将指针后移 3 位，分别经过 x 的最后一位、'a' 和 '^' ，到达这一项的次数 y 的第一位，依然使用读入优化的方式存下 y 。此时记录一下是应该放到 A中还是 B 中，然后放进去即可。

​ 存下之后，如果正好第二次经过 ')'，意味着这个多项式结束了（因为保证一组多项式只有一个 '*'），照例跑 FFT 即可。

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const int N = (1<<21)+5;
const double PI=acos(-1);
struct Complex
{
    double x, y;
    Complex operator+(const Complex &o) const{return{x+o.x,y+o.y};}  
    Complex operator-(const Complex &o) const{return{x-o.x,y-o.y};}  
    Complex operator*(const Complex &o) const{return{x*o.x-y*o.y,x*o.y+y*o.x};} 
}A[N],B[N];
int rev[N];
void init(int k)
{
    int s=1<<k;
    for(int i=1;i<s;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(k-1));
}
void fft(Complex *a,int n,int inv)
{
    for(int i=0;i<n;i++) if(i<rev[i]) swap(a[i],a[rev[i]]);
    for(int len=1;len<n;len<<=1)
    {
        Complex Wn=Complex({cos(PI/len),inv*sin(PI/len)});
        for(int i=0;i<n;i+=len*2)
        {
            Complex w=Complex({1,0});
            for(int j=0;j<len;j++,w=w*Wn)
            {
                Complex x=a[i+j],y=w*a[i+j+len];
                a[i+j]=x+y,a[i+j+len]=x-y;
            }
        }
    }
    if(inv==-1) for(int i=0;i<n;i++) A[i].x = A[i].x/n+0.5; // 精度
}
// --------FFT
int ans[N];
void pre(int n)
{
    int s=2, k=1;
    while(s<=n) k++, s<<=1;
    init(k);
    fft(A,s,1); fft(B,s,1);
    for(int i=0;i<s;i++) A[i] = A[i]*B[i];
    fft(A,s,-1);
    for(int i=n;i>=0;i--)
    {
		ll x=(ll)(A[i].x);
		if(x)
		{
			if(i) printf("%llda^%d+",x,i);
			else printf("%lld\n",x);
		} 
    }
}
int main()
{
    string s1;
    while(getline(cin,s1))
    {
        memset(A,0,sizeof(A));
        memset(B,0,sizeof(B));
        int len=s1.size(), fg=0, n=0, m=0;
        for(int i=0;i<len;i++)
        {
            if(isdigit(s1[i]))
            {
                int x=0, y=0;
                while(isdigit(s1[i])) x=(x<<3)+(x<<1)+(s1[i]-48), i++; 
                // x=x*10+(s1[i]-'a')
                i--;
                if(s1[i+1]=='a') 
                {
                    i += 3;
                    while(isdigit(s1[i])) y=(y<<3)+(y<<1)+(s1[i]-48), i++;
                    i--;
                }
                if(!fg) { n=max(n, y); A[y].x = double(x); }
                else { m=max(m,y); B[y].x = double(x); }
            }
            else if(s1[i]==')')
            {
                fg ^= 1;
                if(!fg) pre(n+m);
            }
        }
    }
	return 0;
}