LeetCode1282.用户分组

LeetCode1282.用户分组

原题链接：LeetCode1282.用户分组

思路

可以看出，一个元素所在组内的值如果和另一个元素所在的组内的值不同，那么这两个元素必然不在同一个组内

把可能同一个组的元素放到一个链内.

比如：groupSizes = [3,3,3,3,3,1,3]，就把g[0~4],g[6]放到一个链内,g[5]放到一个链内.

可以像下图这样，使用一个哈希表hash[3] = [0,1,2][3,4,6],hash[1] = [6];

然后遍历的时候每当hash[x]的大小为x的时候，就把这个vector放入到结果vector中即可

代码

c++代码

class Solution {
public:
    vector<vector<int>> groupThePeople(vector<int>& groupSizes) {
        unordered_map<int,vector<int>> hash;
        vector<vector<int>> res;

        for(int i = 0 ; i < groupSizes.size(); i ++)
        {
            int x = groupSizes[i];
            hash[x].push_back(i);

            if(hash[x].size() == x)
            {
                res.push_back(hash[x]);
                hash[x].clear();
            }
        }

        return res;
    }
};

java代码

class Solution {
    public List<List<Integer>> groupThePeople(int[] groupSizes) {
        
        Map<Integer,List<Integer>> hash = new HashMap<>();
        List<List<Integer>> res = new ArrayList<>();

        for(int i = 0 ; i < groupSizes.length; i ++)
        {
            int x = groupSizes[i];
            if(hash.get(x) == null) hash.put(x,new ArrayList<>());
            hash.get(x).add(i);
            if(hash.get(x).size() == x)
            {
                res.add(hash.get(x));
                hash.put(x,null);
            }
        }

        return res;
    }
}

标签：hash,get,int,res,用户,LeetCode1282,vector,分组,groupSizes
来源： https://www.cnblogs.com/rdisheng/p/16581267.html