题面:
示例输入:
2
9 3 5
0 8 2 1 3
1 1 1
0
示例输出:
9
1
3
4
1
1
思路:
由题干可得对于一个结点x,其子树最左边是x*k+1,最右边是x*k+k(如果x*k+k>=n的话就是n-1)。那么可以用每一层的最右边的编号减去最左边的编号累加来得到结果。
像上图一样一层一层加起来就是要求的值了
贴一段AC代码
#include<cstdio>
#include<iostream>
using namespace std;
void work(){
long long n,k,m,x,l,r;
scanf("%lld%lld%lld",&n/*结点个数*/,&k/*k叉树*/,&m/*询问数*/);
while(m--){
scanf("%lld",&x);
if(k==0){
cout<<1<<'\n';
continue;
}
else if(k==1){
cout<<n-x<<'\n';
continue;
}
long long ans=0;
l=x;
r=x;
while(l<n){
ans=ans+r-l+1;
l=l*k+1;
r=min(n-1,r*k+k);
}
cout<<ans<<'\n';
}
}
int t;
int main(){
cin>>t;
while(t--){
work();
}
return 0;
}
直接在while循环里完成上列步骤,对k==1和k==0的情况特判即可