[백준 24499] Blobyum - 파이썬

문제

https://www.acmicpc.net/problem/24499

 

 

 

풀이

1. 이 문제는 선형이 아니라 원형 큐의 형태이므로 입력받은 배열을 뒤에 하나 더 붙여서 풀면 편하다.

2. 배열의 처음부터 끝까지 누적합을 visited 배열에 저장한다.

3. visited 배열을 돌며 인덱스의 차이가 k인 수들의 차이의 최대값을 구한다.

 

 

 

코드

n, k = map(int, input().split())
arr = list(map(int, input().split()))
arr.extend(arr)

ans = 0
visited = [0] * 200002

for i in range(1, n*2 + 1):
    visited[i] = visited[i-1] + arr[i-1]

for i in range(k, n*2 + 1):
    ans = max(ans, visited[i] - visited[i-k])
    
    
print(ans)