194782025-12-10 16:00:31birozsSípálya (55 pont)pypy3Time limit exceeded 21/55497ms49248 KiB
N,K=map(int,input().split())
T = list(map(int,input().split()))
fold = 0
m = T[0]
for i in range(1,K):
    if T[i] > m - 1:
        fold += (T[i] - (m-1)) * i
        m = T[i]
    else:
        fold += m - 1 - T[i]
        m = m - 1
min = N * 10**9
for i in range(1,N-K+1):
    if T[i-1] < m + K - 1 and T[i+K-1] < m:
        fold -= m + K - 1 - T[i-1]
        m = m - 1
        fold += m - T[i+K-1]
    if T[i-1] < m + K - 1 and T[i+K-1] == m:
        fold -= m + K - 1 - T[i-1]
        fold += K - 1
    if T[i-1] < m + K - 1 and T[i+K-1] > m:
        fold -= m + K - 1 - T[i-1]
        fold += (K - 1) * (T[i+K-1] - m + 1)
        m = T[i+K-1]
    else:
        fold = 0
        m = T[i]
        for j in range(i,i+K):
            if T[j] > m - 1:
                fold += (T[j] - (m-1)) * (j-i)
                m = T[j]
            else:
                fold += m - 1 - T[j]
                m = m - 1
    if fold < min:
        min = fold
print(min)


    

SubtaskSumTestVerdictTimeMemory
base21/55
1Accepted0/039ms19668 KiB
2Accepted0/043ms19544 KiB
3Accepted2/264ms21700 KiB
4Accepted2/256ms21960 KiB
5Accepted2/254ms21668 KiB
6Accepted2/261ms21828 KiB
7Accepted3/352ms21716 KiB
8Accepted1/194ms22016 KiB
9Accepted1/197ms21992 KiB
10Accepted1/1105ms22108 KiB
11Accepted1/1119ms21920 KiB
12Accepted1/1127ms21992 KiB
13Accepted1/1216ms22248 KiB
14Accepted2/2245ms22392 KiB
15Accepted2/2119ms21856 KiB
16Time limit exceeded0/2497ms47640 KiB
17Time limit exceeded0/2495ms47640 KiB
18Time limit exceeded0/2490ms47640 KiB
19Time limit exceeded0/3490ms47616 KiB
20Time limit exceeded0/2486ms47612 KiB
21Time limit exceeded0/2495ms47644 KiB
22Time limit exceeded0/2483ms47628 KiB
23Time limit exceeded0/2488ms47768 KiB
24Time limit exceeded0/2483ms49248 KiB
25Time limit exceeded0/2486ms47612 KiB
26Time limit exceeded0/2488ms47824 KiB
27Time limit exceeded0/2486ms47624 KiB
28Time limit exceeded0/3486ms47636 KiB
29Time limit exceeded0/3492ms47588 KiB
30Time limit exceeded0/3481ms47616 KiB