有 n 位用户参加活动，他们的 ID 从 0 到 n - 1，每位用户都 恰好 属于某一用户组。给你一个长度为 n 的数组 groupSizes，其中包含每位用户所处的用户组的大小，请你返回用户分组情况（存在的用户组以及每个组中用户的 ID）。
你可以任何顺序返回解决方案，ID 的顺序也不受限制。此外，题目给出的数据保证至少存在一种解决方案。
示例 1：
输入：groupSizes = [ 3, 3, 3, 3, 3, 1, 3 ]
输出：[ [ 5 ], [ 0, 1, 2 ] , [ 3, 4, 6 ] ]
其他可能的解决方案有
[ [ 2, 1, 6 ], [ 5 ], [ 0, 4, 3] ]
和
[ [ 5 ], [ 0, 6, 2 ] ,[ 4, 3, 1 ] ]。
示例 2：
输入：groupSizes = [ 2, 1, 3, 3, 3, 2 ]
输出：[ [ 1 ], [ 0, 5 ], [ 2, 3, 4 ] ]
提示：
groupSizes.length == n
1 <= n <= 500
1 <= groupSizes[i] <= n
思路：
使用Hashmap
以groupSize为key, 以ArrayList(ArrayList())为value, 这里存放的Integer就是userId(也就是数组index)
插入策略：
先判断是否有key，无key则直接生成新的ArrayList(ArrayList())， 并且添加一个子ArrayList, 同时插入userId
有key再判断取出的ArrayList内部的最后一个ArrayList长度是否等于对应的groupSize， 如果相等， 则追加到新组， 否则追加
题解：
public class Test {
public static void main(String[] args) {
function(){ //Axitrader代理申请 http://www.kaifx.cn/broker/axitrader.html
System.out.println(groupThePeople(new int[]{3,3,3,3,1,3,3}));
}
private static HashMap<Integer, ArrayList> resultMap = new HashMap<>();
public static List<List<Integer>> groupThePeople(int[] groupSizes) {
ArrayList<List<Integer>> results = new ArrayList<>();
for (int i=0;i<groupSizes.length;i++) {
add(i,groupSizes[i]);
}
for (ArrayList i : resultMap.values()) {
results.addAll(i);
}
return results;
}
static void add(int id, int groupSize) {
if (resultMap.get(groupSize) == null) {
ArrayList<List<Integer>> groupList = new ArrayList<>();
List<Integer> group=new ArrayList<>();
group.add(id);
groupList.add(group);
resultMap.put(groupSize, groupList);
} else {
ArrayList<List<Integer>> a = resultMap.get(groupSize);
if (a.get(a.size() - 1).size() == groupSize ) {
List<Integer> group=new ArrayList<>();
group.add(id);
a.add(group);
} else {
a.get(a.size() - 1).add(id);
}
}
}
}
|
|
收藏
淘帖
踩
