[python] SWEA - 3408. 세가지 합 구하기

🤔문제 해결

• D3 | 수학

💨 1부터 N 까지 숫자의 합: N * (N + 1) / 2

💨 1부터 N 개의 홀수의 합: N ** 2

💨 1부터 N 개의 짝수의 합: N * (N + 1)

 

💻소스 코드

# 인풋
T = int(input())
Ns = [int(input()) for _ in range(T)]

# 풀이
results = []
for tc in range(T):
    n = Ns[tc]
    result = []
    # 전체: n(n+1)//2
    result.append(n*(n+1)//2)
    # 홀수: n**2
    result.append(n**2)
    # 짝수: n(n+1)
    result.append(n*(n+1))

    results.append(result)
# 결과
for tc in range(T):
    print(f'#{tc+1}', end=" ")
    print(*results[tc])
 

📕문제 확인

출처: SW Expert Academy

SW Expert Academy