本文已参与[新人创作礼]活动,一起开启掘金创作之路.

题目描述

标题：版本分支

小明负责维护公司一个奇怪的项目。这个项目的代码一直在不断分支(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及以上版本的特性。 主类的名字必须是：Main，否则按无效代码处理。

java代码

暴力解，规模较大会超时 基本思路是用p保存每一个项目i的直接父项目，比如p[i]的值就是项目i的直接父结点，通过不断向父结点查找，并判断是否相等即可。

import java.util.Scanner;

public class Main {
	static int[] p; // p[i]的值为项目i的父结点
	static int N;
	static int Q;

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		N = sc.nextInt();
		Q = sc.nextInt();
		p = new int[N + 1]; // 从1开始使用
		p[0] = p[1] = 0;
		int parent, son;
		// 处理输入,N个结点的树只有N-1条树枝
		for (int i = 0; i < N - 1; i++) {
			parent = sc.nextInt();
			son = sc.nextInt();
			p[son] = parent;
		}
		// 对Q次查询处理
		for (int i = 0; i < Q; i++) {
			parent = sc.nextInt();
			son = sc.nextInt();
			if (isZuxian(parent, son)) {
				System.out.println("YES");
			} else {
				System.out.println("NO");
			}
		}
		sc.close();
	}

	// 判断parent是否是son的祖先
	private static boolean isZuxian(int parent, int son) {
		while (true) {
			if (son == 0) {
				return false;
			}
			if (p[son] == parent || parent == son) {
				return true;
			}
			son = p[son];
		}
	}
}