如何打印数组中10个最小值的索引

Question

我需要从数组中选择10个最小的数字（有2000个项目）并打印它们的索引。

起初，我尝试对这个数组排序并打印值数组[0到9]。这是最小的数字，但我丢失了这些值的索引，它们是非排序数组。

第二种选择是尝试使用treeMap，它工作得很好，但当我有两个相等的键时，它只打印其中一个，但我需要同时打印这两个键。

将代码与treeMap一起使用的示例：

  TreeMap<Integer, String> treemap = new TreeMap<Integer, String>();

  treemap.put(2, "two");
  treemap.put(1, "one");
  treemap.put(3, "three");
  treemap.put(6, "six");
  treemap.put(6, "six2");
  treemap.put(5, "five");      

  Collection<String> coll=treemap.values();
  System.out.println("Value of the collection: "+coll);  

到目前为止，我还没有使用treeMap，所以有可能存在一些简单的方法来修复它。还是最好使用其他东西？

如果有任何帮助，我将不胜感激

Answer 1

创建一个简单类来保存两个int，或对任何值使用泛型：

class Item<T> { 
    public int index;
    public T value;
    public Item(int index, T value) {
        this.index = index;
        this.value = value;
    }
}

并利用集合。排序。

double[] original = { 1.0, 7.0, 3.0, -1.0, 5.0 };

List<Item> l = new ArrayList<Item<Double>>();
for (int i = 0; i < original.length; i++)
    l.add(new Item<Double>(i, original[i]));

Collections.sort(l, new Comparator<Item<Double>>() {
    public int compare(Item o1, Item o2) {
        return Double.compare(o1.value, o2.value);
    }
});

for (int i = 0; i < 10 && i < l.size(); i++) {
    System.out.printf("index: %f, value: %f%n", 
              l.get(i).index, l.get(i).value);
}

输出：

index: 3, value: -1
index: 0, value: 1
index: 2, value: 3
index: 4, value: 5
index: 1, value: 7

Answer 2

相对于value，对（value， index）对进行排序，而不是对裸值进行排序。然后，您只需获取前10对，就有10个源索引。

例如，您要排序：

6 2 7 4

制作（值、索引）对：

(6, 0) (2, 1) (7, 2) (4, 3) 

关于value的排序：

(2, 1) (4, 3) (6, 0) (7, 2) 

指数：

1 3 0 2

您使用TreeMap的方法很好。您需要进行的唯一修改如下所示：

class Pair implements Comparable<Pair> {
    int value;
    int index;

    @Override
    int compareTo(Pair other) { return value - other.value };
}

Map<Pair, String> map = new TreeMap<Pair, String>();

Answer 3

它是最小的数字，但我丢失了这个值的索引，它们具有i非排序数组

那么为什么不创建一个类来保存这些索引呢？然后，只需按值对数组进行排序，就可以得到关联的索引。

class MyClass implements Comparable<MyClass>{
    private int index;
    private int value;

    public MyClass(int i, int v){
       this.index = i;
       this.value = v;
    }

    @Override
    public String toString(){
        return "Index: "+index+" Value: "+value;
    }

    @Override
    public int compareTo(MyClass m) {
        return value - m.value;
    }
}

public static void main(String[] args){   
    MyClass[] array = new MyClass[20];

    for(int i = 0; i < array.length; i++){
       array[i] = new MyClass(i, someRandomValue); // Here I used (i*3 + 2)%5
    }
    System.out.println(Arrays.toString(array));
    Arrays.sort(array);
    MyClass [] arraySorted = Arrays.copyOfRange(array, 0, 10); //take the first ten elements
    System.out.println(Arrays.toString(arraySorted));
}

如果您想按索引对具有相同值的对象进行排序，您可以这样修改比较器：

@Override
public int compareTo(MyClass m) {
    int compareValue = value - m.value;
    if(compareValue == 0)
        return index - m.index;
    return compareValue;
}

Before :
[Index: 0 Value: 2, Index: 1 Value: 0, Index: 2 Value: 3, Index: 3 Value: 1, Index: 4 Value: 4, Index: 5 Value: 2, Index: 6 Value: 0, Index: 7 Value: 3, Index: 8 Value: 1, Index: 9 Value: 4, Index: 10 Value: 2, Index: 11 Value: 0, Index: 12 Value: 3, Index: 13 Value: 1, Index: 14 Value: 4]

After :
[Index: 1 Value: 0, Index: 6 Value: 0, Index: 11 Value: 0, Index: 3 Value: 1, Index: 8 Value: 1, Index: 13 Value: 1, Index: 0 Value: 2, Index: 5 Value: 2, Index: 10 Value: 2, Index: 2 Value: 3]