[Kotlin, B1] 백준 21771번 가희야 거기서 자는 거 아니야

출처: unsplash.com

 

가희야 거기서 자는 거 아니야(21771번)

Bronze 1

#구현 #문자열

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

 

문제 내용

 

 

문제 접근

가희(G)와 베개(P)가 입력될 때, 가희가 베개 위에 있으면 1, 아니면 0을 출력

베개 위에 있는 조건이 G가 P에 걸쳐 있어도 베개 위에 있다고 친다.

 

1이 출력되려면 P의 영역 안에 G가 있어야 한다.

그것을 확인하기 위해서는 P의 개수를 세면 쉽게 알아낼 수 있다.

P의 개수가 P 영역의 넓이보다 작으면 G가 P 영역 안에 있다는 것을 뜻한다.

입력 예제 1을 예로 들어 보자.

11 11
5 4 6 10
....GGGG...
....GGGG...
....GGGG...
.PPPGGGGPPP
.PPPGGGGPPP
.PPPPPPPPPP
.PPPPPPPPPP
.PPPPPPPPPP
.PPPPPPPPPP
...........
...........

베개의 영역은 총 60개(6 * 10)이다. 여기서 P의 개수는 60 - 8 = 52개이기 때문에 가희(G)가 P 영역 안에 있어서 P의 개수가 줄어든 것이다.

P의 개수가 줄어드려면 무조건 G가 포함되어 있어야 한다.

 

 

문제 해결 코드

import java.io.BufferedReader
import java.io.InputStreamReader

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {
    val (roomH, roomW) = readLine().split(" ").map { it.toInt() }
    val (dogH, dogW, pillowH, pillowW) = readLine().split(" ").map { it.toInt() }
    var pTotalCnt = 0

    for (i in 0 until roomH) {
        val roomInput = readLine()
        pTotalCnt += roomInput.filter { it == 'P' }.length
    }

    if (pTotalCnt < pillowW * pillowH) println(1)
    else println(0)
}

 

문제 풀이

베개의 넓이를 pillowH, pillowW의 곱으로 구하고, P의 개수를 입력받으면서 개수를 세준다.

P의 개수가 베개의 넓이보다 작으면 가희가 베개 위에 있다는 뜻이다.

 

 

문제 해결 과정

P의 개수만 세도 구할 수 있다는 것을 깨닫지 못해서 가희가 베개에 있을 수 있는 모든 경우를 조건으로 잡아서 판별하는 코드를 작성했었다.

 

아래에 코드를 남겨 놓겠다.

이 코드도 정답 코드이다.

import java.io.BufferedReader
import java.io.InputStreamReader

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {
    val (roomH, roomW) = readLine().split(" ").map { it.toInt() }
    val (dogH, dogW, pillowH, pillowW) = readLine().split(" ").map { it.toInt() }
    var answer = 0
    var gCnt = 0
    var pTotalCnt = 0

    for (i in 0 until roomH) {
        val roomInput = readLine()
        gCnt += roomInput.filter { it == 'G' }.length
        pTotalCnt += roomInput.filter { it == 'P' }.length
        
        if (roomInput.contains('G') && roomInput.contains('P')) {
            val pCnt = roomInput.filter { it == 'P' }.length
            if (pCnt != pillowW) answer = 1
        }
    }
    
    if (pTotalCnt < pillowW * pillowH && gCnt == dogW * dogH) answer = 1

    println(answer)
}

P의 개수를 구하는 방법만으로도 편하게 답을 구할 수 있다.