题目描述
给你一份 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
返回 不开心的朋友的数目 。
样例
输入: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 都是开心的。
输入:n = 2, preferences = [[1], [0]], pairs = [[1, 0]]
输出:0
解释:朋友 0 和 1 都开心。
输入: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
- 每位朋友都 恰好 被包含在一对中
算法分析
模拟
感觉是一种阅读理解题,容易把自己绕晕hh
若当前给定两对关系(a, b)
和 (c, d)
,如何让a
不开心?分两种情况,任意一种a
都会不开心
- 1、只要
c
在a
心目中 比b
在a
心目中 更亲密,且a
在c
心目中 比d
在c
心目中更亲密,则a
就会不开心 - 2、只要
d
在a
心目中 比b
在a
心目中 更亲密,且a
在d
心目中 比c
在d
心目中更亲密,则a
就会不开心
枚举所有的关系,将不开心的人都加入到set
中进行去重(因为可能会存在不开心的人是同一个),最后返回set.size()
时间复杂度 $O(n^3)$
Java 代码
class Solution {
static int get(int[] p, int x)
{
for(int i = 0;i < p.length;i ++)
if(p[i] == x)
return i;
return -1;
}
static boolean check(int[][] preferences, int x, int y, int u, int v)
{
//找到u在x中亲密度的位置
int a = get(preferences[x], u);
//找到y在x中亲密度的位置
int b = get(preferences[x], y);
//找到x在u中亲密度的位置
int c = get(preferences[u], x);
//找到v在u中亲密度的位置
int d = get(preferences[u], v);
if(a < b && c < d) return true;
return false;
}
public int unhappyFriends(int n, int[][] preferences, int[][] pairs) {
HashSet<Integer> set = new HashSet<Integer>();
for(int i = 0;i < pairs.length;i ++)
for(int j = 0;j < pairs.length;j ++)
{
if(i == j) continue;
int a = pairs[i][0], b = pairs[i][1], c = pairs[j][0], d = pairs[j][1];
//让a不开心
if(check(preferences, a, b, c, d) || check(preferences, a, b, d, c)) set.add(a);
//让b不开心
if(check(preferences, b, a, c, d) || check(preferences, b, a, d, c)) set.add(b);
}
return set.size();
}
}