-
統計不開心的朋友
顯示英文描述
通過的使用者數 949
嘗試過的使用者數 1240
使用者總通過次數 953
使用者總送出次數 1750
題目難度 Medium
給你一份 n 位朋友的親近程度清單,其中 n 總是 偶數 。
對每位朋友 i,preferences[i] 包含一份 按親近程度從高到低排列 的朋友清單。換句話說,排在清單前面的朋友與 i 的親近程度比排在清單後面的朋友更高。每個清單中的朋友均以 0 到 n-1 之間的整數表示。
所有的朋友被分成幾對,配對情況以清單 pairs 給出,其中 pairs[i] = [xi, yi] 表示 xi 與 yi 配對,且 yi 與 xi 配對。
但是,這樣的配對情況可能會是其中部分朋友感到不開心。在 x 與 y 配對且 u 與 v 配對的情況下,如果同時滿足下述兩個條件,x 就會不開心:
x 與 u 的親近程度勝過 x 與 y,且
u 與 x 的親近程度勝過 u 與 v
傳回 不開心的朋友的數目 。
示例 1:
輸入:n = 4, preferences = [[1, 2, 3], [3, 2, 0], [3, 1, 0], [1, 2, 0]], pairs = [[0, 1], [2, 3]]
輸出:2
解釋:
朋友 1 不開心,因為:
- 1 與 0 配對,但 1 與 3 的親近程度比 1 與 0 高,且
-
3 與 1 的親近程度比 3 與 2 高。
朋友 3 不開心,因為:
- 3 與 2 配對,但 3 與 1 的親近程度比 3 與 2 高,且
-
1 與 3 的親近程度比 1 與 0 高。
朋友 0 和 2 都是開心的。
示例 2:
輸入:n = 2, preferences = [[1], [0]], pairs = [[1, 0]]
輸出:0
解釋:朋友 0 和 1 都開心。
示例 3:
輸入:n = 4, preferences = [[1, 3, 2], [2, 3, 0], [1, 3, 0], [0, 2, 1]], pairs = [[1, 3], [0, 2]]
輸出:4
提示:
2 <= n <= 500
n 是偶數
preferences.length == n
preferences[i].length == n - 1
0 <= preferences[i][j] <= n - 1
preferences[i] 不包含 i
preferences[i] 中的所有值都是獨一無二的
pairs.length == n/2
pairs[i].length == 2
xi != yi
0 <= xi, yi <= n - 1
每位朋友都 恰好 被包含在一對中
- 用鄰接矩陣記錄每條邊的親近值
- 暴力枚舉每個人,根據題意判斷是否不高興即可
int mp[512][512];
class Solution {
public:
int unhappyFriends(int n, vector<vector<int>>& a, vector<vector<int>>& b) {
if(n == 1) return 0;
memset(mp, 0, sizeof(mp));
int ans = 0;
for(int i=0; i<n; i++)
for(int j=0, timer=n; j<n-1; j++)
mp[i][a[i][j]] = --timer;
int ID[MAXN] = { 0 }; //id[u]表示與u配對的人
for(auto it : b) {
int u = it[0], v = it[1];
ID[u] = v, ID[v] = u;
}
for(int x=0; x<n; x++) { //枚舉每個人, 判斷是否高興
int y = ID[x];
int ok = 0;
for(int i=0; i<n-1; i++) {
int u = a[x][i];
int v = ID[u];
if(mp[x][u] > mp[x][y] && mp[u][x] > mp[u][v]) ok = true;
}
ans += ok;
}
return ans;
}
};