傳送門:hihocoder 1041

給定一棵樹,從節點1出發。

如今要求一種遍立方法,使得全部邊都來回各走一遍,問在這樣的條件下能否按指定順序訪問部分點

比較巧妙的dfs深搜。

默認1為根節點,因此從1節點出發,預處理出每一個節點的子樹中能達到的節點。

然後進行深搜,深搜策略例如以下:

對根節點開始,若找到某一個子節點能到達目標節點,且走向該節點的路徑沒有訪問過,則走向子節點繼續深搜直至按指定順序訪問全然部節點,或者某一節點已不可達

/******************************************************
 * File Name:   1041.cpp
 * Author:      kojimai
 * Create Time: 2014年10月31日 星期五 12時43分10秒
******************************************************/

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iostream>
#include<bitset>
using namespace std;
#define FFF 105
int a[FFF],first[FFF],e,cnt = 0,num,next[FFF*2],v[FFF*2];
bitset<FFF> shao[FFF];//用bitset來存每一個節點的子節點中能到達點的情況
bool vis[FFF*2],ans;
void addedge(int x,int y)
{
	v[e] = y;next[e] = first[x];first[x] = e++;
	v[e] = x;next[e] = first[y];first[y] = e++;
}

void solve_can(int now,int fa)//處理出now節點的子樹中能到達的點
{
	shao[now][now] = 1;//首先自身能到達自身位置
	for(int k = first[now]; ~k;k = next[k])
	{
		int val = v[k];
		if(val == fa)
			continue;
		solve_can(val,now);//找出子節點的子樹中能到達的點
		shao[now] |= shao[val];//子節點能到達的點,當前點也能到,用bitset處理能夠直接用或運算比較快捷
	}
	return;
}

void solve(int now,int fa)
{
	if(now == a[cnt])//訪問到指定順序的點,則準備訪問下一個點
	{
		cnt++;
	}
	if(cnt == num)
	{
		ans = true;//全部點都訪問完了,則說明最後可行
		return;
	}
	while(cnt < num)
	{
		int tmp = cnt;
		for(int k = first[now]; ~k;k = next[k])
		{
			int val = v[k];
			if(val == fa)
				continue;
			if(shao[val][a[cnt]] && !vis[k])//目標點可達且該條路沒有訪問過,則向下深搜
			{
				vis[k] = true;//標記路徑已經訪問過
				solve(val,now);
				break;
			}
		}
		if(tmp == cnt)//假如當前節點無法到達指定順序的點或者路徑已訪問過不能走則返回父節點繼續深搜
			break;
	}
	return;
}

int main()
{
	int keng;
	scanf("%d",&keng);
	while(keng--)
	{
		ans = false;
		int x,y;
		memset(first,-1,sizeof(first)); e = 0;
		int n;
		scanf("%d",&n);
		for(int i = 1;i < n;i++)
		{
			scanf("%d%d",&x,&y);
			addedge(x,y);
		}

		scanf("%d",&num);
		for(int i = 0;i < num;i++)
			scanf("%d",&a[i]);

		for(int i = 1;i <= n;i++)
			shao[i].reset();

		solve_can(1,-1);//預處理每一個點的子樹能到達哪些點

		memset(vis,false,sizeof(vis));
		cnt = 0;//記錄按順序已經訪問到第幾個點
		solve(1,-1);
		if(ans)
			printf("YES\n");
		else
			printf("NO\n");
	}
	return 0;
}

hihocoder 1041 國慶出遊 dfs