题目大意:
给你一棵树,然后有q个询问(u,k),问以u为根进行深度优先搜索第k个元素是多少?
思路:
朴素做法:对于q个询问,每个询问dfs一下求第k个元素即可。---超时
正解:
我们可以先求出以1为根的所有节点的dfs序,然后记录一下第i个dfs序是谁,然后对于q个询问输出第(dfn[u] + k - 1)个dfs序是谁即可。
对于-1的情况,我们只需要开一个数组记录每颗子树儿子的数量,然后如果k > son[u]输出-1即可。
代码:
#include<iostream> #include<vector> #include<cstring> #include<algorithm> using namespace std; const int maxn = 2e6 + 10; vector<int>G[maxn]; int dfn[maxn],tot; int ans[maxn],son[maxn]; void dfs(int u,int fu){ dfn[u] = ++tot; ans[tot] = u; son[u] = 1; for(int i = 0; i < G[u].size(); i++){ int v = G[u][i]; if(v != fu){ dfs(v,u); son[u] += son[v]; } } } void solved(){ int n,q;scanf("%d%d",&n,&q); for(int i = 2; i <= n; i++){ int u;scanf("%d",&u); G[i].push_back(u); G[u].push_back(i); } dfs(1,0); while(q--){ int u,k;scanf("%d%d",&u,&k); if(k > son[u]){puts("-1");continue;} printf("%d\n",ans[dfn[u] + k - 1]); } } int main(){ solved(); return 0; }