[백준] python 기초알고리즘 6. 함수

BOJ 15596 - 정수 n개의 합

def solve(a):
    return sum(a)

BOJ 4673 - 셀프 넘버

a = [i for i in range(1,10000)]

def d(n):
    return sum(map(int,str(n))) + n

for j in range(1,10000):
    if d(j) in a:
        if d(j)>10000:
            break
        a.remove(d(j))
    
print('\n'.join(map(str,a)))

BOJ 1065 - 한수

n = int(input())
count = 0

# 양의정수m 대입시 한수가 맞는지를 출력
def d(m):
    # 세자리수라고 가정(배열에 3개의 숫자)
    a = list(map(int,str(m)))
    if len(a)==3:
        if a[1]-a[0]==a[2]-a[1]:
            return True
        else:
            return False
    else:
        return False
        
# n보다 작거나 같은 한수의 개수
if n<100:
    count = n
elif 100<=n<=1000:
    count = 99
    for i in range(100,n+1):
        if d(i):
            count += 1
    
print(count)

1000
144