백준 알고리즘 18856번 피드백(python)

단순 구현 문제이다. 다양한 방법으로 풀 수 있다. 첫 번째 코드가 제일 깔끔한 것 같다.

N = int(input())
res = [1, 2] + [i for i in range(3, 3+N-3)] + [997]
print(N)
print(*res)

N = int(input())
li = [1]*230
for i in range(2, int(230**0.5)+1):
    if li[i] == 1:
        for j in range(i+i, 230, i):
            li[j] = 0
prime = [i for i in range(2, 230) if li[i] == 1]
res = [1] + prime[:N-1]
print(N)
print(*res)