获取数组中最大数量的索引

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

一种方法是简单地沿阵列进行单次传递并跟踪最高数量的所有索引.如果当前条目小于目前为止看到的最高数字,那么no-op.如果当前条目与看到的最大数字相同,则添加该索引.否则,我们已经看到了一个新的最高数字,我们应该抛弃我们最旧的最高数字列表并开始一个新的数字.

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];
    }
}