1. 程式人生 > 其它 >240. 食物鏈(並查集+數論)

240. 食物鏈(並查集+數論)

技術標籤:技術

動物王國中有三類動物A,B,C,這三類動物的食物鏈構成了有趣的環形。

A吃B, B吃C,C吃A。

現有N個動物,以1-N編號。

每個動物都是A,B,C中的一種,但是我們並不知道它到底是哪一種。

有人用兩種說法對這N個動物所構成的食物鏈關係進行描述:

第一種說法是”1 X Y”,表示X和Y是同類。

第二種說法是”2 X Y”,表示X吃Y。

此人對N個動物,用上述兩種說法,一句接一句地說出K句話,這K句話有的是真的,有的是假的。

當一句話滿足下列三條之一時,這句話就是假話,否則就是真話。

1) 當前的話與前面的某些真的話衝突,就是假話;
2) 當前的話中X或Y比N大,就是假話;

3) 當前的話表示X吃X,就是假話。

你的任務是根據給定的N和K句話,輸出假話的總數。

輸入格式

第一行是兩個整數N和K,以一個空格分隔。

以下K行每行是三個正整數 D,X,Y,兩數之間用一個空格隔開,其中D表示說法的種類。

若D=1,則表示X和Y是同類。

若D=2,則表示X吃Y。

輸出格式

只有一個整數,表示假話的數目。

資料範圍

1N500001≤N≤50000,
0K1000000≤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的值分類

三種動物的吃與被吃關係就建立起來了

所以用並查集維護節點到根節點的距離

圖一 給d[px]一個合理的值即可

 

圖二給d[px]一個合理的值即可

  

程式碼:
import java.util.*;

public class Main{
    static final int N=100005;
    static int p[]=new int[N];
    static int d[]=new int[N];
    static int n,k;
    static int find(int x){
        if(p[x]!=x){
            
int t=find(p[x]); d[x]+=d[p[x]]; p[x]=t; } return p[x]; } public static void main(String []args){ Scanner scan=new Scanner(System.in); n=scan.nextInt(); k=scan.nextInt(); for(int i=1;i<=n;i++) p[i]=i; int res=0; while(k-->0){ int D=scan.nextInt(); int x=scan.nextInt(); int y=scan.nextInt(); int px=find(x),py=find(y); if(x>n || y>n) res++; else{
          //如果是同一類動物  
if(D==1){ if(px==py && (d[x]-d[y])%3!=0) res++;//如果兩個動物在一個集合中,那麼它們到根節點的距離模3肯定相等,即差值模3等於0,不等於0就是假話 else if(px!=py){//如果不在一個集合中 見圖1 p[px]=py; d[px]=d[y]-d[x]; } }
          //如果不是同一類動物
else{ if(px==py && (d[x]-d[y]-1)%3!=0) res++;//如果兩個動物在一個集合中,題意x吃y,那麼x到根節點的距離取模肯定是比y到根節點的距離取模大1,所以,d[x]%3同餘(d[y]+1)%3,即差值模3等於0 else if(px!=py){//如果不在一個集合中 見圖2 p[px]=py; d[px]=d[y]+1-d[x]; } } } } System.out.println(res); } }