清北集训Day6T1(生成函数)

听rqy说可以用生成函数做，感觉比较有意思

我们考虑在DP转移的时候，

$5,7,9$这三个数是没有限制的

因此他们出现的次数用01串表示的话就是$1111111111111111......$

$3,5$这两个数只能出现偶数次且必须出现

因此他们出现的次数用01串表示的话是$0010101010101010101....$

因为是组合计数问题，我们考虑用指数型生成函数来搞

对于第一个肯定就是$e^x$

对于第二个，我们首先用$\frac{e^x+e^{-x}}{2}$构造出$1010101010.....$

然后再减个$1$就好了

这样的话我们不难得到答案的方案实际就是

$\left( e^{x}\right) ^{3}\left( \dfrac {e^{x}+e^{-x}}{2}-1\right) ^{2}$

然后暴力推推推就可以得到

$\dfrac {1}{4}e^{5x}+\dfrac {1}{4}e+\dfrac {6}{4}e^{3x}-\dfrac {4}{4}e^{4x}-\dfrac {4}{4}e^{2x}$

然后快速幂搞一搞就好了

生成函数好神奇QWQ。。。

#include<cstdio>

#include<iostream>

#define int long long

using namespace std;

const int MAXN=1e6+;

const int mod=1e9+;

int a[MAXN]={,,,,,};

int k[MAXN]={,,,,-,-};

int fastpow(int a,int p)

{

    int base=;

    while(p)

    {

        if(p&) base=(base*a)%mod;

        a=(a*a)%mod;

        p>>=;

    }

    return base%mod;

}

main()

{

    int N,ans=;

    cin>>N;

    for(int i=;i<=;i++)

        ans =( ans + fastpow(a[i], N) * k[i] ) %mod;

    cout<<( ans * ( (mod + ) /  ) %mod + mod ) %mod;

    return ;

}