프로그래머스 181924번 수열과 구간 쿼리 3 - Python, Java

문제 설명

프로그래머스 181924번 수열과 구간 쿼리 3

정수 배열 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의 길이

입출력 예시

예시 1

입출력 1

arr	queries	result
[0, 1, 2, 3, 4]	[[0, 3],[1, 2],[1, 4]]	[3, 4, 1, 0, 2]

입출력 예 설명 1

각 쿼리에 따라 arr가 다음과 같이 변합니다.

arr
[0, 1, 2, 3, 4]
[3, 1, 2, 0, 4]
[3, 2, 1, 0, 4]
[3, 4, 1, 0, 2]

코드 구현

Python

def solution(arr, queries):
    
    for query in queries:
        i = arr[query[0]]
        j = arr[query[1]]
        
        arr[query[0]] = j
        arr[query[1]] = i
    
    return arr

Java

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

        for (int[] query : queries) {
            int temp = arr[query[0]];
            answer[query[0]] = arr[query[1]];
            answer[query[1]] = temp;
        }

        return answer;
    }
}