[백준/파이썬] 17206번 준석이의 수학 숙제

-Code

n = int(input())
nums = list(map(int, input().split()))
arr = []
res = 0
for i in range(80001):
    if i % 3 == 0 or i % 7 == 0:
        res += i
    arr.append(res)
for num in nums:
    print(arr[num])

 

처음에 일일이 하는 방법을 했으나 틀려서 저장을 모든 수를 해놓고 뽑아서 사용하는 방식으로 해결을 했습니다.