Educational Codeforces Round 57 (Rated for Div. 2) G. Lucky Tickets（NTT）

题目链接：http://codeforces.com/contest/1096/problem/G

 

挺容易想到NTT思路的，然后注意处理数论变换后的数组，直接进行快速幂即可，最后再进行逆变换

 

代码：

#include<bits/stdc++.h>
#define xx first
#define yy second
#define mp make_pair
#define pb push_back
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int MAXN=(1<<22)+5;
namespace NTT
{
    typedef long long ll;
    const int MOD=998244353;
    const int g=3;
    ll qpow(ll a,ll k)
    {
        ll res=1LL;
        while(k>0)
        {
            if(k&1)res=res*a%MOD;
            a=a*a%MOD;
            k>>=1;
        }
        return res;
    }
    void change(ll y[],int len)
    {
        for(int i=1,j=len/2;i<len-1;i++)
        {
            if(i<j)swap(y[i],y[j]);
            int k=len/2;
            while(j>=k)
            {
                j-=k;
                k/=2;
            }
            if(j<k)j+=k;
        }
    }
    void ntt(ll y[],int len,int on)
    {
        change(y,len);
        for(int h=2;h<=len;h<<=1)
        {
            ll wn=qpow(g,(MOD-1)/h);
            if(on==-1)wn=qpow(wn,MOD-2);
            for(int j=0;j<len;j+=h)
            {
                ll w=1LL;
                for(int k=j;k<j+h/2;k++)
                {
                    ll u=y[k];
                    ll t=w*y[k+h/2]%MOD;
                    y[k]=(u+t)%MOD;
                    y[k+h/2]=(u-t+MOD)%MOD;
                    w=w*wn%MOD;
                }
            }
        }
        if(on==-1)
        {
            ll t=qpow(len,MOD-2);
            for(int i=0;i<len;i++)
                y[i]=y[i]*t%MOD;
        }
    }
    void cal(ll x1[],ll n,ll tot,ll &ans)
    {
        int len=1;
        while(len<2*tot)len<<=1;
        ntt(x1,len,1);
        for(int i=0;i<len;i++)
            x1[i]=qpow(x1[i],n/2);
        ntt(x1,len,-1);
		for(int i=0;i<len;i++)
			ans=(ans+x1[i]*x1[i])%MOD;
    }
}
using namespace NTT;
int n,k;
int a[MAXN];
ll x[MAXN];
int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	scanf("%d%d",&n,&k);
	for(int i=1;i<=k;i++)
	{
		int u;
		scanf("%d",&u);
		x[u]=1;
	}
	ll ans=0;
	cal(x,n,n*5,ans);
	printf("%lld\n",ans);
	return 0;
}