29700번: 우당탕탕 영화예매

문제 출처 : https://www.acmicpc.net/problem/29700

 

언어 : Kotlin

 

문제 설명 :

도은이는 동아리 문화의 날을 맞이하여 동아리원들과 함께 좌석이 M행 N열의 직사각형 모양으로 배치되어 있는 영화관에서 영화를 보기로 했다. 도은이는 동아리원의 유대감을 중요하게 생각하기 때문에 이미 예매가 완료된 좌석을 피해 동아리원들이 모두 가로로 이어서 앉을 수 있도록 자리를 예매하고 싶어 한다. 도은이를 도와 모든 동아리원들이 가로로 이어서 앉을 수 있도록 예매하는 경우의 수는 총 몇 가지가 있을지 구해보자. 단, 예매한 좌석은 동일하지만, 각 사람이 앉는 위치만 바뀌는 경우는 한 가지로 본다.

 

입력 :

첫째 줄에 영화관 세로줄의 개수 N(1 ≤ N ≤ 1000)과 가로줄의 개수 M(1 ≤ M ≤ 5000), 영화를 관람할 동아리원의 수 K(1 ≤ K ≤ 10)가 주어진다.

 

둘째 줄부터 N 개의 줄에 걸쳐 그중 i번째 줄에는 i번째 열의 좌석 예매 현황이 길이 M인 문자열 s_i로 주어진다. s_i의 j번째 문자는 i열 j번째 좌석의 예매 현황을 나타내는데, 이 문자가 '0'이라면 예매 가능한 빈 좌석을, 이 문자가 '1'이라면 예매가 완료되어 예매가 불가능한 좌석을 나타낸다.

 

출력 :

동아리원들이 모두 가로로 이어서 앉을 수 있도록 영화를 예매하는 경우의 수를 출력한다. 단, 문제에서 주어진 조건에 맞게 영화를 예매할 수 있는 방법이 없다면 0을 출력한다.

 

제한 사항 :

• 시간 제한 : 1초
• 메모리 제한 : 1024MB

 

입출력 예 :

입력	출력
3 5 3 11000 01010 10000	3
3 4 3 1101 1001 0101	0

 

풀이 : 

import java.io.BufferedWriter
import java.io.OutputStreamWriter

fun main() = with(System.`in`.bufferedReader()) {
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val (n, m, k) = readLine().split(" ").map { it.toInt() }
    val list = MutableList(n) { readLine() }

    var cnt = 0
    for (i in list.indices) {
        for (j in 0 .. list[i].length - k) {
            if (list[i][j] == '0') {
                val cur = list[i].substring(j until j + k)
                if (!cur.contains('1')) cnt++
            }
        }
    }

    bw.write("$cnt")
    bw.flush()
    bw.close()
}