[BOJ] 24089 : ボールの移動 (Moving Balls)

問題

$N$個のボールがあり、１から$N$までの番号が付けられている。
また、何個でもボールを入れることのできる$N$個の箱があり、箱には１から$N$までの番号が付けられている。

箱$i$（$1≦i≦N$）には最初、ボール$i$が入っていた。

JOI高校の生徒である葵は、この状態から箱とボールに対して$M$回の操作を行なった。

$j$回目の（$1≦j≦M$）の操作は、次のように行われた。

ボール$X_j$が入っている箱を探し、その箱からボール$X_j$を取り出す。その後、箱$Y_j$にボール $X_j$を入れる。

葵が$M$回の操作をすべて終えた後、$N$個のボールがそれぞれどの箱に入っているかを求めよ。

INPUT ＆ OUTPUT

INPUT ＆ OUTPUTの例

コード作成

Java

import java.util.Scanner;

public class Main {

    public static void main(String[] args) {

        // BOJ24089
        
        Scanner sc = new Scanner(System.in);
        
        int n = sc.nextInt();                   // 공의 개수
        int m = sc.nextInt();                   // 조작 횟수
        
        int[] box = new int [n];                // 박스를 공의 개수만큼 만들어줌

        for (int i = 0; i < box.length; i++) {
            box[i] = i+1;                       // 조작하기 전 조건에 맞게 공 초기화
        }

        for (int i = 0; i < m; i++) {
            int x = sc.nextInt();               // 꺼낼 공
            int y = sc.nextInt();               // 옮길 상자
            box[x-1] = y;
        }

        for (int ball : box) {
            System.out.println(ball);           // 개행을 두고 공을 순서대로 출력
        }

    }
}

Kotlin

import java.util.Scanner

fun main(args: Array<String>) = with(Scanner(System.`in`)) {

    val n:Int = nextInt()    // 공의 개수
    val m:Int = nextInt()    // 조작 횟수

    var index = 1
    val box:Array<Int> = Array(n) { index++ }


    for(i in 0 until m) {
        val x:Int = nextInt()
        val y:Int = nextInt()
        box[x-1] = y
    }

    for(ball in box) {
        println(ball)
    }

}