[JAVA]백준 알고리즘 10813번 : 공 바꾸기

https://www.acmicpc.net/problem/10813

10813번: 공 바꾸기

문제

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 바구니에는 공이 1개씩 들어있고, 처음에는 바구니에 적혀있는 번호와 같은 번호가 적힌 공이 들어있다.

도현이는 앞으로 M번 공을 바꾸려고 한다. 도현이는 공을 바꿀 바구니 2개를 선택하고, 두 바구니에 들어있는 공을 서로 교환한다.

공을 어떻게 바꿀지가 주어졌을 때, M번 공을 바꾼 이후에 각 바구니에 어떤 공이 들어있는지 구하는 프로그램을 작성하시오.

 

풀이

1번부터 N번까지 번호가 매겨져 있는 바구니에 해당 번호와 같은 공이 담겨 있다.

이 상황에서 두 공을 선택하여 자리를 바꾸는 과정을 여러 번 진행한 후, 최종 결과를 출력하는 문제이다.

 

바구니의 상태를 일차원 배열에 저장한 후, 이 과정을 진행하면 되는데,

자리를 바꿀 때 유의할 점이 한 가지 있다.

I번째 바구니와 J번째 바구니에 담겨 있는 공의 자리를 바꿀 때 아래와 같이 진행하면, 정답을 구할 수 없다.

basket[I] = basket[J];
basket[J] = basket[I];

 

언뜻 보면 J번째 바구니에 담겨있는 공을 I번째 바구니로, I번째 바구니에 담겨있는 공을 J번째 바구니로 넣는 것 같지만,

윗줄에서 이미 I번째 바구니에 담겨있는 공이 바뀌었고, 따라서 아랫줄의 J번째 바구니와 I번째 바구니에 담긴 공은 같은 공이 된다.

 

아래와 같이 새로운 변수에 I번째 바구니의 값을 저장해 놓는 방법을 사용하면 이를 방지할 수 있다.

매우 많이 사용하는 방법이니 알아두면 좋을 것이다.

int tmp = basket[I];
basket[I] = basket[J];
basket[J] = tmp;

 

코드

BufferedReader 클래스와 StringBuilder 클래스를 이용한 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        StringBuilder sb = new StringBuilder();

        st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());
        int[] basket = new int[N + 1];
        for(int i = 1; i <= N; i++) {
            basket[i] = i;
        }
        for(int i = 0; i < M; i++) {
            st = new StringTokenizer(br.readLine());
            int I = Integer.parseInt(st.nextToken());
            int J = Integer.parseInt(st.nextToken());

            // 공 바꾸기
            int tmp = basket[I];
            basket[I] = basket[J];
            basket[J] = tmp;
        }
        for(int i = 1; i <= N; i++) {
            sb.append(basket[i]).append(" ");
        }
        System.out.println(sb);
    }
}

 

틀린 부분이 있다면 정정해 주시면 감사하겠습니다.
궁금한 부분이 있거나, 다른 아이디어가 있으시면 자유롭게 댓글 남겨주세요!