leetcode刷题记录15-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] 互不相同

问题分析：

虽然这道每日一题是中等难度，但我撇了一眼评论区发现大家都觉得是简单题的水平。。。自己写完发现时间消耗巨大，主要是写这种题的时候脑子总会想很多多余的东西。总之说一下我的方法：首先通过一个set来存储到底有几个人在比赛，记录对应的序号。然后用一个unordered_map来记录谁输了几场比赛，接着for遍历set，看看能不能在map里面找到，找不到就是没输，找到了值为1就是只输一场比赛。

代码如下：

class Solution {
public:
    vector<vector<int>> findWinners(vector<vector<int>>& matches) {
        unordered_map<int, int> mp;
        set<int> st;
        for(auto& v : matches){
            st.insert(v[0]);
            st.insert(v[1]);
            mp[v[1]] ++;
        }
        vector<vector<int>> answer(2);

        for(auto& s : st){
            if(mp.find(s) == mp.end()){
                answer[0].push_back(s);
            }
            else if(mp[s] == 1){
                answer[1].push_back(s);
            }
        }
        return answer;
    }
};