백준 1527 금민수의 개수 파이썬

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

1527번: 금민수의 개수

 

브루트 포스 냄새가 살살 나길래 permutation 으로 첨엔 접근했다.

어케 접근했냐면, 입력된 수의 길이 만큰 [4] 와 [7] 배열을 늘림. 

늘린 [4] , [7] 배열을 합침. -> [4] * len(인풋값) + [7] * len(인풋값) 

합쳐진 배열을 permutations 에 넣고 경우의 수를 구함. 당연히 시간 초과남...

 

불필요하게 4 와 7 배열을 늘려놓은게 시간초과의 원인이었다. 

이럴 땐 product 를 사용해보자 (다른 분 답 참고함 🥹)

product 의 repeat 속성을 이용하면 입력값의 길이만큼 경우의 수를 만들어 낼 수 있고, 시간초과도 나지 않는다.

그 후엔 기존에 짠 로직이랑 비슷해서 통과할 수 있었다.

from itertools import product

a, b = map(int, input().split())
a_len = len(str(a))
b_len = len(str(b))
count = 0
for i in range(a_len, b_len + 1):
    for n in product([4, 7], repeat=i):
        join = ''.join(list(map(str, n)))
        if a <= int(join) <= b:
            count += 1
print(count)