[BOJ#29700] 우당탕탕 영화예매(Python)

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

1 초

1024 MB

690

439

378

66.549%

문제

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

입력

 

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

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

출력

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

---

제출 코드

N, M, K = map(int, input().split())
arr = [input() for _ in range(N)]

result = 0
for i in range(N):
    st = 0
    for j in range(M):
        if arr[i][j] == '0':
            st += 1
            if st >= K:
                result += 1
        elif arr[i][j] == '1':
            st = 0

print(result)

 

구현으로 풀었고 st에 부원들을 채워서 숫자가 같아질때부터 1씩 더해주고,

자리가 끊겼다면 st을 0으로 초기화 해주었다.