文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

給你一個整數 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

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/process-restricted-friend-requests
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

參考：數據結構 并查集

• 使用并查集維護好友關系
• 對于每次的請求[a, b]，查找請求兩端的代表 fa，fb，遍歷所有的限制條件 [r0, r1]，也查找其代表 f0, f1，如果能匹配上 (fa==f0 && fb==f1) || (fa==f1 && fb==f0)，則他們不能連通

class dsu{vector<int> f; public:dsu(int n){f.resize(n);for(int i = 0; i < n; ++i)f[i] = i;}void merge(int a, int b){if(!isfriend(a, b))f[a] = b;}int find(int a){if(f[a] == a) return a;return f[a] = find(f[a]);}bool isfriend(int a, int b){int fa = find(a), fb = find(b);return fa==fb;} }; class Solution { public:vector<bool> friendRequests(int n, vector<vector<int>>& restrictions, vector<vector<int>>& requests) {dsu u(n);//并查集vector<bool> ans(requests.size(), true);for(int i = 0; i < requests.size(); ++i){int a = requests[i][0], b = requests[i][1];if(u.isfriend(a, b))continue;int fa = u.find(a), fb = u.find(b);bool flag = true;for(auto& r : restrictions){int f0 = u.find(r[0]), f1 = u.find(r[1]);if((fa==f0 && fb==f1) || (fa==f1 && fb==f0)){flag = false;break;}}ans[i] = flag;// if(flag) u.merge(a, b); // 錯誤解if(flag) u.merge(fa, fb); // 正確的解，把最頂層的父節點合并}return ans;} };

256 ms 21.4 MB C++

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的LeetCode 2076. 处理含限制条件的好友请求（并查集）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。