본문 바로가기
코딩테스트 연습

수열과 구간 쿼리 3

by 신재은👩🏼‍💻 2024. 4. 17.

문제 설명
정수 배열 arr와 2차원 정수 배열 queries이 주어집니다. queries의 원소는 각각 하나의 query를 나타내며, [i, j] 꼴입니다.

각 query마다 순서대로 arr[i]의 값과 arr[j]의 값을 서로 바꿉니다.

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

제한사항
1 ≤ arr의 길이 ≤ 1,000
0 ≤ arr의 원소 ≤ 1,000,000
1 ≤ queries의 길이 ≤ 1,000
0 ≤ i < j < arr의 길이


입출력 예
arr queries result
[0, 1, 2, 3, 4] [[0, 3],[1, 2],[1, 4]] [3, 4, 1, 0, 2]


class Solution {
    public int[] solution(int[] arr, int[][] queries) {
        for(int i=0; i<queries.length; i++) {
            int a = queries[i][0];
            int b = queries[i][1];
            int temp = arr[a];
            arr[a] = arr[b];
            arr[b] = temp;
        }
        return arr;
    }
}


좋아요가 가장 많은 풀이를 봐도

 

import java.util.Arrays;

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

        for (int[] query : queries) {
            int i = query[0];
            int j = query[1];

            int temp = answer[i];
            answer[i] = answer[j];
            answer[j] = temp;
        }

        return answer;
    }
}

 

int[] answer에 맞추기 위해 copyOf 쓴 거, for each 돌린 거 외에는 별 특이 사항이 없네.

평이한 문제라고 생각해도 되겠다.

'코딩테스트 연습' 카테고리의 다른 글

수열과 구간 쿼리 4  (0) 2024.04.18
수열과 구간 쿼리 2  (0) 2024.04.17
수 조작하기 2  (0) 2024.04.17
주사위 게임 2  (0) 2024.04.16
등차수열의 특정한 항만 더하기  (0) 2024.04.16