java - 获取数组中最大数的索引

Question

我有一个包含排名数字的数组。

像这样:

0 4 2 0 1 0 4 2 0 4 0 2

此处 0 对应最低等级，max 数字对应最高等级。可能有多个包含最高排名的索引。

我想找到数组中所有排名最高的索引。我已经通过以下代码实现了:

import java.util.*;

class Index{

    public static void main(String[] args){

        int[] data = {0,4,2,0,1,0,4,2,0,4,0,2};
        int max = Arrays.stream(data).max().getAsInt();
        ArrayList<Integer> indexes = new ArrayList<Integer>();

        for(int i=0;i<12;i++){
            if(data[i]==max){
               indexes.add(i);
            }
        }

        for(int j=0;j<indexes.size();j++){
            System.out.print(indexes.get(j)+" ");   
        }
        System.out.println();
    }
}

我得到的结果是:1 6 9

还有比这更好的方法吗？

因为，在我的例子中，可能有一个包含数百万个元素的数组，因此我在性能方面遇到了一些问题。

所以，

如有任何建议，我们将不胜感激。

Answer 1

一种方法是简单地沿着数组进行单次传递，并跟踪所有最高数字的索引。如果当前条目小于到目前为止看到的最高数字，则不执行任何操作。如果当前条目与看到的最高数字相同，则添加该索引。否则，我们已经看到了一个新的最高数字，我们应该扔掉旧的最高数字列表并开始一个新的。

int[] data = {0,4,2,0,1,0,4,2,0,4,0,2};
int max = Integer.MIN_VALUE;
List<Integer> vals = new ArrayList<>();

for (int i=0; i < data.length; ++i) {
    if (data[i] == max) {
        vals.add(i);
    }
    else if (data[i] > max) {
        vals.clear();
        vals.add(i);
        max = data[i];
    }
}