🔻PS/Baekjoon

[Baekjoon] 백준 21921 블로그 Python

_니지 2024. 10. 4. 19:10

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

 

import sys
input = sys.stdin.readline


n, x = map(int, input().split(" "))

arr = list(map(int, input().split(" ")))
prefix = [0 for _ in range(n)]
prefix[0] = arr[0]

max_val = 0
cnt = 0

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

prefix = [0] +prefix


for i in range(n, -1, -1):
    hap = prefix[i] - prefix[i-x]

    if hap > max_val:
        max_val = hap
        cnt = 1
    elif hap == max_val:
        cnt += 1


if max_val == 0:
    print("SAD")
else:
    print(max_val)
    print(cnt)
728x90
반응형