[ 파이썬(python) ] 백준 1292 - 쉽게 푸는 문제

📍 백준 1292 - 쉽게 푸는 문제

백준 1292 - 쉽게 푸는 문제

---

⚡️ 나의 풀이

문제 이름은 쉽게 푸는 문제였는데 나는 어렵게 푼 문제였다. 구간의 시작과 끝을 기준으로 구간 합을 구해야 하기 때문에 prefix_sum을 이용했다. 1, 2, 2, 3, 3, 3, 4, 4, 4, 4 ... 수열을 만들 때 더 쉽게 떠올릴 수 있어야 하는데 str형으로 어렵게 푼 것 같다.

 

나의 풀이

1. arr에 0을 넣은 상태로 선언한다. (index 고려)
2. a, b를 입력받는데 b는 index를 고려할 값이기 때문에 반복문의 범위를 반으로 줄였다.
3. cnt는 1씩 증가하면서 똑같은 값이 하나씩 증가한다.
4. 현재 항과 이전 항을 더해 누적 합을 계산해준다.
5. arr[b] - arr[a-1]으로 구간 합을 구한다.

 

정답판정을 맞고 다른사람의 코드를 봤는데 코드길이가 반으로 줄 정도로 간단한 문제였는지 몰랐다. 😅 😅

 

다른사람의 풀이

1. a, b의 범위는 1,000까지였기 때문에 반복문의 범위는 46까지만 구해도 된다. (len(arr) = 1036)
2. i값을 기준으로 이중 반복문을 사용하는데 빈 arr에 i를 추가한다. 값이 1부터 arr에 들어간다. (j를 추가하는 것이 아님!!)
3. sum함수를 이용해서 풀었다.(구간합을 구하지 않고 기존 arr에서 사용 가능하다.)

구간 합을 구할 때 누적 합을 구하지 않고도 sum + slicing 함수로 구할 수 있다는 것을 배웠다.

 

# 나의 코드
a, b = map(int, input().split())

arr = [0]
cnt = 1
for i in range(1, (b+1) // 2 + 1):    # 범위를 반으로 나눠서 시간 단축
    for j in range(cnt):
        arr.append(cnt)
    cnt += 1

for i in range(1, b+1):
    arr[i] += arr[i-1]
print(arr[b] - arr[a-1])

# 다른 사람의 코드
a,b = map(int,input().split())

arr = [0]
for i in range(46):
    for j in range(i):
        arr.append(i)

print(sum(arr[a:b+1]))