< 과제 >
[ 문제 ]
동호는 내년에 초등학교를 입학한다. 그래서 동호 어머니는 수학 선행 학습을 위해 쉽게 푸는 문제를 동호에게 주었다.이 문제는 다음과 같다. 1을 한 번, 2를 두 번, 3을 세 번, 이런 식으로 1 2 2 3 3 3 4 4 4 4 5 .. 이러한 수열을 만들고 어느 일정한 구간을 주면 그 구간의 합을 구하는 것이다.
하지만 동호는 현재 더 어려운 문제를 푸느라 바쁘기에 우리가 동호를 도와주자.
[ 입력 ]
첫째 줄에 구간의 시작과 끝을 나타내는 정수 A, B(1 ≤ A ≤ B ≤ 1,000)가 주어진다. 즉, 수열에서 A번째 숫자부터 B번째 숫자까지 합을 구하면 된다.[ 출력 ]
첫 줄에 구간에 속하는 숫자의 합을 출력한다.[ 예제 입력 1 ]
3 7[ 예제 출력 1 ]
15
< 내 코드 >
A, B = map(int, input().split())
arr = []
for i in range(100):
for j in range(i):
arr.append(i)
# A, B 둘 다 최대 값이 1000이므로 arr의 길이가 1000이 되면 탈출
if len(arr) == 1000:
break
if len(arr) == 1000:
break
print(sum(arr[A-1 : B]))- 문제의 배열을 구하는 것이 먼저이다. 반복문을 도는데 첫 번째 반복분에서 i를 지정하고, 두 번째 반복문에서 i 만큼 i를 리스트에 추가한다. 그러면 문제에서 제시한 1, 2, 2, 3, 3, 3, 4, 4, 4, 4, ...가 된다.
- 조건에서 A와 B의 최대값은 1000이라고 한다. 따라서 배열의 최대 길이도 1000까지만 되면된다.(참고로 탈출 조건을 지정하지 않으면 배열의 길이가 4950이 된다.)
- break는 바로 앞에 오는 반복문만 끝낼 수 있으므로 동일한 탈출 조건이지만 반복문 두 개 모두 적용한다.
< 피드백 >
- break의 조건이 한 번만 있으면 되는 줄 알고 있었다. 똑같은 코드를 반복해서 쓰는 것이 싫어서 두 번째 break는 조건 없이 썼더니 배열이 계속 비어있는 것이다. 아! break의 조건은 각각 써줘야 하는구나!
< 출처 >
