[List] 01. 최댓값 인덱스 구하기

 

---

import java.util.*;
import java.util.stream.*;

class Solution {
    public int[] solution(int[] arr) {
        int max = Arrays.stream(arr).max().getAsInt();
        int n=0;
        for(int x : arr){
        
           if(x==max) break;
           n++;
         
//         if(x!=max) n++;
//         else break;

        }
        return n;
    }
}

 

+) 세련된 풀이

import java.util.*;
import java.util.stream.*;

class Solution {
    public int[] solution(int[] arr) {
        int max = Arrays.stream(arr).max().getAsInt();
        
        return IntStream.range(0, arr.length)
            .filter(i -> arr[i] == max)
            .toArray();
    }
}