[파이썬] 백준 2559 수열

< 해결 방법 > 

이 문제 역시도 시간 초과가 나서 미리 값을 저장을 하는 방식으로 해결을 하였다. b가 1이거나 a인 경우 예외 처리를 해주었고 k[b-1] 값을 따로 append 해주어서 오류를 해결했다.

 

 

import sys
 
a,b = map(int,sys.stdin.readline().split(" "))
 
k = list(map(int,sys.stdin.readline().split(" ")))
 
answer = [ ]
 
if b==1:
    print(max(k))
elif b==a:
    print(sum(k))
else:        
 
    for i in range(1,len(k)):
        k[i]+=k[i-1]    
 
    answer.append(k[b-1])
    for i in range(b,len(k)):
        answer.append(k[i]-k[i-b])
 
    print(max(answer))