[Java] 백준(브론즈-2) 10813번 - 공 바꾸기

문제 풀이

---

공 바꾸기 문제는 배열의 원소를 교체하는 메커니즘을 이용하면 쉽게 풀 수 있다.
이번 문제는 배열 내에서 2개의 원소를 교체하는 방식을 구현하면 된다.

아이디어 도출

N개의 바구니에 M번만큼 공을 바꾸게 되는데, 바구니에는 이미 바구니와 같은 번호의 공이 들어가 있음을 윺의해야 한다.
즉, 배열의 초기값을 [인덱스+1] 값으로 고정시켜놓아야 한다는 것이다.
그리고 배열의 원소를 교체하면 되는데 i-1번째 값과 j-1번째 값을 변경하면 된다.

• N개의 길이를 가지는 배열을 만들고 인덱스+1 값으로 초기화한다.
  • 예를 들어 N이 5라고 하면 배열은 [1,2,3,4,5] 로 초기화되어야 한다.
• M번 횟수만큼 배열의 원소를 교체(swap)한다.
  • 배열의 (i-1)번째 값과 (j-1)번째 값을 변경한다.

생각해낸 아이디어는 간단하다. 이제 코드를 작성해보자.

int N = Integer.parseInt(st.nextToken());
int M = Integer.parseInt(st.nextToken());
int[] arr = new int[N];

for(int i=0; i<arr.length; i++) {
    arr[i] = i+1;
}

N과 M을 공백 기준으로 입력받게되니 StringTokenizer를 이용해 N과 M을 저장한다.
그리고 N개의 크기를 가지는 int 배열을 만들고 [인덱스+1] 값을 설정하여 해당 배열을 초기화한다.

int temp = 0;
for(int idx=0; idx<M; idx++) {
    st = new StringTokenizer(br.readLine());
    int i = Integer.parseInt(st.nextToken()) - 1;
    int j = Integer.parseInt(st.nextToken()) - 1;
    temp = arr[i];
    arr[i] = arr[j];
    arr[j] = temp;
    temp = 0;
}
for(int ball : arr) {
    bw.write(ball + " ");
}

다음으로 배열의 원소를 교체하기 위한 임시변수 temp를 하나 생성해두고 i와 j를 공백 기준으로 입력받아 저장한 뒤
배열의 (i-1)번째 값과 (j-1)번째 값에 대한 교체(swap) 작업을 진행하면 된다.

작성코드

---

import java.io.*;
import java.util.*;

class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());
        int[] arr = new int[N];

        for(int i=0; i<arr.length; i++) {
            arr[i] = i+1;
        }
        
        int temp = 0;
        for(int idx=0; idx<M; idx++) {
            st = new StringTokenizer(br.readLine());
            int i = Integer.parseInt(st.nextToken()) - 1;
            int j = Integer.parseInt(st.nextToken()) - 1;
            temp = arr[i];
            arr[i] = arr[j];
            arr[j] = temp;
            temp = 0;
        }
        for(int ball : arr) {
            bw.write(ball + " ");
        }
        
        bw.flush();
        bw.close();
        br.close();
    }
}

회고

• 배열의 값에 대해 Swap(교체) 작업을 하기 위한 임시변수를 만들어 활용할 수 있었다.

출처

• 해당 문제의 저작권은 문제를 만든이에게 있으며 자세한 내용은 문제 링크에서 참조바랍니다.