动物王国中有三类动物 A,B,CA,B,C,这三类动物的食物链构成了有趣的环形。
AA 吃 BB,BB 吃 CC,CC 吃 AA。
现有 NN 个动物,以 1∼N1∼N 编号。
每个动物都是 A,B,CA,B,C 中的一种,但是我们并不知道它到底是哪一种。
有人用两种说法对这 NN 个动物所构成的食物链关系进行描述:
第一种说法是 1 X Y
,表示 XX 和 YY 是同类。
第二种说法是 2 X Y
,表示 XX 吃 YY。
此人对 NN 个动物,用上述两种说法,一句接一句地说出 KK 句话,这 KK 句话有的是真的,有的是假的。
当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
- 当前的话与前面的某些真的话冲突,就是假话;
- 当前的话中 XX 或 YY 比 NN 大,就是假话;
- 当前的话表示 XX 吃 XX,就是假话。
你的任务是根据给定的 NN 和 KK 句话,输出假话的总数。
输入格式
第一行是两个整数 NN 和 KK,以一个空格分隔。
以下 KK 行每行是三个正整数 D,X,YD,X,Y,两数之间用一个空格隔开,其中 DD 表示说法的种类。
若 D=1D=1,则表示 XX 和 YY 是同类。
若 D=2D=2,则表示 XX 吃 YY。
输出格式
只有一个整数,表示假话的数目。
数据范围
1≤N≤500001≤N≤50000,
0≤K≤1000000≤K≤100000
输入样例:
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
输出样例:
3
解题思路:本题是一道经典并查集问题,主要考察并查集维护信息,可以设置距离根节点的距离为判断依据,如距离根节点距离模3为1表示该动物吃根节点,距离模3为2表示该动物被根节点吃,距离模3为0表示该动物与根节点动物同类。
具体可在并查集常规查找根节点方法上进行距离改变
并且在后续改变中,由于可先进行循环查找根节点,故可直接将根节点距离调整
代码如下:
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
@SuppressWarnings("all")
public class Main {
//分别表示并查集数组和距离数组
static int[] arr;
static int[] brr;
//分别表示n个动物和k句话
static int n,k;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String[] s = br.readLine().split(" ");
n = Integer.valueOf(s[0]);
k = Integer.valueOf(s[1]);
init();
int ans = 0;
int d,x,y,px,py;
while(k-- > 0){
s = br.readLine().split(" ");
d = Integer.valueOf(s[0]);
x = Integer.valueOf(s[1]);
y = Integer.valueOf(s[2]);
if(x > n || y > n){
ans++;
}else{
px = find(x);
py = find(y);
if(d == 1){
if(px == py && (brr[x] - brr[y]) % 3 != 0){
ans++;
}else if(px != py){
arr[px] = py;
brr[px] = brr[y] - brr[x];
}
}else if(d == 2){
if(px == py && (brr[x] - brr[y] - 1) % 3 != 0){
ans++;
}else if(px != py){
arr[px] = arr[py];
brr[px] = brr[y] - brr[x] + 1;
}
}
}
}
System.out.println(ans);
}
//初始化数组
public static void init(){
arr = new int [n + 5];
brr = new int [n + 5];
for(int i = 1; i <= n; i++){
arr[i] = i;
}
}
//并查集查找并且维护距离数组
public static int find(int x){
if(x != arr[x]){
int t = find(arr[x]);
brr[x] += brr[arr[x]];
arr[x] = t;
}
return arr[x];
}
}