hiboCoder 1041 国庆出游 dfs+思维

先抽象出一棵以1做为根结点的树。给定了访问序列a[1..m]。

考虑两种特殊情况:

1.访问了某个a[j],但是存在a[i]没有访问且i < j,出现这种情况说明a[j]一定是a[i]的祖先节点,必须访问过a[j]才能访问a[i],这种情况是不符合条件的。

2.当某个点被访问后并且这个节点被回溯之后,它的子树中仍然存在序列中的节点没被访问,那么些节点不会有机会再被访问,不符合情况。

思路:根据访问序列中节点的下标,给定所有待访问节点一个权值,代表他们的访问顺序,例如数据:

7
1 2
1 3
2 4
2 5
3 6
3 7
3
3 2 7

这组数据中,分别给3 2 7 一个权值,w[3] = 0, w[2] = 1, w[7] = 2, 其他节点的权值都是inf,dp[i]表示以节点i作为根结点的子树中最小的权值,如果某个节点i下面存在权值比它小的节点j,说明满足第一种情况,肯定是不合理的。

当处理了第一种情况,并且得到了所有dp[i],那么现在要开始处理第二种情况。模拟深度优先遍历这棵树即可,在遍历的同时,一定是按照给定的访问序列来顺序访问,给一个全局变量f,代表当前需要访问的节点的权值,假设当前在i节点处,它的子节点是Vj,那么一定是优先访问dp[Vj] = f的节点,然后继续找,知道有节点u满足w[u] = f,说明找到了这个节点,让f++,该寻找下一个节点了,一直重复即可,注意每个节点只访问一次,当出现要访问某个被访问过的节点时,就说明符合第二种情况。其实这种方法也可以处理第一种情况,由于要预处理dp[i],所以我在建树的同时就把第一种情况处理了。最后判断序列中是否有没被访问过得节点,有则说明是满足第二种情况的。总的复杂度是O(n*logn)

贴一组我认为很容易WA的测试数据:

6

4

1 2

2 3

3 4

3

4 3 2

7

1 2

1 3

2 4

2 5

3 6

3 7

3

3 7 2

7

1 2

1 3

2 4

2 5

3 6

3 7

3

3 2 7

4

1 2

2 3

3 4

4

4 3 2 1

7

1 2

1 3

2 4

2 5

3 6

3 7

4

2 4 3 5

9

1 2

1 3

2 4

2 5

3 6

3 7

4 8

4 9

5

2 4 8 5 9

正确答案是:

NO

YES

NO

NO

NO

NO

AC代码:

#include <cstdio>
#include <cmath>
#include <cctype>
#include <bitset>
#include <algorithm>
#include <cstring>
#include <utility>
#include <string>
#include <iostream>
#include <map>
#include <set>
#include <vector>
#include <queue>
#include <stack>
using namespace std;
#pragma comment(linker, "/STACK:1024000000,1024000000")
#define eps 1e-10
#define inf 0x3f3f3f3f
#define PI pair<int, int>
typedef long long LL;
const int maxn = 100 + 5;
vector<int>G[maxn];
int vis[maxn], w[maxn], a[maxn], dp[maxn];
int n, m;

bool build_tree(int u, int par) {
	if(G[u].size() == 1 && G[u][0] == par) {
		dp[u] = w[u];
		return true;
	}
	int ans = inf;
	for(int i = 0; i < G[u].size(); ++i) {
		int v = G[u][i];
		if(v == par) continue;
		if(!build_tree(v, u)) return false;
		if(w[u] != inf && w[u] > dp[v]) return false;
		ans = min(ans, dp[v]);
	}
	dp[u] = min(ans, w[u]);
	return true;
}

int f;
void dfs(int u,int par) {
	vis[u]++;
	if(w[u] == f) f++;
	if(f >= m) return;
	for(int i = 0; i < G[u].size(); ++i) {
		int v = G[u][i];
		if(v == par) continue;
		if(!vis[v] && dp[v] == f) { //未访问过f能在以v为根结点的子树中找到
			dfs(v, u);
		}
	}
}

bool cmp(int a, int b) {
	return dp[a] < dp[b];
}

bool solve(int n, int m) {
	//以第一个点为根建树
	bool ok = build_tree(1, -1);
	if(!ok) return false;
	for(int i = 1; i <= n; ++i) {
		//排序,根据节点的权值确定访问顺序
		sort(G[i].begin(), G[i].end(), cmp);
	}
	f = 0;
	dfs(1, -1);
	for(int i = 0; i < m; ++i) {
		if(!vis[a[i]]) return false;
 	}
 	return true;
}
int main() {
	int T;
	scanf("%d", &T);
	while(T--) {
		scanf("%d", &n);
		for(int i = 1; i <= n+1; ++i) G[i].clear();
		int u, v;
		for(int i = 0; i < n-1; ++i) {
			scanf("%d%d", &u, &v);
			G[u].push_back(v);
			G[v].push_back(u);
		}
		memset(vis, 0, sizeof(vis));
		for(int i = 1; i <= n; ++i) w[i] = inf;
		scanf("%d", &m);
		//赋予权值
		for(int i = 0; i < m; ++i) {
			scanf("%d", &a[i]);
			w[a[i]] = i;
		}
		int ok = solve(n, m);
		if(ok) printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}

如有不当之处欢迎指出!

上一篇:Java基础知识陷阱(九)


下一篇:【Python 函数对象 命名空间与作用域 闭包函数 装饰器 迭代器 内置函数】