问题:详见http://hihocoder.com/problemset/problem/1041
有n个城市,城市编号为1-n,城市间有n-1条路(所以,城市路网是一棵树)。给定一个序列S,要求判断是否存在一条路径,使得
- 从城市 1 开始出发,每条路都需通过2次,最后回到1
- 对于S中的城市,访问的顺序要从前到后。因为每条路可以走两次,所以这里要求的是第一次经过的顺序
思路:DFS
1. 预处理:找出所有结点的可达结点。
通过这个,我们就可以快速判断某个结点是否在某个子树中。比如下一个的访问目标为3,当前的位置是1,那么通过判断3是否在1的某个子树中,可以直接决定下一步应该选择哪个结点。
2. 从结点1开始深搜。直到按序访问到序列S中的每个点
#include <iostream>
#include <bitset>
#include <string> using namespace std;
#define maxCity 100
bitset<maxCity> roads[maxCity];//标识两个城市之间有路
bitset<maxCity> reach[maxCity];//标识一个点的子节点,也就是到哪些点是可达的 void setReach(int curr, int father, int cityCnt){
reach[curr][curr] = ;//自己到自己是可达的
for (int son = ; son <= cityCnt; son++){
if (father == son)
continue;
if (roads[curr][son]){//如果father 和 son 之间是有路的
setReach(son, curr, cityCnt);//计算son的可达点
reach[curr] |= reach[son];//通过 或 操作,把son的可达点加入到curr的可达集合中
}
}
}
//当前访问到的城市 currID
//下一个应该访问在要求序列中的第 rIndex 个城市
//要求的序列包含 rCnt 个城市
int rIndex = ;
bool flag = false;//是否找到满足条件的路径
void dfs(int currID, int father, int rCnt, int* rCity, int cityCnt){
if (currID == rCity[rIndex])//当前要访问的正好是序列中的下一个城市
++rIndex;
if (rIndex == rCnt){//序列的要求已经满足
flag = true;
return;
}
while (rIndex < rCnt){//判断从当前点继续是否能走完整个要求的序列
int i = ;
for (; i <= cityCnt; i++){
if (i == currID || i == father)
continue;
//遍历currID的所有子节点,每个节点代表了一棵子树
//roads[currID][i] 不为0, 表明currID和i有路
//reach[i][rCity[rIndex]] 不为0,表明从i可以直到要求序列的下一个城市
//同时满足条件,也就是说currID的下一个可以访问i,然后通过 i 走到下一个要求的城市
if (roads[currID][i] && reach[i][rCity[rIndex]]){
roads[currID][i] = ;
dfs(i, currID, rCnt, rCity, cityCnt);//访问 i 这棵子树
break;//每个节点只能属于一棵子树,所以其它的就不用再判断。已经知道下一个要求的节点在 i 的子树中,所以不需要再去判断其它节点的子树
}
}
if (i > cityCnt)//没有在currID中的任何一个子树中找到下一个要求的结点
break;
}
}
int main(){
int cnt, cityCnt, rCnt, *rCity;
cin >> cnt; while (cnt-- > ){
flag = false;
cin >> cityCnt;
rIndex = ;
//初始化,城市间都没有路,都没有子节点
for (int i = ; i <= cityCnt; i++){
roads[i].reset();
reach[i].reset();
}
for (int i = ; i < cityCnt - ; i++){
int a, b;
//a, b间有路
cin >> a >> b;
roads[a][b] = ;
roads[b][a] = ;
}
cin >> rCnt;
rCity = new int[rCnt];
for (int i = ; i < rCnt; i++)
cin >> rCity[i];
setReach(, -, cityCnt);
dfs(, -, rCnt, rCity, cityCnt);
if (flag)
cout << "YES" << endl;
else
cout << "NO" << endl;
}
return ;
}
注:
1. bitset 需要#include <bitset>
bitset<maxCity> roads[maxCity] 声明了一个数组,数组的元素类型为bitset<maxCity>,每一个数用maxCity位的二进制来表示
2. 在计算每个点可达的集合,以及寻找路径的时候都有一个判断条件:判断当前点的son是否和父节点father一样
如果不进行这个判断,假设树的结构为 1-2,也就是只有两个城市1、2,它们中间有一条路。以setReach为例,递归的过程如下:
setReach(1, -1)->(因为1可以到达2)setReach(2, 1)->(从城市1开始判断,2可以到达1)setReach(1, 2)->setReach(2,1)...死循环
这里只需要考虑当前点的父结点,而不用继续向上考虑,是因为当前点和父结点的上层并不会有路径。