3 回答
TA贡献1818条经验 获得超7个赞
只需Collections.sort(temp);在最后一个 for 循环中包含这个,这是必要的,因为多个数字可以具有相同的 digitRoot 并且应该放在已排序的列表中。
for (int i=0;i<a.length;i++) {
ArrayList<Integer> temp = map.get(a1[i]);
Collections.sort(temp);
for(int j=0;j<temp.size();j++) {
a[i]=temp.get(j);
if (j<temp.size()-1)
i++;
}
}
Input: a: [13, 20, 7, 4]
Output: [20, 4, 13, 7]
编辑:关于错误
因为在put(a1[i], Collections.sort(list))put 方法中是期望的put(int, List),但是你给了它put(int, void),因为返回类型Collections.sort()是void,你只需要先对列表进行排序然后再通过
TA贡献1876条经验 获得超7个赞
您似乎正在尝试插入集合的排序值。Collections.sort(List)
,无论好坏,就地对该列表进行排序并且不返回任何内容。先对列表进行排序,然后将其插入到地图中。
TA贡献1868条经验 获得超4个赞
放入地图时不需要对数组进行排序。相反,您可以在最后一个循环中检索时对其进行排序:
Arrays.sort(a1);
for (int i=0;i<a.length;i++) {
ArrayList<Integer> temp = map.get(a1[i]);
Collections.sort(temp);
for(int j=0;j<temp.size();j++) {
a[i]=temp.get(j);
if (j<temp.size()-1)
i++;
}
}
如果您需要在放入地图时进行排序,您应该使用 SortedSet 因为它会自动保持元素排序:
int[] digitRootSort(int[] a) {
HashMap<Integer, TreeSet<Integer>> map = new HashMap<Integer, TreeSet<Integer>>();
int[] a1 = new int[a.length];
for (int i = 0; i < a.length; i++) {
a1[i] = digitRoot(a[i]);
if (map.containsKey(a1[i])) {
TreeSet<Integer> set = map.get(a1[i]);
set.add(a[i]);
map.put(a1[i], set);
} else {
TreeSet<Integer> set = new TreeSet<Integer>();
set.add(a[i]);
map.put(a1[i], set);
}
}
Arrays.sort(a1);
for (int i = 0; i < a.length;) {
TreeSet<Integer> set = map.get(a1[i]);
for (int j : set) {
a[i] = j;
i++;
}
}
return a;
}
但是,通过使用适当的比较器,还有一种更简单的方法来处理排序集:
int[] digitRootSort(int[] a) {
SortedSet<Integer> set = new TreeSet<Integer>(new Comparator<Integer>() {
@Override
public int compare(Integer a, Integer b) {
int result = Integer.compare(digitRoot(a), digitRoot(b));
result = result == 0 ? Integer.compare(a, b) : result;
return result;
}
});
for (int i : a) {
set.add(i);
}
int i = 0;
for (int j : set) {
a[i++] = j;
}
return a;
}
添加回答
举报