版本分支(蓝桥杯)

小明负责维护公司一个奇怪的项目。这个项目的代码一直在不断分支(branch)但是从未发生过合并(merge)。
现在这个项目的代码一共有N个版本,编号1~N,其中1号版本是最初的版本。
除了1号版本之外,其他版本的代码都恰好有一个直接的父版本;即这N个版本形成了一棵以1为根的树形结构。  

如下图就是一个可能的版本树:

    1
   / \
  2   3
  |  / \
  5 4   6

现在小明需要经常检查版本x是不是版本y的祖先版本。你能帮助小明吗?

输入
----
第一行包含两个整数N和Q,代表版本总数和查询总数。  
以下N-1行,每行包含2个整数u和v,代表版本u是版本v的直接父版本。  
再之后Q行,每行包含2个整数x和y,代表询问版本x是不是版本y的祖先版本。  

对于30%的数据,1 <= N <= 1000  1 <= Q <= 1000  
对于100%的数据,1 <= N <= 100000  1 <= Q <= 100000  

输出
----
对于每个询问,输出YES或NO代表x是否是y的祖先。  

【样例输入】
6 5
1 2
1 3
2 5
3 6
3 4
1 1
1 4
2 6
5 2
6 4

【样例输出】
YES
YES
NO
NO
NO

资源约定:
峰值内存消耗(含虚拟机) < 256M
CPU消耗  < 1000ms


请严格按要求输出,不要画蛇添足地打印类似:“请您输入...” 的多余内容。

所有代码放在同一个源文件中,调试通过后,拷贝提交该源码。

不要使用package语句。不要使用jdk1.7及以上版本的特性。(Java)
主类的名字必须是:Main,否则按无效代码处理。


Java代码:

import java.util.*;
public class Main {
	static Map<Integer,Integer> map=new HashMap<Integer,Integer>();
	
	static boolean dfs(int x,int y){//x是不是y的祖先
		if(x==y)
			return true;
		else if(map.containsKey(y)){
			int father=map.get(y);
			if(father==x)
				return true;
			else
				return dfs(x,father);
		}
		else
			return false;
	}
	
    public static void main(String[] args) {
        Scanner in=new Scanner(System.in);
        int n=in.nextInt(),q=in.nextInt();
        for(int i=1;i<n;i++){
        	int u=in.nextInt(),v=in.nextInt();//u是v的直接父
        	map.put(v,u);
        }
        for(int i=0;i<q;i++){
        	int x=in.nextInt(),y=in.nextInt();//x是不是y的祖先
        	if(dfs(x,y))
        		System.out.println("YES");
        	else
        		System.out.println("NO");
        }
    }
}

Python代码:

n,q=map(int,input().split())  # 版本总数,查询总数
d={}  # 字典:value是key的直接父版本
def dfs(x,y):  # 询问x是不是y的祖先版本
    if x==y:
        return True
    while y in d:
        if d[y]==x:
            return True
        else:
            y=d[y]
    return False

for i in range(n-1):
    u,v=map(int,input().split())  # u是v的直接父版本
    d[v]=u
for i in range(q):
    x,y=map(int,input().split())  # 询问x是不是y的祖先版本
    if dfs(x,y):
        print("YES")
    else:
        print("NO")

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值