Submission #1213443


Source Code Expand

from itertools import combinations
n, k = map(int, raw_input().split())
a = map(int, raw_input().split())
ans = 10 ** 12
for p in combinations(range(n), k):
    p = list(p)[::-1]
    ma = 0
    res = 0
    for j in xrange(n):
        if p and j == p[-1]:
            if ma < a[j]:
                ma = a[j]
            else:
                res += ma + 1 - a[j]
                ma += 1
            p.pop()
        else:
            if ma < a[j]:
                ma = a[j]
    if ans > res:
        ans = res
print ans

Submission Info

Submission Time
Task B - Buildings are Colorful!
User nwin
Language Python (2.7.6)
Score 350
Code Size 540 Byte
Status AC
Exec Time 28 ms
Memory 2692 KB

Judge Result

Set Name Sample Subtask1 Subtask2 Subtask3
Score / Max Score 0 / 0 120 / 120 90 / 90 140 / 140
Status
AC × 2
AC × 2
AC × 3
AC × 9
Set Name Test Cases
Sample sub0_in1.txt, sub0_in2.txt
Subtask1 sub1_in1.txt, sub1_in2.txt
Subtask2 sub2_in1.txt, sub2_in2.txt, sub2_in3.txt
Subtask3 sub0_in1.txt, sub0_in2.txt, sub1_in1.txt, sub1_in2.txt, sub2_in1.txt, sub2_in2.txt, sub2_in3.txt, sub3_in1.txt, sub3_in2.txt
Case Name Status Exec Time Memory
sub0_in1.txt AC 11 ms 2692 KB
sub0_in2.txt AC 11 ms 2692 KB
sub1_in1.txt AC 11 ms 2692 KB
sub1_in2.txt AC 11 ms 2692 KB
sub2_in1.txt AC 11 ms 2692 KB
sub2_in2.txt AC 11 ms 2692 KB
sub2_in3.txt AC 11 ms 2692 KB
sub3_in1.txt AC 14 ms 2692 KB
sub3_in2.txt AC 28 ms 2692 KB