【Python】【蓝桥杯】小明喜欢0、1、2、9

判断1~n内含有0、1、2、9的数字，并求出它们的和
样例输入 ： 40      样例输出 ： 547
样例输入 ： 2019      样例输出 ： 2658417853

print('-----方法1------')

n = int(input())
sum = 0
s = ['0','1','2','9']
for i in range(1, n+1):
    for x in s:
        if x in str(i):
            sum += i
            break
print(sum)

print('-----方法2------')

n = int(input())
sum = 0
for i in range(1,n+1):
    for j in str(i):
        if j in '2019':
            sum += i
            break
print(sum)

拓展延伸：

判断1~n内含有0、1、2、9的数字，并将求出它们的平方和  （PS：平方和是指将每个数分别平方后求和）
样例输入 ： 40       样例输出 ：14362

n = int(input())
sum = 0
for i in range(1,n+1):
    for j in str(i):
        if j in '2019':
            sum += i*i
            break
print(sum)