【并查集】

并查集

[NOIP2010 提高组] 关押罪犯

题目背景

NOIP2010 提高组 T3

题目描述

S 城现有两座监狱，一共关押着 $N$ 名罪犯，编号分别为 $1\sim N$。他们之间的关系自然也极不和谐。很多罪犯之间甚至积怨已久，如果客观条件具备则随时可能爆发冲突。我们用“怨气值”（一个正整数值）来表示某两名罪犯之间的仇恨程度，怨气值越大，则这两名罪犯之间的积怨越多。如果两名怨气值为 $c$ 的罪犯被关押在同一监狱，他们俩之间会发生摩擦，并造成影响力为 $c$ 的冲突事件。

每年年末，警察局会将本年内监狱中的所有冲突事件按影响力从大到小排成一个列表，然后上报到 S 城 Z 市长那里。公务繁忙的 Z 市长只会去看列表中的第一个事件的影响力，如果影响很坏，他就会考虑撤换警察局长。

在详细考察了 $N$ 名罪犯间的矛盾关系后，警察局长觉得压力巨大。他准备将罪犯们在两座监狱内重新分配，以求产生的冲突事件影响力都较小，从而保住自己的乌纱帽。假设只要处于同一监狱内的某两个罪犯间有仇恨，那么他们一定会在每年的某个时候发生摩擦。

那么，应如何分配罪犯，才能使 Z 市长看到的那个冲突事件的影响力最小？这个最小值是多少？

输入格式

每行中两个数之间用一个空格隔开。第一行为两个正整数 $N,M$，分别表示罪犯的数目以及存在仇恨的罪犯对数。接下来的 $M$ 行每行为三个正整数 $a_j,b_j,c_j$，表示 $a_j$ 号和 $b_j$ 号罪犯之间存在仇恨，其怨气值为 $c_j$。数据保证 $1<a_j\le b_j\le N,0<c_j\le 10^9$，且每对罪犯组合只出现一次。

输出格式

共一行，为 Z 市长看到的那个冲突事件的影响力。如果本年内监狱中未发生任何冲突事件，请输出 0。

样例 #1

样例输入 #1

4 6
1 4 2534
2 3 3512
1 2 28351
1 3 6618
2 4 1805
3 4 12884

样例输出 #1

3512

提示

输入输出样例说明

罪犯之间的怨气值如下面左图所示，右图所示为罪犯的分配方法，市长看到的冲突事件影响力是 $3512$（由 $2$ 号和 $3$ 号罪犯引发）。其他任何分法都不会比这个分法更优。

数据范围

对于 $30\%$ 的数据有 $N\le 15$。

对于 $70\%$ 的数据有 $N\le 2000,M\le 50000$。

对于 $100\%$ 的数据有 $N\le 20000,M\le 100000$。

# 代码
N,M = map(int,input().split())
crime = []
for _ in range(M):
    crime.append(list(map(int,input().split())))
crime = sorted(crime,key=lambda x:x[-1],reverse=True) # 这里排序用了贪心的思想，优先保证排影响最大的犯人不在同一个监狱

fa = [i for i in range(2*N+1)] # 一倍存自己 二倍存的是敌人
def find(x):
    if fa[x]!=x:
        fa[x] = find(fa[x])
    return fa[x]
ans = 0
for cr in crime:
    a,b,c = cr[0],cr[1],cr[2]
    if find(a)==find(b): # 说明这两个在一个集合里面
        ans = c
        break
    fa[find(a)] = find(b+N) # 把a存放到b的敌人里面
    fa[find(b)] = find(a+N) # 把b存放到a的敌人里面
print(ans)

[NOI2001] 食物链

题目描述

动物王国中有三类动物 $A,B,C$，这三类动物的食物链构成了有趣的环形。$A$ 吃 $B$，$B$ 吃 $C$，$C$ 吃 $A$。

现有 $N$ 个动物，以 $1\sim N$ 编号。每个动物都是 $A,B,C$ 中的一种，但是我们并不知道它到底是哪一种。

有人用两种说法对这 $N$ 个动物所构成的食物链关系进行描述：

• 第一种说法是 1 X Y，表示 $X$ 和 $Y$ 是同类。
• 第二种说法是2 X Y，表示 $X$ 吃 $Y$。

此人对 $N$ 个动物，用上述两种说法，一句接一句地说出 $K$ 句话，这 $K$ 句话有的是真的，有的是假的。当一句话满足下列三条之一时，这句话就是假话，否则就是真话。

• 当前的话与前面的某些真的话冲突，就是假话；
• 当前的话中 $X$ 或 $Y$ 比 $N$ 大，就是假话；
• 当前的话表示 $X$ 吃 $X$，就是假话。

你的任务是根据给定的 $N$ 和 $K$ 句话，输出假话的总数。

输入格式

第一行两个整数，$N,K$，表示有 $N$ 个动物，$K$ 句话。

第二行开始每行一句话（按照题目要求，见样例）

输出格式

一行，一个整数，表示假话的总数。

样例 #1

样例输入 #1

100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5

样例输出 #1

3

提示

对于全部数据，$1\le N\le 5\times 10^4$，$1\le K\le 10^5$。

N, K = map(int, input().split())
fa = [i for i in range(N*3+1)] #一倍存自己，二倍存猎物，三倍存天敌
def find(x):
    if fa[x] != x:
        fa[x]=find(fa[x])
    return fa[x]
def union(x,y):
    fa[find(x)] = find(y)

guanxi = []
for _ in range(K):
    guanxi.append(list(map(int, input().split())))
ans = 0
for gx in guanxi:
    flat,x,y = gx[0],gx[1],gx[2]
    if x>N or y>N:
        ans+=1
        continue
    if flat==1: # x与y是同类
        if find(x+N)==find(y) or find(x+2*N)==find(y): # 如果在x的猎物或者x的天敌中找到了y那么说明这句话是假话
            ans+=1
            continue
        else:
            union(x,y);union(x+N,y+N);union(x+2*N,y+2*N) # 合并x与y的关系
    if flat==2: # x吃y 意味着y的天敌是x,x的猎物是y,y的猎物是x的天敌
        if find(x)==find(y) or find(x+2*N)==find(y): # 如果x跟y的关系是同类，或者x的天敌是y,那么是假话
            ans+=1
            continue
        else:
            union(x,y+2*N);union(x+N,y);union(x+2*N,y+N) #y的天敌是x,x的猎物是y,y的猎物是x的天敌
print(ans)