[프로그래머스(Java)] 수열과 구간 쿼리 2 / Arrays.fill()

 

Level. 0

 

문제

정수 배열 arr와 2차원 정수 배열 queries이 주어집니다. 
queries의 원소는 각각 하나의 query를 나타내며, [s, e, k] 꼴입니다.
각 query마다 순서대로 s ≤ i ≤ e인 모든 i에 대해 k보다 크면서 가장 작은 arr[i]를 찾습니다.
각 쿼리의 순서에 맞게 답을 저장한 배열을 반환하는 solution 함수를 완성해 주세요.
단, 특정 쿼리의 답이 존재하지 않으면 -1을 저장합니다.

* 제한사항
- 1 ≤ arr의 길이 ≤ 1,000
   0 ≤ arr의 원소 ≤ 1,000,000
- 1 ≤ queries의 길이 ≤ 1,000
   0 ≤ s ≤ e < arr의 길이
   0 ≤ k ≤ 1,000,000

 

풀이

import java.util.*;

class Solution {
    public int[] solution(int[] arr, int[][] queries) {
        int[] answer = new int[queries.length];
        int s, e, k;
        for(int i=0; i<queries.length; i++){
            s = queries[i][0];
            e = queries[i][1];
            k = queries[i][2];
            
            int[] arr2 = Arrays.copyOfRange(arr,s, e+1);
            Arrays.sort(arr2);
            answer[i] = -1;
            for(int j=0; j<arr2.length; j++){
                if(k < arr2[j]){
                    answer[i] = arr2[j];
                    break;
                }
            }
        }
        
        return answer;
    }
}

 

 

---

다른 풀이

import java.util.Arrays;

class Solution {
    public int[] solution(int[] arr, int[][] queries) {

        int[] answer = new int[queries.length];
        Arrays.fill(answer, -1);

        for (int idx = 0; idx < queries.length; idx++) {
            int[] query = queries[idx];
            int s = query[0], e = query[1], k = query[2];

            for (int i = s; i <= e; i++) {
                if (k < arr[i]) {
                    answer[idx] = answer[idx] == -1 ? arr[i] : Math.min(answer[idx], arr[i]);
                }
            }
        }
        return answer;
    }
}

 

Arrays.fill()

배열의 모든 값을 지정한 값으로 초기화시킨다. 

// 1차원 배열 초기화
int arr[] = new int[10];
Arrays.fill(arr, 1);

// 2차원 배열 초기화
int arr2[][] = new int[3][2];
for(int i=0; i<arr2.length; i++){
	Arrays.fill(arr2[i], 1);
}

 

 

---

https://school.programmers.co.kr/learn/courses/30/lessons/181923#

프로그래머스