【BUAA 610】前前前世【前缀和计数】

题目:给出一棵无穷节点的二叉树,根为1,对于节点i,他的两个子节点为2i和2i+1,求(x,y)满足x是y的父节点的父节点的父节点,并且x%k=1&&y%k=1的对数

题解：

x%k=1,y=8x+d(0<=d<8)  => (8+d)%k=1

用cnt[d]表示满足条件的d的个数

由上式可知k>=15时,无解

用f[k][i][j]表示长度为2^i且起点%k=j的区间中%k=1的个数

f[k][i][j]=f[k][i-1][j]+f[k][i-1][j+2^(i-1)]

这样ans=sigma(f[k][1..n-3][p%k]*cnt[k])

我们可以在O(n)的时间内处理每个回答

我们也可以处理f的前缀和

g[k][i][j]表示起点%k=j的到第i层为止的f的前缀和

g[k][i][j]=g[k][i-1][j]+f[k][i-1][j*2^(i-1)%k]

#include<cstdio>
#include<iostream>
using namespace std;
#define ll long long
const int N=5e4+2;
const int M=3e5+10;
const int Mbit=1e6+10;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
ll Pow[N],f[15][N][15],g[15][N][15];
int cnt[15];
const int MAXN=50000+5;
const int MOD=1e9+7;
ll a[20][MAXN][20];
void In(){freopen("in.in","r",stdin);}
void init()
{
    for(int k=2;k<15;k++){
        ll ret=1;
        f[k][0][1]=1;
        for(int i=1;i<N;i++){
            for(int j=0;j<k;j++)
                f[k][i][j]=(f[k][i-1][j]+f[k][i-1][(j+ret)%k])%mod;
            ret=ret*2%k;
        }
    }
    //f[k][i][j] 2^i个数 起点为j %k=1的个数
    //g[k][i][j] k个数 到第i层 起点%k=j %k=1的前缀和
    for(int k=2;k<15;k++){
        ll ret=1;
        for(int i=1;i<N;i++){
            for(int j=0;j<k;j++)
                g[k][i][j]=(g[k][i-1][j]+f[k][i-1][1ll*ret*j%k])%mod;
            ret=ret*2%k;
        }
    }
    for(int i=1;i<15;i++)
        for(int j=7;j<=14;j++)
            if(j%i==0)cnt[i]++;
}
int main()
{
   //In();
    init();
    int T,kase=0;
    ll k,n,p;
    scanf("%d",&T);
    while(T--){
        scanf("%lld%lld%lld",&k,&n,&p);
        if(k>=15){
            puts("0");
            continue;
        }
        if(n>=3)printf("%lld\n",1ll*g[k][n-3][p%k]*cnt[k]%mod);
        else  printf("0\n");
    }
    return 0;
}