05 August 2008
给你一个整数 n ，表示网络上的用户数目。每个用户按从 0 到 n - 1 进行编号。
给你一个下标从 0 开始的二维整数数组 restrictions ，其中 restrictions[i] = [xi, yi] 意味着用户 xi 和用户 yi 不能 成为 朋友 ，不管是 直接 还是通过其他用户 间接 。
最初，用户里没有人是其他用户的朋友。给你一个下标从 0 开始的二维整数数组 requests 表示好友请求的列表，其中 requests[j] = [uj, vj] 是用户 uj 和用户 vj 之间的一条好友请求。
如果 uj 和 vj 可以成为 朋友 ，那么好友请求将会 成功 。每个好友请求都会按列表中给出的顺序进行处理（即，requests[j] 会在 requests[j + 1] 前）。一旦请求成功，那么对所有未来的好友请求而言， uj 和 vj 将会 成为直接朋友 。
返回一个 布尔数组 result ，其中元素遵循此规则：如果第 j 个好友请求 成功 ，那么 result[j] 就是 true ；否则，为 false 。
注意：如果 uj 和 vj 已经是直接朋友，那么他们之间的请求将仍然 成功 。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
示例 1：
输入：n = 3, restrictions = [[0,1]], requests = [[0,2],[2,1]]
输出：[true,false]
解释：
请求 0 ：用户 0 和 用户 2 可以成为朋友，所以他们成为直接朋友。
请求 1 ：用户 2 和 用户 1 不能成为朋友，因为这会使 用户 0 和 用户 1 成为间接朋友 (1--2--0) 。
示例 2：
输入：n = 3, restrictions = [[0,1]], requests = [[1,2],[0,2]]
输出：[true,false]
解释：
请求 0 ：用户 1 和 用户 2 可以成为朋友，所以他们成为直接朋友。
请求 1 ：用户 0 和 用户 2 不能成为朋友，因为这会使 用户 0 和 用户 1 成为间接朋友 (0--2--1) 。
用res数组记录不能成为朋友的状态，用pre相当于记录朋友圈的老大。并查集的具体算法就不多介绍了，许多大佬都讲的很清楚。 find函数用来查找朋友中的老大。 每一次更改老大，将小弟的res列表复制给老大
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
class Solution {
private int[] pre;
private boolean[][] res;
public boolean[] friendRequests(int n, int[][] restrictions, int[][] requests) {
pre = new int[n];
res = new boolean[n][n];
for (int i = 0; i < n; i++) {
pre[i] = i;
}
for (int[] restriction : restrictions) {
int x = restriction[0], y = restriction[1];
res[x][y] = true;
res[y][x] = true;
}
int m = requests.length;
boolean[] results = new boolean[m];
for (int i = 0; i < m; i++) {
int x = find(requests[i][0]), y = find(requests[i][1]);
if (x == y) {
results[i] = true;
} else {
if (res[x][y] || res[y][x]) {
results[i] = false;
} else {
pre[x] = y;
for (int j = 0; j < n; j++) {
if (res[x][j]) {
res[y][j] = true;
res[j][y] = true;
}
}
results[i] = true;
}
}
}
return results;
}
public int find(int x) {
int r = x;
while (pre[r] != r) {
r = pre[r];
}
int i = x, j;
while (i != r) {
j = pre[i];
pre[i] = r;
i = j;
}
return r;
}
}