CodeForces 300 D.Painting Square（dp+NTT）

Description

给出一个$n\times n$的棋盘，每次操作选取一个方形小棋盘将其分割成四个方形小棋盘，问进行$k$次操作有多少种不同的方案

Input

第一行一整数$q$表示查询数，之后$q$行每行输入两个整数$n,k$表示查询将$n\times n$的棋盘操作$k$次的方案数$(1\le q\le 10^5,1\le n\le 10^9,0\le k\le 1000)$

Output

对于每次查询，输出方案数，结果模$7340033=7\cdot 2^{20}+1$

Sample Input

8
1 0
1 1
3 0
3 1
2 0
2 1
3 2
7 2

Sample Output

1
0
1
1
1
0
0
4

Solution

若想将$n\times n$的棋盘分成四个方形小棋盘，显然$n$是奇数，设$n=2m+1$，则四个小棋盘均为$m\times m$的，假设$n$的二进制表示后面有连续的$s$个$1$，分割一次后，就变成四个二进制表示后有连续$s-1$个$1$的小棋盘，故一个棋盘的分割方案只与其边长二进制表示后连续$1$的个数有关，即只要两个数字二进制表示后连续$1$的数量相同，那么其分割方案数相同

令$f[i][j]$为二进制表示后连续$i$个$1$的棋盘边长被分割$j$次的方案数，对其分割一次后变成四个二进制表示后连续$i-1$个$1$的小棋盘，这些小棋盘各自再分割，分割次数之和为$j-1$，故有$f[i][j]=\sum\limits_{a+b+c+d=j-1}f[i-1][a]\cdot f[i-1][b]\cdot f[i-1][c]\cdot f[i-1][d]$，把$f[i]$看作多项式则有$f[i]=f^4[i-1]$，做$NTT$即可快速由$f[i-1]$得到$f[i]$，对于每次查询，只要求出$n$二进制表示后有多少连续的$1$的个数（设为$s$）,则答案为$f[s][k]$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=1000005;
#define mod 1000000007
int fact[maxn],inv[maxn];
void init(int n=1e6)
{
    fact[0]=1;
    for(int i=1;i<=n;i++)fact[i]=(ll)i*fact[i-1]%mod;
    inv[1]=1;
    for(int i=2;i<=n;i++)inv[i]=mod-(ll)(mod/i)*inv[mod%i]%mod;
    inv[0]=1;
    for(int i=1;i<=n;i++)inv[i]=(ll)inv[i-1]*inv[i]%mod;
}
int C(int n,int m)
{
    return (ll)fact[n]*inv[m]%mod*inv[n-m]%mod;
}
bool check(int n,int a,int b)
{
    while(n)
    {
        if(n%10!=a&&n%10!=b)return 0;
        n/=10;
    }
    return 1;
}
int main()
{
    init();
    int a,b,n;
    while(~scanf("%d%d%d",&a,&b,&n))
    {
        int ans=0;
        for(int i=0;i<=n;i++)
            if(check(i*a+(n-i)*b,a,b))ans=(ans+C(n,i))%mod;
        printf("%d\n",ans);
    }
    return 0;
}