codeforces 1037D 模拟

链接： http://codeforces.com/problemset/problem/1037/D

思路： 在vector 中二分找就可以了。

模拟记录每个节点的直接孩纸。

代码：

#include<stdio.h>
#include<string.h>
#include<vector>
#include<queue>
#include<iostream>
#include<algorithm>

using namespace std;
typedef long long ll;
const int N =2e5+5;

vector<int >ve[N];
int a[N];
int n;
int inde[N];
int vis[N];
int sz[N];

void dfs(int u,int fa)
{
	sz[u]=0;
	for(int i=0;i<ve[u].size();i++){
		int v=ve[u][i];
		if(v==fa) continue;
		dfs(v,u);
		sz[u]++;
	}
	return ;
}

int main()
{
	cin>>n;
	int u,v;
	for(int i=1;i<n;i++){
		scanf("%d %d",&u,&v);
		ve[u].push_back(v);
		ve[v].push_back(u);
		inde[v]++;
		inde[u]++;
	}
	
	for(int i=1;i<=n;i++){
		sort(ve[i].begin(),ve[i].end());
	}
	
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	if(a[1]!=1){
		printf("No\n");
		return 0;
	}
	
	dfs(1,-1);
	
	queue< int >q;
	q.push(1);
	int f=0;
	vector<int >::iterator it;
	int p=2;
	while(!q.empty()){
		u=q.front(); q.pop();
		//cout<<"uuu "<<u<<endl;
		
		int aim=sz[u];
		int cnt=0;
		//cout<<"aim "<<aim<<endl;
		while(cnt<aim){
			
			it=lower_bound(ve[u].begin(),ve[u].end(),a[p]);
			if(it==ve[u].end()||*it!=a[p]){ // 没有加 后半句wa 到心碎 
				f=1; break;
			}
			cnt++;
			q.push(a[p++]);
			if(p>n) break;
		}
		if(p>n) break;
		if(f) break;
	}
	if(f) printf("No\n");
	else printf("Yes\n");
	
	return 0;
}