洛谷 P2256 一中校运会之百米跑 (Java)

洛谷 P2256 一中校运会之百米跑 (Java)

传送门：P2256 一中校运会之百米跑

题目： 一中校运会之百米跑

题目背景

在一大堆秀恩爱的 ** 之中，来不及秀恩爱的苏大学神踏着坚定（？）的步伐走向了 $100$ 米跑的起点。这时苏大学神发现，百米赛跑的参赛同学实在是太多了，连体育老师也忙不过来。这时体育老师发现了身为体育委员的苏大学神，便来找他帮忙。

可是苏大学神需要热身，不然跑到一半就会抽（筋）、于是他就找到了你。。。如果你帮助体育老师解决了问题，老师就会给你 $5$ 个积分。

题目描述

假设一共有 $N$（$2\le N\le 2\times 10^4$）个参赛选手。（尼玛全校学生都没这么多吧）

老师会告诉你这 $N$ 个选手的名字。

接着会告诉你 $M$（$1\le M\le 10^6$）句话，即告诉你学生 A 与学生 B 在同一个组里。

如果学生 A 与学生 B 在同一组里，学生 B 与学生 C 也在同一组里，就说明学生 A 与学生 C 在同一组。

然后老师会问你 $K$（$1\le K\le 10^6$）句话，即学生 X 和学生 Y 是否在同一组里。

若是则输出 Yes.，否则输出 No.。

输入格式

第一行输入 $N$ 和 $M$。

接下来 $N$ 行输入每一个同学的名字。

再往下 $M$ 行每行输入两个名字，且保证这两个名字都在上面的 $N$ 行中出现过，表示这两个参赛选手在同一个组里。

再来输入 $K$。

接下来输入 $K$ 个体育老师的询问。

输出格式

对于每一个体育老师的询问，输出 Yes. 或 No.。

样例 #1

样例输入 #1

10 6
Jack
Mike
ASDA
Michel
brabrabra
HeHe
HeHE
papapa
HeY
Obama
Jack Obama
HeHe HeHE
brabrabra HeHe
Obama ASDA
papapa Obama
Obama HeHE
3
Mike Obama
HeHE Jack
papapa brabrabra

样例输出 #1

No.
Yes. 
Yes.

分析：

实现了一个并查集的数据结构，并对给定的一系列操作进行处理。

定义全局变量：
N：指定一个足够大的常量，用于表示选手数量的上限。
map：HashMap类型的变量，用于将选手的名字映射到编号。
p：整型数组，用于存储每个选手的祖先编号。
n、m、k：分别表示选手数量、组数、查询次数。

union() 方法：合并两个集合。它通过查找两个选手的祖先，并将其中一个选手的祖先设为另一个选手的祖先，从而合并两个集合。

find() 方法：找到选手所在集合的祖先。它通过递归地查找选手的祖先，直到找到祖先节点。

sear() 方法：判断两个选手是否在同一集合中。它通过比较两个选手的祖先是否相同来判断是否在同一集合中。

主函数 main()：
从输入中获取选手数量 n 和组数 m。

n：使用循环将选手名字与编号建立映射关系，并初始化并查集。

m：循环读入每组选手的信息，并通过 union() 方法将同一组的选手合并到一个集合中。

k：循环处理每次查询，通过 sear() 方法判断两个选手是否在同一集合中，并输出结果。

复杂度分析：
建立映射关系的时间复杂度为 O(n)。
处理每组选手信息的时间复杂度为 O(m)。
处理每次查询的时间复杂度为 O(k)。
整体时间复杂度为 O(n + m + k)。
空间复杂度为 O(n)。

代码：

import java.util.HashMap;
import java.util.Scanner;

public class Main {
	static final int N = 10010;
	// map集合存储 名字即编号
	static HashMap<String,Integer> map = new HashMap<>();  
	// p数组 存储每个点的祖宗编号
	static int [] p = new int [N];
	static int n,m,k;
	
	// 合并
	public static void union(int x,int y) {
		int rootx = find(x);
		int rooty = find(y);
		if(rootx!=rooty) p[rootx] = rooty;
	}
	
	// 找到x的祖宗节点
	public static int find(int x) {
		if(p[x]!=x) p[x] = find(p[x]);
		return p[x];
	}
	
	// 查询
	public static boolean sear(int x,int y) {
		int rootx = find(x);
		int rooty = find(y);
		return rootx==rooty;
	}
	
    public static void main(String[] args) {
    	Scanner sc = new Scanner(System.in);
    	n = sc.nextInt();m = sc.nextInt();
    	for(int i = 1;i <= n;i++) {
    		// 将第i个名字对应的选手编号设为i
    		map.put(sc.next(),i);
    		p[i] = i;
    	}
    	
    	String s1,s2;
    	while(m-->0) {
    		s1 = sc.next();s2 = sc.next();
    		int x = map.get(s1);int y = map.get(s2);
    		union(x,y);
    	}
    	
    	k = sc.nextInt();
    	while(k-->0) {
    		s1 = sc.next();s2 = sc.next();
    		// 特判两个要查找的名字，可能不存在
    		if(map.containsKey(s1)&&map.containsKey(s2)) {
    			int x = map.get(s1);int y = map.get(s2);
        		if(sear(x,y)) System.out.println("Yes.");
        		else System.out.println("No.");
    		}else System.out.println("No.");
    		
    	}
    }
}