力扣每日一题2021.08.14统计不开心的朋友

---

1583.统计不开心的朋友

题目描述

给你一份n位朋友的亲近程度列表，其中n总是偶数。
对每位朋友i，preferences[i]包含一份按亲近程度从高到低排列的朋友列表。换句话说，排在列表前面的朋友与i的亲近程度比排在列表后面的朋友更高。每个列表中的朋友均以0到n-1之间的整数表示。
所有的朋友被分成几对，配对情况以列表pairs给出，其中pairs[i] = [x_i, y_i]表示x_i与y_i配对，且y_i与x_i配对。
但是，这样的配对情况可能会使其中部分朋友感到不开心。在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
• x_i != y_i
• 0 <= x_i, y_i <= n-1
• 每位朋友都恰好被包含在一对中

---

思路：模拟

首先对preferences和pairs进行预处理。
创建一个n行n列的二维列表order，其中order[i][j]表示朋友j在i的朋友列表中的亲近程度下标。遍历preferences就可以填入全部数据。特别的，order[i][i]都以初始值作为结果，不做改变。
创建长度为n的match列表，如果x和y配对，则match[x] = y，match[y] = x。
做完这些预处理后，就可以开始模拟了。
遍历0到n-1的每一位朋友作为x，在match列表中找到与x配对的朋友y。找到朋友y在朋友x的朋友列表中的亲近程度下标，并记录为index。在朋友x的朋友列表中的下标从0到index-1的朋友都是可能的u。遍历所有的u，找到与朋友u配对的朋友v。如果order[u][x] < order[u][v]，则x是不开心的朋友。
对于每一个x，只要找到一个满足条件的四元组(x, y, u, v)，则x就是不开心的。

Python代码

class Solution:
    def unhappyFriends(self, n: int, preferences: List[List[int]], pairs: List[List[int]]) -> int:
        order = [[0] * n for i in range(n)]
        for i in range(n):
            for j in range(n-1):
                order[i][preferences[i][j]] = j
        
        match = [0] * n
        for x, y in pairs:
            match[x] = y
            match[y] = x
            
        unhappyPerson = 0
        for x in range(n):
            y = match[x]
            index = order[x][y]
            for i in range(index):
                u = preferences[x][i]
                v = match[u]
                if order[u][x] < order[u][v]:
                    unhappyPerson += 1
                    break
        return unhappyPerson