[Kotlin, B1] 백준 16433번 주디와 당근농장

출처: unsplash.com

 

주디와 당근농장(16433번)

Bronze 1

#구현

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

 

문제 내용

 

 

문제 접근

정사각형으로 이루어진 밭의 넓이인 n과 당근이 심어져 있는 위치 r, c가 순서대로 주어진다.

당근을 밭에 심어야 하는데 조건이 있다.

당근이 심어져 있는 위치에 위, 아래, 오른쪽, 왼쪽에 다른 당근이 심어져 있으면 안 된다.

예제 1을 예로 들면 다음과 같이 당근이 심어져야 한다.

사진 1-1

위 사진과 같이 기존에 심어져 있던 당근을 기준으로 지그재그로 당근을 심어야 한다.

첫 당근의 위치에 맞게 다른 당근의 위치를 구하려면 조건이 필요하다.

 

<조건 1>

우선 r부터 봐보자. 지그재그가 되려면, 현재 당근이 심어져 있는 위치와 심을 당근의 거리 차이가 짝수면 똑같은 x위치에 당근을 심어도 된다. 이를 코드로 나타내 보면 다음과 같다.

if(abs(r - i) % 2 == 0)

첫 당근의 위치인 r로부터 거리가 짝수만큼 떨어져 있으면 동일한 c에 당근을 심어야 한다.

첫 당근의 위치가 r로부터 홀수만큼 떨어져 있으면 동일한 c에 당근을 심을 수 없다.

 

<조건 2>

이제 c를 살펴보자. 조건 1을 만족하는 경우는 현재 심어진 당근의 r과 심어질 당근의 r 거리의 차이가 짝수라는 뜻이기 때문에

첫 당근이 심어진 c위치에서 짝수거리만큼 떨어진 위치에 당근을 심어야 한다.

if(abs(c - j % 2 == 0)

이 조건을 만족하면 c와의 거리가 짝수이고 r과의 거리가 짝수인 위치에 당근을 모두 심을  수 있다.

 

<조건 3>

조건 1이 만족하지 못하는 경우에 사용한다.

조건 1이 만족하지 않는 경우는 r과 현재 심을 위치의 y거리 차이가 홀수이기 때문에 당근을 심을 수 없게 된다.

가로의 경우에도 c와 현재 심을 위치가 홀수인 상태여야 당근을 심을 수 있을 것이다.

if(abs(c - j) % 2 != 0)

이 조건을 만족하면서 조건 1을 불만족하는 경우에만 당근을 심어야 한다.

 

 

문제 해결 코드

import java.io.BufferedReader
import java.io.InputStreamReader
import kotlin.math.abs

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {
    val (n, r, c) = readLine().split(" ").map { it.toInt() }
    val map = Array(n + 1) { Array(n + 1) { "." } }
    val res = StringBuilder()

    map[r][c] = "v"

    for (i in 1 until map.size) {
        for (j in 1 until map[i].size) {
            if (abs(r - i) % 2 == 0) {
                if (abs(c - j) % 2 == 0) map[i][j] = "v"
            } else {
                if (abs(c - j) % 2 != 0) map[i][j] = "v"
            }
        }

        res.append("${map[i].filterIndexed { index, _ -> index != 0 }.joinToString("")}\n")
    }

    println(res)
}

 

문제 풀이

밭을 2차원 배열로 설정한다 size는 n + 1로 설정한다. 문제에서 왼쪽 위의 값을 1, 1로 가정하고 있기 때문에 index가 0인 값들을 이용하지 않기 위해서이다.

초기 당근이 심어진 위치를 v로 설정한다.

 

2중 for문을 돌면서 조건 1을 만족하는지 확인한다.

조건 1을 만족한다면 조건 2를 확인해서 조건 2도 만족하는 위치에 당근(v)을 심어주고,

조건 1을 만족하지 않는다면 조건 3을 확인해 조건 3을 만족하는 위치에 당근(v)을 심는다.

 

조건을 모두 확인해 각 줄에 적절한 위치에 당근을 모두 심었으면 당근을 심은 줄을 출력한다. 0번째 index는 제거하고 출력한다.

출력 형식을 따르는 것을 잊지 말자.

 

 

문제 해결 과정

심어진 위치를 이용해 나머지 당근이 심어질 위치를 모두 확인할 수 있다.

심어진 위치와 심을 위치의 홀수 짝수 거리를 적절히 활용해 지그재그 모양으로 당근을 심어야 한다.

아이디어 생각은 어렵진 않지만, 구현하는 데 조금 시간이 걸렸다.