题目:给出一棵无穷节点的二叉树,根为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;
}