题目解析
给定几堆部落群
先输出人数以及部落群数
然后在判断两个人是否在同一个部落中
解题思路
可以先用并查集 统计出每个人所在的部落群
- 为啥想到并查集呢？因为他后面需要判断两个人是否为同一部落中，以及部落群数
然后输出想要的答案即可
这题java我基本都是t 欸 过不去
代码
import java.io.*;
import java.util.*;
public class Main
{
static int N = (int) 1e4;
static int fa[] = new int[N + 10];
static int find(int x)
{
if (fa[x] != x)
fa[x] = find(fa[x]);
return fa[x];
}
public static void main(String[] args) throws IOException
{
for (int i = 1; i <= N; i++)
fa[i] = i;
TreeSet<Integer> tr = new TreeSet<Integer>(); // 存储共有多少个人
int n = sc.nextInt();
for (int i = 1; i <= n; i++)
{
int k = sc.nextInt(); // k个人
int a = sc.nextInt(); // 第一个人
int x = find(a); // 第一个人的祖宗节点
tr.add(a);
while (--k > 0)
{
int b = sc.nextInt(), y = find(b); // 之后的人与祖宗节点
fa[y] = x; // 之后的人的祖宗节点认祖归宗
tr.add(b);
}
}
int cnt = 0; // 存储共有多少个圈子
for (int i : tr)
{
if (fa[i] == i)
cnt++;
}
out.println(tr.size() + " " + cnt);
int q = sc.nextInt();
while (q-- > 0)
{
int a = sc.nextInt(), b = sc.nextInt();
int x = find(a), y = find(b);
out.println(x == y ? "Y" : "N");
}
out.flush();
out.close();
}
static Scanner sc = new Scanner(System.in);
static PrintWriter out = new PrintWriter(System.out);
}
c++
#include <iostream>
#include <unordered_set>
using namespace std;
unordered_set<int> se;
const int N = 1e4;
int fa[N + 10];
int find(int x)
{
if(fa[x] != x)
fa[x] = find(fa[x]);
return fa[x];
}
int main()
{
for(int i = 1; i <= N; i ++)
fa[i] = i;
int n; scanf("%d", &n);
while (n -- > 0)
{
int k; scanf("%d", &k);
int a; scanf("%d", &a);
int x = find(a);
se.insert(a);
while (-- k > 0)
{
int b; scanf("%d", &b);
int y = find(b);
fa[y] = x;
se.insert(b);
}
}
int cnt = 0;
for (int i : se)
{
if(fa[i] == i)
cnt ++;
}
printf("%d %d\n", se.size(), cnt);
int q; scanf("%d", &q);
while (q -- > 0)
{
int a, b; scanf("%d%d", &a, &b);
int x = find(a), y = find(b);
printf("%s\n", x == y ? "Y" : "N");
}
return 0;
}