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

题目

给你一个整数数组 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 <= 10^5
matches[i].length == 2
1 <= winneri, loseri <= 10^5
winneri != loseri
所有 matches[i] 互不相同

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-players-with-zero-or-one-losses
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

使用哈希表记录每个人失败的次数，再使用一个集合记录战胜的人，战胜的人后面还可能战败，因此还需要再移除。

因为需要返回有序的结果，因此和都使用有序集合。

代码

class Solution {
    public List<List<Integer>> findWinners(int[][] matches) {
        List<List<Integer>> ans = new ArrayList<>();
        // 一次都没战败过的
        TreeSet<Integer> win = new TreeSet<>();
        // 记录失败次数
        TreeMap<Integer, Integer> loss = new TreeMap<>();
        for (int[] m : matches) {
            // 目前是候选的
            win.add(m[0]);
            loss.put(m[1], loss.getOrDefault(m[1], 0) + 1);
        }
        List<Integer> one = new ArrayList<>();
        for (Map.Entry<Integer, Integer> entry : loss.entrySet()) {
            // 将战败过的但是在win集合中的移除
            win.remove(entry.getKey());
            if (entry.getValue() == 1) {
                one.add(entry.getKey());
            }
        }
        ans.add(new ArrayList<>(win));
        ans.add(one);
        return ans;
    }
}