BOJ / 1475 / 방 번호 / Python

[BOJ : 1475]

💻 code

01 lambda 이용

import sys
input = sys.stdin.readline

N = input().rstrip()
ans = 1
nums = [1 for _ in range(10)]

for n in N:
    n = int(n)
    if nums[n] > 0:
        nums[n] -= 1
    else:
        if (n == 6 and nums[9] > 0):
            nums[9] -= 1
        elif (n == 9 and nums[6] > 0):
            nums[6] -= 1
        else:
            nums = list(map(lambda x: x+1, nums))  # *
            nums[n] -= 1
            ans += 1


print(ans)

초반에 N을 입력받을 때 .rstrip()을 통해 개행문자가 함께 받아지지 않도록 주의한다.
사용 가능한 각 숫자들의 개수가 담긴 nums 리스트에 해당 숫자가 남아 있는지 체크한다.
만약 사용 가능한 숫자가 없다면 숫자 세트를 추가한다. 즉, nums내 모든 원소들의 값을 1씩 늘린다.
숫자 6과 9에 대해서는 조건문을 따로 작성했다.

02

import sys
input = sys.stdin.readline

N = input().rstrip()
cnts = [0] * 9  # 0 ~ 8

for n in N:
    n = int(n)
    if n == 9:
        n = 6
    cnts[n] += 1

cnts[6] = (cnts[6]+1) // 2

print(max(cnts))

N을 구성하는 숫자들의 개수를 카운트한다.
이때 9는 6으로 취급하여 카운트한다.

✍️👀

lambda function

익명 함수, 함수를 선언하지 않고 사용할 수 있다.
lambda 매개변수 : 표현식

def func(x, y):
    return x+y

func(1, 2)와 (lambda x, y : x+y)(1, 2)의 값은 같다.

map

map(함수, 리스트)
리스트에서 원소를 하나씩 꺼내서 함수를 적용하고 그 결과를 리스트로 반환해 준다.