[Python] 백준 28214번 - 크림빵

문제

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

 

---

n, k, p에 해당하는 세 개의 정수 입력.

n x k개의 정수를 입력받고, 리스트에 저장.

개수를 세기 위한 변수 생성(0으로 초기화).

n, k, p = map(int, input().split())
arr = list(map(int, input().split()))
count = 0

 

0부터 n*k-1까지 반복 (k씩 증가).

- 상태변수 생성(True: 팔 수 있음)

- 크림이 들어있지 않는 빵의 수를 세기 위한 변수 생성.

- i부터 i+k전까지 반복.

-- 만일 해당 번째의 값이 0이면 변수 값 증가. 해당 변수의 값이 p와 같아지면 팔 수 없으므로 False로 상태 변환 후 반복 종료.

- 상태변수가 True를 유지하고 있다면 팔 수 있으므로 개수 증가.

결과 출력.

for i in range(0, n*k, k):
    check = True
    fail = 0
    for j in range(i, i+k):
        if arr[j] == 0:
            fail += 1
            if fail == p:
                check = False
                break
    if check:
        count += 1
print(count)

 

전체 코드.

n, k, p = map(int, input().split())
arr = list(map(int, input().split()))
count = 0
for i in range(0, n*k, k):
    check = True
    fail = 0
    for j in range(i, i+k):
        if arr[j] == 0:
            fail += 1
            if fail == p:
                check = False
                break
    if check:
        count += 1
print(count)