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

来源:互联网 发布:python经典书籍 编辑:程序博客网 时间:2024/05/01 22:45

题目地址: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个值之一。

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

原创粉丝点击