import sys, random input = lambda : sys.stdin.readline().rstrip() write = lambda x: sys.stdout.write(x+"\n"); writef = lambda x: print("{:.12f}".format(x)) debug = lambda x: sys.stderr.write(x+"\n") YES="Yes"; NO="No"; pans = lambda v: print(YES if v else NO); INF=10**18 LI = lambda : list(map(int, input().split())); II=lambda : int(input()) def debug(_l_): for s in _l_.split(): print(f"{s}={eval(s)}", end=" ") print() def dlist(*l, fill=0): if len(l)==1: return [fill]*l[0] ll = l[1:] return [dlist(*ll, fill=fill) for _ in range(l[0])] import random def kmean(xy, k): n = len(xy) # cs = [(random.randint(0,1000), random.randint(0,1000)) for _ in range(k)] if k>=n: cs = [xy[i%n] for i in range(k)] index = [[] for i in range(k)] for i in range(n): index[i].append(i) return cs, index nokori = set(range(n)) i0 = random.randint(0,n-1) cs = [xy[i0]] nokori.remove(i0) for i in range(m-1): i1 = max(nokori, key=lambda ii: min((xy[ii][0]-x)**2 + (xy[ii][1]-y)**2 for x,y in cs)) cs.append(xy[i1]) for _ in range(10): index = [[] for _ in range(k)] for i in range(n): x,y = xy[i] ind = min(range(k), key=lambda j: (x-cs[j][0])**2 + (y-cs[j][1])**2) index[ind].append(i) ncs = [] for i in range(k): sx = sy = 0 if not index[i]: ncs.append((random.randint(0,1000), random.randint(0,100))) else: for ind in index[i]: x,y = xy[ind] sx += x sy += y ncs.append((int(round(sx/len(index[i]))), int(round(sy/len(index[i]))))) cs = ncs return cs, index def main(n,m,xy,show=False): cs,index = kmean(xy, m) from itertools import permutations as ps best = (INF, None) for inds in ps(range(m)): cost = 0 for i in range(m): x,y = cs[inds[i]] xx,yy = cs[inds[i-1]] cost += (x-xx)**2 + (y-yy)**2 if cost