k의 개수.python

문제 설명

1부터 13까지의 수에서, 1은 1, 10, 11, 12, 13 이렇게 총 6번 등장합니다. 정수 i, j, k가 매개변수로 주어질 때, i부터 j까지 k가 몇 번 등장하는지 return 하도록 solution 함수를 완성해주세요.

제한사항
1 ≤ i < j ≤ 100,000
0 ≤ k ≤ 9

입출력 예
i	j	k	result
1	13	1	6
10	50	5	5
3	10	2	0

풀이 :

def solution(i, j, k):
    count = 0
    for num in range(i,j+1) :
        for i in str(num):
            if i in str(k) :
                count += 1

    return count

갈결하게 만들고 싶어서 시도해 보았다.

def solution(i, j, k):
	return sum([str(num).count(str(k)) for num in range(i,j+1)])
# count,sum 함수 사용