[JAVA] 백준 10813번 공 바꾸기

문제

 

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

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

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

 

 

입력

 

첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.

둘째 줄부터 M개의 줄에 걸쳐서 공을 교환할 방법이 주어진다. 각 방법은 두 정수 i j로 이루어져 있으며, i번 바구니와 j번 바구니에 들어있는 공을 교환한다는 뜻이다. (1 ≤ i ≤ j ≤ N)

도현이는 입력으로 주어진 순서대로 공을 교환한다.

 

 

풀이

import java.util.Scanner;

public class Main {
    public static void main(String[] args){
        Scanner scanner = new Scanner(System.in);
        int x = scanner.nextInt();
        int y = scanner.nextInt();

        int[] bag = new int[x];

        // bag 안에 차례대로 숫자 넣기 -> {1,2,3,4,5}
        for (int i = 0; i < bag.length; i++) {
            bag[i] = i+1;
        }


        int temp = 0;
        for (int i = 0; i < y; i++) {
            int a = scanner.nextInt();
            int b = scanner.nextInt();
            temp = bag[a-1];
            bag[a-1] = bag[b-1];
            bag[b-1] = temp;
            }

        // 배열 출력
        String res = "";
        for (int i = 0; i < bag.length; i++) {
            res += bag[i] + " ";
        }

        System.out.println(res);
    }
}

 

 

1.

int[] bag = new int[x];

// bag 안에 차례대로 숫자 넣기 -> {1,2,3,4,5}
for (int i = 0; i < bag.length; i++) {
    bag[i] = i+1;
}

 

배열 bag 안에 1번부터 x번까지 번호가 매겨져있다고 했으니 for문으로 숫자를 넣어준다.

 

2.

int temp = 0;
for (int i = 0; i < y; i++) {
    int a = scanner.nextInt();
    int b = scanner.nextInt();
    temp = bag[a-1];
    bag[a-1] = bag[b-1];
    bag[b-1] = temp;
    }

 

입력받은 바구니 번호 a 와 b 안에 담겨있는 공의 번호를 바꿔치기 해야하기 때문에

임시로 숫자를 담을 temp 를 선언해 초기화 해준다.

temp에 a의 숫자를 담고,

a 안에 b의 숫자를 바꿔 넣어준 뒤,

b 안에 temp에 담았던 a의 숫자를 넣어준다.

 

 

3.

// 배열 출력
String res = "";
for (int i = 0; i < bag.length; i++) {
    res += bag[i] + " ";
}

System.out.println(res);

 

for문으로 배열을 출력해준다.