본문 바로가기

개발/코테

[JAVA] 수열과 구간 쿼리3

[문제]

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

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

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

[내 풀이]

class Solution {
    public int[] solution(int[] arr, int[][] queries) {
        for (int i =0; i<queries.length;i++){
            int tmp1 = queries[i][0];
            int tmp2 = queries[i][1];
            int n = arr[tmp1];
            arr[tmp1] = arr[tmp2];
            arr[tmp2] = n;
        }
        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;
    }
}

queries의 배열들을 for문으로 -> 길이 지정할 필요없음

- 기본 코드에서는 answer의 길이 지정이 안되어 있어 outofindex에러 발생하기 때문에 arr를 copy해주어야함

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

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