【题解 | 并查集】给前来参宴的各位宾客安排座位，敌对关系不能同席

排座位

布置宴席最微妙的事情，就是给前来参宴的各位宾客安排座位。无论如何，总不能把两个死对头排到同一张宴会桌旁！这个艰巨任务现在就交给你，对任何一对客人，请编写程序告诉主人他们是否能被安排同席。

输入格式：

输入第一行给出3个正整数：N（≤100），即前来参宴的宾客总人数，则这些人从1到N编号；M为已知两两宾客之间的关系数；K为查询的条数。随后M行，每行给出一对宾客之间的关系，格式为：宾客1 宾客2 关系，其中关系为1表示是朋友，-1表示是死对头。注意两个人不可能既是朋友又是敌人。最后K行，每行给出一对需要查询的宾客编号。

这里假设朋友的朋友也是朋友。但敌人的敌人并不一定就是朋友，朋友的敌人也不一定是敌人。只有单纯直接的敌对关系才是绝对不能同席的。

输出格式：

对每个查询输出一行结果：如果两位宾客之间是朋友，且没有敌对关系，则输出No problem；如果他们之间并不是朋友，但也不敌对，则输出OK；如果他们之间有敌对，然而也有共同的朋友，则输出OK but…；如果他们之间只有敌对关系，则输出No way。

输入样例：

7 8 4
5 6 1
2 7 -1
1 3 1
3 4 1
6 7 -1
1 2 1
1 4 1
2 3 -1
3 4
5 7
2 3
7 2

输出样例：

No problem
OK
OK but...
No way

解题思路

纯模拟题，使用并查集来维护宾客之间的朋友关系，使用二维数组记录宾客之间的敌对关系。

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.io.StreamTokenizer;

public class Main {
	static final int MAXN = 101;
	
	static boolean[][] d = new boolean[MAXN][MAXN];
	
    static int[] pre = new int[MAXN];
    
    static BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
    static StreamTokenizer in = new StreamTokenizer(bf);
    static PrintWriter out = new PrintWriter(System.out);

    static int find(int x) {
    	if(x == pre[x]) {
    		return x;
    	}
    	return pre[x] = find(pre[x]);
    }

    static void union(int a, int b) {
        int fa = find(a);
        int fb = find(b);
        if(fa != fb) {
        	pre[fa] = fb;
        }
    }
    
    static void build() {
    	for(int i = 0; i<MAXN; i++) {
    		pre[i] = i;
    		for(int j = 0; j<MAXN; j++) {
    			d[i][j] = false;
    		}
    	}
    }
    
    static int n, m, k;

    public static void main(String[] args) throws IOException {
    	// 初始化
        build();
        n = nextInt();	// n个人
        m = nextInt();	// m个关系
        k = nextInt();	// k次查询
        
        for(int i = 0, u, v, p; i<m; i++) {
        	u = nextInt();
        	v = nextInt();
        	p = nextInt();
        	if(p == 1) {
        		// 朋友的朋友也是朋友
        		union(u, v);
        	} else {
        		// 如果是敌对关系
        		d[u][v] = true;
        		d[v][u] = true;
        	}
        }
        
        for(int i = 0, u, v; i<k; i++) {
        	u = nextInt();
        	v = nextInt();
        	
        	if(find(v) == find(u)) {
        		if(!d[u][v]) {
        			// 如果两位宾客之间是朋友，且没有敌对关系，则输出No problem
        			out.println("No problem");
        		} else {
        			// 如果他们之间有敌对，然而也有共同的朋友，则输出OK but...
        			out.println("OK but...");
        		}
        	} else {
        		// 如果他们之间并不是朋友，但也不敌对，则输出OK；
        		if(!d[u][v]) {
        			out.println("OK");
        		} else {
        			// 如果他们之间只有敌对关系，则输出No way
        			out.println("No way");
        		}
        	}
        }
        
        
        // 刷新输出缓冲区
        out.flush();
    }
    
    static int nextInt() throws IOException {
        in.nextToken();
        return (int) in.nval;
    }
}

查阅文档，了解并查集：【数据结构 | 并查集】维护元素分组信息，支持高效合并集合、查询元素所在集合