1583. 統計不開心的朋友
給你一份 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
每位朋友都 恰好 被包含在一對中
解題思路
- 使用數組l[i][j][k]=true記錄對于i來說j的親密度大于k
- 使用map記錄下,每個朋友配對的另一位朋友
- 對每一位朋友x,檢查其余的所有朋友u,y為x的配對朋友,v為u的配對朋友
- 如果存在x 與 u 的親近程度勝過 x 與 y,u 與 x 的親近程度勝過 u 與 v,那么就將x和u加入不開心朋友的集合
代碼
class Solution {public int unhappyFriends(int n, int[][] preferences, int[][] pairs) {//l[i][j]boolean[][][] l=new boolean[n][n][n];for (int k=0;k<n;k++) {for (int i=0;i<n-2;i++)for (int j=i+1;j<n-1;j++)l[k][preferences[k][i]][preferences[k][j]]=true;}Map<Integer,Integer>map=new HashMap<>();for (int[] pair : pairs) {map.put(pair[0],pair[1]);map.put(pair[1],pair[0]);}int cnt=0;Set<Integer> set=new HashSet<>();for (int i=0;i<n;i++){int o1=i,o2=map.get(i);for (int j=0;j<n;j++){if (j==o1||j==o2) continue;if (l[i][j][o2]&&l[j][i][map.get(j)]){set.add(i);set.add(j);}}}return set.size();}
}