BOJ/백준-20953-python

문제

풀이

• 원본 코드가 3중 반복문이므로 시간초과가 나서 고대로 구현하면 안된다.
• 반복문의 횟수를 계산해서 바로 반환하면 된다.

코드

# https://www.acmicpc.net/problem/20953
# boj, 20953: 고고학자 예린, python3
import sys

input = sys.stdin.readline

# def dolmen(a: int, b: int) -> int:
#     sum, k = 0, 0
#
#     for _ in range(0, a + b):
#         for j in range(0, a + b):
#             while k < j:
#                 sum += 1
#                 k += 1
#             k = 0
#
#     return sum
def dolmen(a: int, b: int) -> int:
    return (a + b) * (a + b - 1) * (a + b) // 2


if __name__ == '__main__':
    t = int(input())  # 테스트 케이스의 개수

    for _ in range(t):
        a, b = map(int, input().split())

        print(dolmen(a, b))

결과

출처 & 깃허브

boj 20953
github