[ Algorithm ] Baekjoon_10813 : Kotlin

안녕하세요 공공돌🧸 입니다 !!

오늘은 백준 알고리즘의 10813 : 공 바꾸기 를 풀어보고 리뷰를 해보려 합니다.

먼저 문제를 풀기 전 저의 목표는 알고리즘 문제 풀이도 있지만, 기능들을 나누어 구현하는것을 목표로 삼았습니다.

 

문제는 아래 링크를 참고해주세요.

10813번: 공 바꾸기

---

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)

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

출력

1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다.

예제 입력 1

5 4
1 2
3 4
1 4
2 2

예제 출력 1

3 1 4 2 5

---

알고리즘 분류

• 구현
• 시뮬레이션

---

기능 정리

✅ 바구니 n과 반복횟수 m을 입력 받는다.

✅ 1번부터 n번까지의 바구니를 생성한다.

✅ 정수 i와 j를 입력 받는다.

✅ i번 바구니와 j번 바구니의 공을 교환한다.

✅ 교환 과정을 m번 반복한다.

✅ 1번부터 n번 바구니에 들어있는 공의 번호를 공백으로 구분하여 출력한다. 

---

핵심

해당문제는 1차원 배열의 문제입니다.

 

1번부터 5번까지의 바구니에 1부터 5까지의 숫자가 들어있습니다.

이때, i와 j의 입력값이 1과 3이라고 가정하면 아래와 같이 temp변수를 활용하여 아래와 같이 변경할 수 있습니다.

 

이와 같은 방식으로 m 번을 반복하게 구현하여 1번부터 n번까지의 바구니를 출력하면 됩니다.

---

코드 구현

main

fun main() {
    val (n, m) = readln().split(" ").map { it.toInt() }
    val initialBasketList = generateNumberList(n)
    val finalBasketList = executeSwaps(initialBasketList, m)
    println(finalBasketList.joinToString(" "))
}

 

바구니의 개수 n과 교환할 반복횟수 m을 입력받습니다.

생성된 바구니 리스트를 변수에 저장합니다.

교환이 끝난 바구니 리스트를 변수에 저장합니다.

교환이 끝난 바구니 리스트를 공백으로 구분해 출력합니다.

---

generateNumberList

private fun generateNumberList(n: Int): List<Int> {
    return (1..n).toList()
}

 

입력받은 정수 n만큼 1 ~ n 까지의 바구니 리스트를 생성합니다.

 

---

executeSwaps

private fun executeSwaps(basketList: List<Int>, m: Int): List<Int> {
    val currentBasketList = basketList.toMutableList()
    repeat(m) {
        val (i, j) = readln().split(" ").map { it.toInt() }
        swapBaskets(currentBasketList, i - 1, j - 1)
    }
    return currentBasketList
}

 

i와 j를 입력 받습니다.

공을 교환해줍니다.

입력받은 정수 m반복합니다.

교환이 끝난 바구니 리스트를 리턴합니다.

---

swapBaskets

private fun swapBaskets(basketList: MutableList<Int>, i: Int, j: Int) {
    val temp = basketList[i]
    basketList[i] = basketList[j]
    basketList[j] = temp
}

 

i번 바구니의 공과 j번 바구니의 공을 교환합니다.

---

최종 코드 구현

fun main() {
    val (n, m) = readln().split(" ").map { it.toInt() }
    val initialBasketList = generateNumberList(n)
    val finalBasketList = executeSwaps(initialBasketList, m)
    println(finalBasketList.joinToString(" "))
}

private fun generateNumberList(n: Int): List<Int> {
    return (1..n).toList()
}

private fun executeSwaps(basketList: List<Int>, m: Int): List<Int> {
    val currentBasketList = basketList.toMutableList()
    repeat(m) {
        val (i, j) = readln().split(" ").map { it.toInt() }
        swapBaskets(currentBasketList, i - 1, j - 1)
    }
    return currentBasketList
}

private fun swapBaskets(basketList: MutableList<Int>, i: Int, j: Int) {
    val temp = basketList[i]
    basketList[i] = basketList[j]
    basketList[j] = temp
}

---

결과

---

공부하는 공돌이, 공공돌입니다🐻

@sheep1sik