在java中排序2d数据结构

时间:2011-08-24 02:17:08

标签: java arrays sorting 2d hashmap

我需要按值对2d键/值对进行排序。我已经在网上阅读了很多关于这个的参考资料,并且最终编写了我自己的类来使用HashMaps(见下文)。我将代码放入一个精简的工作类中,用最少量的代码重现问题,这样您就可以将其剪切并粘贴到IDE中以便快速诊断。

正如您所看到的,我编写的方法是在将值输入sortedMap之前正确对值进行排序。但是,出于某种原因,当我尝试随后遍历sortedMap时,它们的值再次以不同的方式排序。

任何人都可以告诉我如何修复下面的代码,以便迭代生成的2D数据对象,以降序为我提供数据吗?


编辑: 使用TreeMaps重新编写,我仍然遇到类似的问题。这是重写的代码:

import java.util.*;

public class HashMapDemo {
    public static void main(String args[]) {

        // Code that creates and populates the unordered HashMap:
        TreeMap<Integer, Double> unSortedMap = new TreeMap<Integer, Double>();
        unSortedMap.put(1343, 0.521851);
        unSortedMap.put(1950, -0.301208);
        unSortedMap.put(3667, -0.0280762);
        unSortedMap.put(3879, 0.154724);
        unSortedMap.put(4124, 0.022583);

        // Code that calls the ordering method:
        TreeMap<Integer, Double> sortedMap = new TreeMap<Integer, Double>(
                sortTreeMap(unSortedMap));

        // Code that iterates through the "sorted" hashmap.
        System.out.println("now iterate through sortedMap: ");
        for (Integer key : sortedMap.keySet()) {
            System.out.println("key, sortedMap.get(key) are:  " + key + ", "
                    + sortedMap.get(key));
        }
    }

    // Code for the ordering method. Note that the println tests indicate that
    // this method is correctly sorting the key/value pairs in the hashmap:
    private static TreeMap<Integer, Double> sortTreeMap(
            TreeMap<Integer, Double> input) {

        System.out
                .println("input.size() upon entering sortHasMap() function is: "
                        + input.size());
        int startSize = input.size();

        // create a hashmap to store sorted output
        TreeMap<Integer, Double> sortedMap = new TreeMap<Integer, Double>();

        // repeat the following process once for every key/value pair in the
        // hashmap
        for (int i = 0; i < startSize; i++) {
            int mySize = input.size();
            System.out.println("mySize is: " + mySize);
            double maxVal = Double.NEGATIVE_INFINITY;
            int maxKey = 0;

            // iterate through each key in hashmap to find key/value of max
            // value
            for (Integer key : input.keySet()) {
                if (maxVal < input.get(key)) {
                    maxVal = input.get(key);
                    maxKey = key;
                }
            }

            // add key/value of max of that iteration to sorted map and remove
            // from input before next iteration
            sortedMap.put(maxKey, maxVal);

            input.remove(maxKey);

            System.out.println("sortedMap.put(maxKey, maxVal) are: " + maxKey
                    + ", " + maxVal);
        }
        return sortedMap;
    }
}

2 个答案:

答案 0 :(得分:5)

您需要仅两行才能达到您想要的效果。以下是这两行:

    Map<Integer, Double> sortedMap = new TreeMap<Integer, Double>(new Comparator<Integer>() {
        public int compare(Integer o1, Integer o2) {
            return map.get(o2).compareTo(map.get(o1)); // reverse order of values
        }
    });
    sortedMap.putAll(map);

这是完整的可运行代码:

public static void main(String[] args) {
    final Map<Integer, Double> map = new HashMap<Integer, Double>();
    map.put(1343, 0.521851);
    map.put(1950, -0.301208);
    map.put(3667, -0.0280762);
    map.put(3879, 0.154724);
    map.put(4124, 0.022583);
    Map<Integer, Double> sortedMap = sortMap(map);

    for (Map.Entry<Integer, Double> entry : sortedMap.entrySet()) {
        System.out.println(entry.getKey() + ", " + entry.getValue());
    }
}

public static Map<Integer, Double> sortMap(final Map<Integer, Double> map) {
    Map<Integer, Double> sortedMap = new TreeMap<Integer, Double>(new Comparator<Integer>() {
        public int compare(Integer o1, Integer o2) {
            return map.get(o2).compareTo(map.get(o1));
        }
    });
    sortedMap.putAll(map);
    return sortedMap;
}

输出:

1343, 0.521851
3879, 0.154724
4124, 0.022583
3667, -0.0280762
1950, -0.301208

注意:通过将要使用的Comparator传递给构造函数,指定如何在TreeSet中排序条目。 TreeSet实现完成剩下的工作。

其他说明:

  • 迭代地图的键/值的最佳方法是迭代Map.entrySet()
  • 始终对您的变量使用抽象类型 - 例如Map<?, ?> myMap 具体实现(即HashMap<?, ?> myMap

以下是sortMap方法的通用版本,它将根据值的相反顺序对任何合适的地图进行排序:

public static <K, V extends Comparable<V>> Map<K, V> sortMap2(final Map<K, V> map) {
    Map<K, V> sortedMap = new TreeMap<K, V>(new Comparator<K>() {
        public int compare(K o1, K o2) {
            return map.get(o2).compareTo(map.get(o1));
        }
    });
    sortedMap.putAll(map);
    return sortedMap;
}

答案 1 :(得分:4)

使用TreeMap。它是SortedMap,它使用键的自然顺序。在您的情况下,它将按Integer键排序。

它为你排序。

java.util.HashMap无序。

  

此课程不保证地图的顺序;在   特别是,它不保证订单将保持不变   随着时间的推移。

EDIT 正确阅读问题后

使用Comparator作为比较值的TreeMap构造函数。