我有一个数组,其中包含排名.
像这样的东西：
0 4 2 0 1 0 4 2 0 4 0 2
这里0对应于最低等级,最大数对应于最高等级.可能有多个索引包含最高排名.
我想找到数组中所有最高等级的索引.我用以下代码实现了：
import java.util.*;
class Index{
public static void main(String[] args){
int[] data = {0,4,2,1,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
还有比这更好的方法吗？
因为,在我的情况下,可能有一个包含数百万个元素的数组,因此我对性能有一些问题.
所以,
任何建议表示赞赏.
解决方法
一种方法是简单地沿阵列进行单次传递并跟踪最高数量的所有索引.如果当前条目小于目前为止看到的最高数字,那么no-op.如果当前条目与看到的最大数字相同,则添加该索引.否则,我们已经看到了一个新的最高数字,我们应该抛弃我们最旧的最高数字列表并开始一个新的数字.
int[] data = {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];
}
}
