[BOJ / 백준] 1475번 방번호 파이썬 / python 리스트 인덱스

출처 : https://www.acmicpc.net/problem/1475

1475번: 방 번호

---

plastic 리스트에 0부터 9까지 개수를 표시해주고

새로운 세트가 추가 될 때마다 cnt에 1 더해주고,

plastic리스트의 각 인덱스에 해당하는 숫자에도 1씩 더해줬다.

 

그리고 6이나 9일때를 나눠서 처리해줬다.

# BOJ 1475 방번호
N = input()
cnt = 1
plastic = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
for i in N:
    i = int(i)
    if i != 6 and i != 9:
        if plastic[i] != 0:
            plastic[i] -= 1
        else:
            cnt += 1
            for j in range(10):
                plastic[j] += 1
            plastic[i] = 0
    else: # 6이나 9일때
        if plastic[6] != 0:
            plastic[6] -= 1
        elif plastic[9] != 0:
            plastic[9] -= 1
        elif plastic[6] == 0 and plastic[9] == 0:
            cnt += 1
            for k in range(10):
                plastic[k] += 1
            plastic[6] = 0
print(cnt)

 

ㅇ