日渐头秃的代码日记 -- 8 LC第287周赛第二题

找出输掉零场或一场比赛的玩家

给你一个整数数组 matches 其中 matches[i] = [winneri, loseri] 表示在一场比赛中 winneri 击败了 loseri 。

返回一个长度为 2 的列表 **answer ：

• answer[0] 是所有 没有 输掉任何比赛的玩家列表。
• answer[1] 是所有恰好输掉 一场 比赛的玩家列表。

两个列表中的值都应该按 递增 顺序返回。

注意：

• 只考虑那些参与 至少一场 比赛的玩家。
• 生成的测试用例保证 不存在 两场比赛结果 相同 。

示例 1：

输入： matches = [[1,3],[2,3],[3,6],[5,6],[5,7],[4,5],[4,8],[4,9],[10,4],[10,9]]
输出： [[1,2,10],[4,5,7,8]]
解释：
玩家 1、2 和 10 都没有输掉任何比赛。
玩家 4、5、7 和 8 每个都输掉一场比赛。
玩家 3、6 和 9 每个都输掉两场比赛。
因此，answer[0] = [1,2,10] 和 answer[1] = [4,5,7,8] 。

示例 2：

输入： matches = [[2,3],[1,3],[5,4],[6,4]]
输出： [[1,2,5,6],[]]
解释：
玩家 1、2、5 和 6 都没有输掉任何比赛。
玩家 3 和 4 每个都输掉两场比赛。
因此，answer[0] = [1,2,5,6] 和 answer[1] = [] 。

提示：

• 1 <= matches.length <= 105
• matches[i].length == 2
• 1 <= winneri, loseri <= 105
• winneri != loseri
• 所有 matches[i] 互不相同

这也是一道简单题目，读懂题目意思即可，利用集合的减法就可以求出来。需要注意的是，不能直接使用list的count函数，会超时。

from collections import defaultdict
class Solution:
    def findWinners(self, matches: List[List[int]]) -> List[List[int]]:
        winner_list = [m[0] for m in matches]
        loser_list = [m[1] for m in matches]
        a0 = sorted(list(set(winner_list) - set(loser_list)))
        a1 = []
        lose_count = defaultdict(int)
        for l in loser_list:
            lose_count[l] += 1
        for l, count in lose_count.items():
            if count == 1:
                a1.append(l)
        return a0, sorted(a1)