题目
给定一棵树,N个节点,N - 1条边。给定m个节点,能否找出一种遍历方法,使得首次到达节点ai的时间小于首次到达节点aj的时间(i < j)。且经过的路径上的每条边都最多走两遍
分析
我的想法:
深度优先搜索的策略,在进入某个节点A时,以该节点A为根的子树中的所有节点构成一个集合,该集合内的点将在 m个点集合中连在一块,
而在节点A为根的子树之外的点不能混在其中。
在实现的时候,定义全局数组 gBlocked[],gBlocked[i] 表示节点i将不能再访问(需要判断并标记), gVisited[i] 表示节点i在当前已经被访问过(
在之前访问其他节点的路径上被经过)。
从头到尾遍历m个点,对于点Mi:
1)如果该节点在之前被访问过,则失败!
2)找出从Mi向上到达根的路径 path_new,并将路径上的点都标记为 访问过;
在从节点Mi出发寻找Mi到根节点的路径的时候,如果中间某个节点的状态为“不能在访问”,则失败!
3)然后,和前一个点Mi-1 形成的路径 path_old 进行比较,找出 Mi 和 Mi-1的最低的公共祖先节点 P. 然后将 从P到Mi-1的路径中,P之后的那个节点t标记
为 “不能再访问” 状态(gBlocked[t] = true),因为此时已经从节点t的子树中出来了。
遍历完所有m个节点之后,返回成功!
参考网上的想法:
1.预处理,确定每个节点出发可以到达的所有节点。(通过父节点可以到达子节点到达的所有节点进行递归)
2.从根节点开始按照顺序访问m个点,在访问点 Mi 的时候,若当前点为P, 则P的子节点中寻找能够到达 Mi
的节点 Q,然后递归到Q,并将Q标记为“访问过”, 从P找不到可以访问到Mi的子节点,则回溯到P的父节点T, 从T
再递归
直到访问完所有的节点,或者中间的某个节点递归到根节点仍然无法访问
实现
解法1
/*
深度优先搜索的策略,在进入某个节点A时,以该节点A为根的子树中的所有节点构成一个集合,该集合内的点将在 m个点集合中连在一块,
而在节点A为根的子树之外的点不能混在其中。
在实现的时候,定义全局数组 gBlocked[],gBlocked[i] 表示节点i将不能再访问(需要判断并标记), gVisited[i] 表示节点i在当前已经被访问过(
在之前访问其他节点的路径上被经过)。
从头到尾遍历m个点,对于点Mi:
1)如果该节点在之前被访问过,则失败!
2)找出从Mi向上到达根的路径 path_new,并将路径上的点都标记为 访问过;
在从节点Mi出发寻找Mi到根节点的路径的时候,如果中间某个节点的状态为“不能在访问”,则失败!
3)然后,和前一个点Mi-1 形成的路径 path_old 进行比较,找出 Mi 和 Mi-1的最低的公共祖先节点 P. 然后将 从P到Mi-1的路径中,P之后的那个节点t标记
为 “不能再访问” 状态(gBlocked[t] = true),因为此时已经从节点t的子树中出来了。
遍历完所有m个节点之后,返回成功!
*/
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<deque>
using namespace std;
#define N 120
struct Edge{
int to;
int next;
Edge(int t = -1, int n = -1) :to(t), next(n){};
};
Edge gEdges[N];
int gEdgeIndex;
int gHead[N];
bool gVisited[N];
int gPre[N];
int gTravelSeq[N];
bool gBlocked[N];
void InsertEdge(int u, int v){
int e = gEdgeIndex++;
gEdges[e].to = v;
gEdges[e].next = gHead[u];
gHead[u] = e;
e = gEdgeIndex++;
gEdges[e].to = u;
gEdges[e].next = gHead[v];
gHead[v] = e;
}
//Dfs确定树的结构,因为一开始给的点对不一定是 父--子 ,所以需要遍历图来进行确定树的父子关系
void Dfs(int u){
gVisited[u] = true;
for (int e = gHead[u]; e != -1; e = gEdges[e].next){
int v = gEdges[e].to;
if (!gVisited[v]){
gPre[v] = u;
Dfs(v);
}
}
}
//寻找从节点u向上到根节点的路径
bool Path(int u, deque<int>& path){
path.clear();
while (u != -1){
if (gBlocked[u]){
return false;
}
gVisited[u] = true;
path.push_back(u);
u = gPre[u];
}
return true;
}
//判断是否可以按照顺序遍历这m个节点,n为节点的数目
bool CanTravel(int m, int n){
if (m <= 0)
return false;
if (gTravelSeq[0] > n)
return false;
memset(gVisited, false, sizeof(gVisited));
deque<int> last_path; //保存路径
bool can_path = Path(gTravelSeq[0], last_path);
deque<int> path;
for (int i = 1; i < m; i++){
int u = gTravelSeq[i];
if (u > n || gVisited[u]) //节点u在访问之前的节点过程中被访问该,返回失败
return false;
can_path = Path(gTravelSeq[i], path);
if (!can_path) //寻找从节点u到根节点的路径时,遇到了 “不能再访问”的节点,返回失败
return false;
int k1 = last_path.size() - 1;
int k2 = path.size() - 1;
//寻找最近公共祖先
while (k1 >= 0 && k2 >= 0 && last_path[k1] == path[k2])
{
--k1;
--k2;
}
if (k1 >= 0) //最近公共祖先到 gTravelSeq[i-1]的的下一个点,标记为不能再访问
gBlocked[last_path[k1]] = true;
last_path = path;
}
return true;
}
void Init(){
memset(gVisited, false, sizeof(gVisited));
memset(gPre, -1, sizeof(gPre));
memset(gEdges, -1, sizeof(gEdges));
memset(gHead, -1, sizeof(gHead));
memset(gBlocked, false, sizeof(gBlocked));
gEdgeIndex = 0;
}
int main2(){
int T, n, m, u, v;
scanf("%d", &T);
while (T--){
Init();
scanf("%d", &n);
for (int i = 0; i < n - 1; i++){
scanf("%d %d", &u, &v);
InsertEdge(u, v);
}
Dfs(1);
scanf("%d", &m);
for (int i = 0; i < m; i++){
scanf("%d", &(gTravelSeq[i]));
}
bool ret = CanTravel(m, n);
if (ret)
printf("YES\n");
else
printf("NO\n");
}
return 0;
}
解法2
/*
参考网上的做法:
1.预处理,确定每个节点出发可以到达的所有节点。(通过父节点可以到达子节点到达的所有节点进行递归)
2.从根节点开始按照顺序访问m个点,在访问点 Mi 的时候,若当前点为P, 则P的子节点中寻找能够到达 Mi
的节点 Q,然后递归到Q,并将Q标记为“访问过”, 从P找不到可以访问到Mi的子节点,则回溯到P的父节点T, 从T
再递归
直到访问完所有的节点,或者中间的某个节点递归到根节点仍然无法访问
*/
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<deque>
#include<bitset>
using namespace std;
#define N 120
struct Edge{
int to;
int next;
Edge(int t = -1, int n = -1) :to(t), next(n){};
};
Edge gEdges[N];
int gEdgeIndex;
int gHead[N];
bool gVisited[N];
int gPre[N];
int gTravelSeq[N];
bitset<N> gCanReach[N]; //判断节点是否可以到达其他节点,用bitset可以方便的进行位操作
void InsertEdge(int u, int v){
int e = gEdgeIndex++;
gEdges[e].to = v;
gEdges[e].next = gHead[u];
gHead[u] = e;
e = gEdgeIndex++;
gEdges[e].to = u;
gEdges[e].next = gHead[v];
gHead[v] = e;
}
//确定从u可以到达的所有节点
void CanReach(int u){
gCanReach[u][u] = 1;
gVisited[u] = true;
for (int e = gHead[u]; e != -1; e = gEdges[e].next){
int v = gEdges[e].to;
if (!gVisited[v]){ //若为true,则说明v为u的父节点
gPre[v] = u;
CanReach(v);
gCanReach[u] |= gCanReach[v];
}
}
}
//当前节点为u,需要 按照顺序遍历的索引为 index, m为需要访问的节点数目
bool CanTravel(int u, int index, int m){
if (index == m) //所有节点都访问过了
return true;
if (u == -1) //最后无法继续向下访问
return false;
gVisited[u] = true;
if (u == gTravelSeq[index]){ //访问到一个节点,从当前节点开始继续下一个
return CanTravel(u, index + 1, m);
}
for (int e = gHead[u]; e != -1; e = gEdges[e].next){
int v = gEdges[e].to;
//从当前节点的子节点中查找之前没有经过的,且能够到达目标节点的
if (!gVisited[v] && gCanReach[v][gTravelSeq[index]]){
return CanTravel(v, index, m);
}
}
//回溯到上一层节点
return CanTravel(gPre[u], index, m);
}
void Init(){
memset(gVisited, false, sizeof(gVisited));
memset(gEdges, -1, sizeof(gEdges));
memset(gHead, -1, sizeof(gHead));
memset(gPre, -1, sizeof(gPre));
for (int i = 0; i < N; i++){
gCanReach[i].reset(0);
}
gEdgeIndex = 0;
}
int main(){
int T, n, m, u, v;
scanf("%d", &T);
while (T--){
Init();
scanf("%d", &n);
for (int i = 0; i < n - 1; i++){
scanf("%d %d", &u, &v);
InsertEdge(u, v);
}
CanReach(1);
scanf("%d", &m);
for (int i = 0; i < m; i++){
scanf("%d", &(gTravelSeq[i]));
}
memset(gVisited, false, sizeof(gVisited));
bool ret = CanTravel(1, 0, m);
if (ret)
printf("YES\n");
else
printf("NO\n");
}
return 0;
}