当前位置: 代码迷 >> 综合 >> hihoCoder #1041 : 国庆出游 ( 深搜,树的遍历 )
  详细解决方案

hihoCoder #1041 : 国庆出游 ( 深搜,树的遍历 )

热度:91   发布时间:2023-12-27 01:39:12.0

题目地址:http://hihocoder.com/problemset/problem/1041

时间限制:1000ms
单点时限:1000ms
内存限制:256MB

描述
小Hi和小Ho准备国庆期间去A国旅游。A国的城际交通比较有特色:它共有n座城市(编号1-n);城市之间恰好有n-1条公路相连,形成一个树形公路网。小Hi计划从A国首都(1号城市)出发,自驾遍历所有城市,并且经过每一条公路恰好两次——来回各一次——这样公路两旁的景色都不会错过。

令小Hi苦恼的是他的小伙伴小Ho希望能以某种特定的顺序游历其中m个城市。例如按3-2-5的顺序游历这3座城市。(具体来讲是要求:第一次到达3号城市比第一次到达2号城市早,并且第一次到达2号城市比第一次到达5号城市早)。

小Hi想知道是否有一种自驾顺序满足小Ho的要求。

输入
输入第一行是一个整数T(1<=T<=20),代表测试数据的数量。

每组数据第一行是一个整数n(1 <= n <= 100),代表城市数目。

之后n-1行每行两个整数a和b (1 <= a, b <= n),表示ab之间有公路相连。

之后一行包含一个整数m (1 <= m <= n)

最后一行包含m个整数,表示小Ho希望的游历顺序。

输出
YES或者NO,表示是否有一种自驾顺序满足小Ho的要求。

样例输入

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

样例输出

YES
NO

Java 代码如下:

import java.util.HashSet;
import java.util.Scanner;
import java.util.Set;
import java.util.Stack;
public class Main {public static void main(String[] args) {Scanner in=new Scanner(System.in);int t=in.nextInt();while(t-->0){int n=in.nextInt();int[][] map=new int[n+2][n+2];for(int i=0;i<n-1;i++){int a=in.nextInt();int b=in.nextInt();map[a][b]=1;map[b][a]=1;}int m=in.nextInt();int[] order=new int[m+2];int ocount=0;Set<Integer> set=new HashSet<Integer>();for(int i=0;i<m;i++){                   //要求顺序序列order[i]=in.nextInt();set.add(new Integer(order[i]));}int[] vis=new int[n+2];int vcoun=0;Stack<Integer> st=new Stack<Integer>();st.push(new Integer(1));int flag=0;while(!st.isEmpty()){int sttop=st.peek().intValue();if(vis[sttop]==0){                      //未访问过的点if(set.contains(st.peek())){
   //在要求的顺序中if(sttop!=order[ocount]){    //不是下一个顺序点,回溯st.pop();continue;}ocount++;}vis[sttop]=1;vcoun++;if(vcoun==n){               //判断结束System.out.println("YES");flag=1;break;}for(int i=1;i<=n;i++){if(sttop!=i&&map[sttop][i]==1&&vis[i]==0){st.push(new Integer(i));}}}else if(vis[sttop]==1){ //访问过1次的,添加vis[sttop]++;for(int i=1;i<=n;i++){if(sttop!=i&&map[sttop][i]==1&&vis[i]==0){st.push(new Integer(i));}}}else{ //访问2次,直接弹出st.pop();}}if(flag==0)System.out.println("NO");}}}

思路:这是一道深搜的题,用给的游历顺序剪枝。而且n个点一共有n-1条边,是一个树(先开始没有注意这点当成图的遍历了),每条边可以走两回。
用一个分别用数组和HashSet记录所有的m个值,数组是有顺序的,用来判断下一个应该遇到的值,HashSet只是用来判断当前点是否是m个值之一。

很久没写过搜索 了,写的很费劲,而且很不美观,继续努力吧。