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

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

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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值