본문 바로가기

개발/코테

[JAVA] 수열과 구간 쿼리 4

[문제]

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

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

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

[내 풀이]

class Solution {
    public int[] solution(int[] arr, int[][] queries) {
        for( int[] query:queries){
            for(int i = query[0]; i< query[1]+1; i++){
                if(i % query[2] == 0){
                    arr[i] +=1;
                }
            }
        }
        return arr;
    }
}

 

'개발 > 코테' 카테고리의 다른 글

[JAVA] 수열과 구간 쿼리2  (0) 2024.04.12
[JAVA] 수열과 구간 쿼리3  (1) 2024.04.12
[JAVA] 수 조작하기 2  (1) 2024.04.12
[JAVA] 수 조작하기1  (0) 2024.04.11
[JAVA] 마지막 두 원소  (0) 2024.04.11