CCPC-Wannafly Winter Camp Day5 E.Fast Kronecker Transform(离散化+NTT)

题目

思路来源

dls

题解

先离散化,把a和b搞进一个序列,然后排序去重,把a和b赋为rank值

然后遍历这个不重复的序列,两个大于1W就建2e5数列用NTT搞,否则直接暴力

 

既然要在ai和bj值相同,下标i+j==k时求i*j的值,

那么就把位置i和j分别放入值的vector里,对相同值的vector进行NTT

注意多项式乘法一定会遍历所有x^{i}*x^{j}==x^{k}的情形,所以就令f(x)的第i位为i,g(x)的第j位为j

是一种很自然的做法,注意多项式乘法的定义

 

这个1W卡的真是刚刚好啊……昨天T了若干发

注意 NTT里面取模不要判if 直接模

上面的是有if的 下面的是没if的

NTT和FFT类似,处理整数取模的情况,

可以把大素数搞成它的原根,998244353的原根是3

FFT卡精度过不了

代码

#include 
#include  
#include 
#include 
#include 
#include 
#include  
#define g 3
using namespace std;
typedef long long ll;
const double PI = acos(-1.0);
const int mod=998244353;
const int maxn=1e5+10;
const int G=3;
ll inv,N;
ll n,m,tmp1[maxn*3],tmp2[maxn*3],rev[maxn*3],back[maxn*3];
vectorq[maxn*2],p[maxn*2];
vectornum;
ll a[maxn],b[maxn],c[maxn*2];
ll power(ll x,ll y)
{
    ll res=1ll;
    for(;y;y>>=1)
    {
        if(y&1)res=res*x%mod;
        x=x*x%mod;
    }
    return res;
}
void init()
{
    int len=0;
    while((n+m+2)>=(1<>=1;
        }
        back[i]=rev[i]=pos;
    }
}
void init2()
{
	for(int i=0;ii)
        {
            swap(a[i],a[rev[i]]);
        }
    }
    for(int i=2;i<=n;i<<=1)
    {
        ll mid=i>>1;
        ll wn=power(G,(mod-1)/i);
        if(re) wn=power(wn,(mod-2));
        for(int j=0;j

你可能感兴趣的:(FFT/NTT/FWT/FMT)