题目:给出一棵树,问与节点v的第k个祖先相同的节点数有多少个。
分析:
寻找节点v的第k个祖先,这不就是qtree2简化版吗,但是怎么统计该祖先拥有多少个深度为k的儿子?
我们可以对于深度为d的所有节点放到一个数组中,这时需要知道的是深度为d的数组中某个连续区间都属于该子树的长度。
某棵子树的信息?这时可以考虑一下dfs序列。。。
dfs序列把整棵子树放在同一个区间,不懂dfs序的可以做做这两题:
BZOJ 1103 [POI2007]大都市meg,BZOJ 1782 [Usaco2010 Feb]slowdown 慢慢游。
于是,我们dfs时把节点x的进入时的时间戳放到深度为d的数组中,二分出询问节点v进入、出去时的时间戳所在的位置,算出该区间长度即可。
具体实现可以看代码。
#include <set> #include <map> #include <list> #include <cmath> #include <queue> #include <stack> #include <string> #include <vector> #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; typedef long long ll; typedef unsigned long long ull; #define debug puts("here") #define rep(i,n) for(int i=0;i<n;i++) #define rep1(i,n) for(int i=1;i<=n;i++) #define REP(i,a,b) for(int i=a;i<=b;i++) #define foreach(i,vec) for(unsigned i=0;i<vec.size();i++) #define pb push_back #define RD(n) scanf("%d",&n) #define RD2(x,y) scanf("%d%d",&x,&y) #define RD3(x,y,z) scanf("%d%d%d",&x,&y,&z) #define RD4(x,y,z,w) scanf("%d%d%d%d",&x,&y,&z,&w) #define All(vec) vec.begin(),vec.end() #define MP make_pair #define PII pair<int,int> #define PQ priority_queue #define cmax(x,y) x = max(x,y) #define cmin(x,y) x = min(x,y) #define Clear(x) memset(x,0,sizeof(x)) /* #pragma comment(linker, "/STACK:1024000000,1024000000") int size = 256 << 20; // 256MB char *p = (char*)malloc(size) + size; __asm__("movl %0, %%esp " :: "r"(p) ); */ /******** program ********************/ const int MAXN = 1e5+5; const int LOG = 18; vector<int> adj[MAXN]; int a[MAXN]; int dep[MAXN]; int p[MAXN][LOG+1]; int st[MAXN],ed[MAXN],tim; vector< int > vec[MAXN]; void dfs(int x,int fa,int depth){ st[x] = tim++; dep[x] = depth; vec[depth].pb( tim ); p[x][0] = fa; rep1(i,LOG) p[x][i] = p[ p[x][i-1] ][i-1]; foreach(i,adj[x]) dfs(adj[x][i],x,depth+1); ed[x] = tim++; } int main(){ #ifndef ONLINE_JUDGE freopen("sum.in","r",stdin); //freopen("sum.out","w",stdout); #endif int n; int ncase = 0; while(cin>>n){ ncase?puts("------------------------"):ncase = 1; rep1(i,n){ adj[i].clear(); vec[i].clear(); } int tot = 0; rep1(i,n){ RD(a[i]); tot += !a[i]; adj[ a[i] ].pb(i); } tim = 0; dfs(0,0,0); int m,x,k; RD(m); while(m--){ RD2(x,k); int d = dep[x]; rep(i,LOG) if( k>>i & 1 ) x = p[x][i]; if(!x){ printf("0 "); continue; } int ans = lower_bound(All(vec[d]),ed[x])-lower_bound(All(vec[d]),st[x]-1)-1; printf("%d ",ans); } cout<<endl; } return 0; }