문제
풀이
- 수학시험 수열의 규칙은 1 2 2 3 3 3 4 4 4 4 5 5 5 5 5 와 같은 규칙이다
- num * num 을 list에 append or extend하면 되는데 sum을 하기에 일차원 list가 용이하니 extend를 사용하였다.
코드
# https://www.acmicpc.net/problem/8974
# boj, 8974: 희주의 수학시험, python3
import sys
input = sys.stdin.readline # 변수 입력 속도 향상
# 희주가 대답해야할 정답을 구하는 함수
def solve(a: int, b: int) -> int:
result = []
# extand로 1차원 확장
for x in range(1, b+1):
result += [x] * x
# a번째부터 b까지의 list합 슬라이싱
return sum(result[a-1:b])
if __name__ == '__main__':
a, b = map(int, input().split())
print(solve(a, b))결과
출처 & 깃허브
