Codeforces Round #498 (Div. 3)E. Military Problem

dfs and similar

题目链接

题解：

/*
Keep clam  Believe youself
*/
#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<queue>
#include<map>
#include<vector>
#include<stack>
#include<set>
#include<cmath>
#define Xiaobo main
using namespace std;
const int maxn=2e5+7;
const int mod=1e9+7;
const double eps=1e-15;
const double pi=acos(-1);
const int INF=0x3f3f3f;
typedef long long ll;
ll read(){ll c = getchar(),Nig = 1,x = 0;while(!isdigit(c) && c!='-')c = getchar();if(c == '-')Nig = -1,c = getchar();while(isdigit(c))x = ((x<<1) + (x<<3)) + (c^'0'),c = getchar();return Nig*x;}
ll gcd(ll a,ll b){ return b==0?a:gcd(b,a%b);}
int jin[maxn],out[maxn],a[maxn];
int num[maxn];
vector<int>G[maxn];
int cnt=0;
/*PS*/
/*
 本人不太会dfs，只会简单的搜索
 对于本题思路就是一定要预处理所有情况才能在q次询问中不超时
 主要的思路就是记忆化搜索，设置一个数组jin 记录的是值x第一次搜索时的步数
 num[]数组记录第cnt步时候对应的x值
 然后对x分支下的孩子进行搜索，当搜索完所有分支后用 out[]数组记录对应的x值
 所处的步数  通过回溯可以记录从x开始搜索最大能达到的步数
 然后对于每q次询问 
 判断jin[x]+k-1是否大于out[x]如果不大于就输出从x开始搜索k-1次所对应步数
 即为num[jin[x]+k-1]
 总体思路通过步数之间的差值来输出对应的数值，也可以理解为把每步所对应的的x
 都记录了下来剩下的就是判断从第几步开始搜索的x和搜索到k-1次所对应的值 
*/
void dfs(int x) {
	jin[x]=++cnt;
	num[cnt]=x;
	for(int i=0;i<G[x].size();i++) {
		dfs(G[x][i]);
	}
	out[x]=cnt;
}
int Xiaobo()
{
	int n,q;
	n=read();
	q=read();
	for(int i=2;i<=n;i++) {
		int k;
		scanf("%d",&k);
		G[k].push_back(i);
	}
	dfs(1);
	while(q--) {
		int u,k;
		cin>>u>>k;
		if(jin[u]+k-1>out[u]) {
			printf("-1\n");
		}
		else {
			cout<<num[jin[u]+k-1]<<'\n';
		}
	}
}

PS: 第一次尝试超时不过dfs思路是对的

/*
Keep clam  Believe youself
*/
#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<queue>
#include<map>
#include<vector>
#include<stack>
#include<set>
#include<cmath>
#define Xiaobo main
using namespace std;
const int maxn=2e5+7;
const int mod=1e9+7;
const double eps=1e-15;
const double pi=acos(-1);
const int INF=0x3f3f3f;
typedef long long ll;
ll read(){ll c = getchar(),Nig = 1,x = 0;while(!isdigit(c) && c!='-')c = getchar();if(c == '-')Nig = -1,c = getchar();while(isdigit(c))x = ((x<<1) + (x<<3)) + (c^'0'),c = getchar();return Nig*x;}
ll gcd(ll a,ll b){ return b==0?a:gcd(b,a%b);}
int a[maxn];
int u,k,n;
int flag;
int step=1;
void dfs(int x) {
	if(flag) exit(0);
	if(step>=k&&!flag) {
		printf("%d\n",x);
		flag=1;
		return ;
	}
	for(int i=2;i<=n;i++) {
		if(a[i]==x&&!flag) {
			step++;
			dfs(i);
		}
	}
}
int Xiaobo()
{
	n=read();
	int q;
	q=read();
	a[1]=1;
	for(int i=2;i<=n;i++) {
		a[i]=read();
	}
	while(q--) {
		scanf("%d%d",&u,&k);
		flag=0;
		step=1;
		dfs(u);
		if(!flag) printf("-1\n");
	}
}