Source Code
a , b = input("").split(" ")[:2]
a = int(a)
b = int(b)
d = input("").split(" ")

c = [int(i) for i in d]



lists = [[] for i in range(b)]

if(b==2):
    
    for i in range(a):
        if(i<a/2):
            lists[0].append(min(c))
            c.remove(min(c))
        else:
            lists[1].append(min(c))
            c.remove(min(c))
            

for i in range(int(a/b)):
    if(b==1 or b==2):
        break
    else:
        for i in range(b):
            lists.append(list())
            lists[i].append(min(c))
            c.remove(min(c))

mex = []

for i in range(b):
    if(b==1):
        break
    min = 0 
    while(min in lists[i]):
        min = min+1
    mex.append(min)
    
if(b==1):
    min = 0
    while(min in c):
        min = min + 1
    
    
   
if(b==1):
    print(min)
else:
    print(sum(mex))
Copy
Bitar The Handy Man cheesy
Python 3
18 ms
2.9 MB
Wrong Answer