백준 단계별로 풀어보기 - 함수 (Python)

처음부터 다시 풀어보는 것의 첫 시작입니다.
이 게시물은 제가 문제를 풀 때 마다 업데이트 할 예정입니다.

정수 N개의 합 - 15596번

def solve(a: list):
    return sum(a)

셀프 넘버 - 4673번

all_num = set(range(1, 10001))
generated_num = set()

for i in all_num:
    for j in str(i):
        i += int(j)
    generated_num.add(i)

self_num = sorted(all_num - generated_num)

for i in self_num:
    print(i)

한수 - 1065번

n = int(input())
answer = n

def is_arithmetic_sequence(n):
    num = str(n)
    
    if int(num[1]) * 2 == int(num[0]) + int(num[2]):
        return 1
        
    return 0
            
if n >= 100:
    answer = 99
    for i in range(100, n + 1):
        answer += is_arithmetic_sequence(i)

print(answer)