給你一個整數 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:
輸入: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) 。
示例 3:
輸入:n = 5, restrictions = [[0,1],[1,2],[2,3]], requests = [[0,4],[1,2],[3,1],[3,4]]
輸出:[true,false,true,false]
解釋:
請求 0 :用戶 0 和 用戶 4 可以成為朋友,所以他們成為直接朋友。?
請求 1 :用戶 1 和 用戶 2 不能成為朋友,因為他們之間存在限制。
請求 2 :用戶 3 和 用戶 1 可以成為朋友,所以他們成為直接朋友。?
請求 3 :用戶 3 和 用戶 4 不能成為朋友,因為這會使 用戶 0 和 用戶 1 成為間接朋友 (0--4--3--1) 。
?
提示:
2 <= n <= 1000
0 <= restrictions.length <= 1000
restrictions[i].length == 2
0 <= xi, yi <= n - 1
xi != yi
1 <= requests.length <= 1000
requests[j].length == 2
0 <= uj, vj <= n - 1
uj != vj
思路:并查集的運用。我們可以借助并查集來維護已經構成朋友關系的集合,對于當前查詢x和y,他們可以組成朋友的情況如下:
1. x和y已經在同一個朋友集合中, 則它們一定可以成為朋友
2. x和y不在一個朋友集合中,我們需要考慮所有不能成為朋友的元素對。假設元素對為u和v,判斷他們的朋友集合與x和y集合的關系,若f(x)==f(u) && f(y)==f(v) || f(x)==f(v) && f(y)==f(u)時,他們無法構成朋友關系,否則可以。【f(x)指的是x所在的朋友集合編號】
class Solution {
public:
int f[1005];
int find(int x) {
if (f[x] == x)
return x;
return f[x] = find(f[x]);
}
vector<bool> friendRequests(int n, vector<vector<int>>& restrictions, vector<vector<int>>& requests) {
vector<bool> ans(requests.size());
for (int i = 0; i < n; i++)
f[i] = i;
for (int i = 0;i < requests.size(); i++) {
int t1 = find(requests[i][0]);
int t2 = find(requests[i][1]);
if (t1 == t2)
ans[i] = true;
else {
bool flag = true;
for (int j = 0;j < restrictions.size(); j++) {
int u = find(restrictions[j][0]);
int v = find(restrictions[j][1]);
if (t1 == u && t2 == v || t1 == v && t2 == u) {
flag = false;
break;
}
}
ans[i] = flag;
if (flag)
f[t1] = t2;
}
}
return ans;
}
};