[프로그래머스 - JAVA] 수열과 구간 쿼리 4

문제 설명

 

정수 배열 arr와 2차원 정수 배열 queries이 주어집니다. queries의 원소는 각각 하나의 query를 나타내며, [s, e, k] 꼴입니다.

각 query마다 순서대로 s ≤ i ≤ e인 모든 i에 대해 i가 k의 배수이면 arr[i]에 1을 더합니다.

위 규칙에 따라 queries를 처리한 이후의 arr를 return 하는 solution 함수를 완성해 주세요.

 

제한사항

• 1 ≤ arr의 길이 ≤ 1,000
  • 0 ≤ arr의 원소 ≤ 1,000,000
• 1 ≤ queries의 길이 ≤ 1,000
  • 0 ≤ s ≤ e < arr의 길이
  • 0 ≤ k ≤ 5

 

결과적으로는 어렵지 않았으나 문제자체가 헷갈리게 설명을 해준다. 문제에서 요구하는 사항은 arr[s] ~ arr[e] 인덱스 중 k 값의 배수를 구하는 문제이다 처음에는 각각의 인덱스 내부에 있는 값이 k의 배수인 경우인줄 알았으나 아니였다... 

 

arr 인덱스 값을 k로 나눴을 때 0인경우를 찾아 해당 인덱스 값이 1씩 더해주면 된다.

 

오답

import java.util.Arrays;

class Solution {
    public int[] solution(int[] arr, int[][] queries) {
        int[] answer = Arrays.copyOf(arr, arr.length);

        for (int i = 0; i < queries.length; i++) {
            int k = queries[i][2];

            for (int j = queries[i][0]; j <= queries[i][1]; j++) {
                if (arr[j] % k == 0) {
                    answer[j]++;
                }
            }
        }

        return answer;
    }
}

 

해당 풀이를 가지고 거의 1시간을 고민했다 문제 요구 조건을 다 맞춘거 같은데 왜 틀릴까... 위에서 말한 것처럼 인덱스 값이 아닌 인덱스에 나머지를 구해줘야한다

 

정답

import java.util.Arrays;

class Solution {
    public int[] solution(int[] arr, int[][] queries) {
        int[] answer = Arrays.copyOf(arr, arr.length);

        for (int i = 0; i < queries.length; i++) {
            int k = queries[i][2];

            for (int j = queries[i][0]; j <= queries[i][1]; j++) {
                if (j % k == 0) {
                    answer[j]++;
                }
            }
        }

        return answer;
    }
}

 

수정된 코드는 단 하나이다 arr[j] -> j