結果

問題 No.5007 Steiner Space Travel
ユーザー titan23titan23
提出日時 2022-10-06 14:51:17
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 889 ms / 1,000 ms
コード長 3,174 bytes
コンパイル時間 235 ms
実行使用メモリ 85,212 KB
スコア 7,021,237
最終ジャッジ日時 2022-10-06 14:51:37
合計ジャッジ時間 19,293 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 521 ms
83,936 KB
testcase_01 AC 490 ms
83,584 KB
testcase_02 AC 491 ms
84,136 KB
testcase_03 AC 516 ms
83,624 KB
testcase_04 AC 491 ms
84,008 KB
testcase_05 AC 547 ms
83,964 KB
testcase_06 AC 506 ms
84,028 KB
testcase_07 AC 532 ms
84,020 KB
testcase_08 AC 517 ms
83,876 KB
testcase_09 AC 496 ms
84,096 KB
testcase_10 AC 527 ms
83,960 KB
testcase_11 AC 518 ms
84,028 KB
testcase_12 AC 502 ms
84,036 KB
testcase_13 AC 539 ms
83,976 KB
testcase_14 AC 502 ms
84,232 KB
testcase_15 AC 498 ms
84,100 KB
testcase_16 AC 504 ms
83,756 KB
testcase_17 AC 493 ms
83,424 KB
testcase_18 AC 512 ms
83,892 KB
testcase_19 AC 527 ms
83,844 KB
testcase_20 AC 514 ms
83,972 KB
testcase_21 AC 889 ms
85,212 KB
testcase_22 AC 517 ms
83,980 KB
testcase_23 AC 510 ms
84,020 KB
testcase_24 AC 508 ms
84,280 KB
testcase_25 AC 490 ms
84,260 KB
testcase_26 AC 513 ms
83,936 KB
testcase_27 AC 529 ms
84,132 KB
testcase_28 AC 497 ms
84,056 KB
testcase_29 AC 625 ms
84,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import itertools
import random
import time
start = time.time()
N, M = map(int, input().split())
AB = [list(map(int, input().split())) for _ in range(N)]

class KMeans:

  def __init__(self, n_clusters, max_iter = 10, random_seed = 0):
    self.n_clusters = n_clusters
    self.max_iter = max_iter

  def fit(self, X):
    self.labels_ = []
    now = 0
    while len(self.labels_) < len(X):
      self.labels_.append(now)
      now += 1
      now %= self.n_clusters
    random.shuffle(self.labels_)
    labels_prev = [0]*len(X)
    count = 0
    self.cluster_centers_ = [(0, 0)] * self.n_clusters

    while (not (self.labels_ == labels_prev) and count < self.max_iter):
      syuukei = [[] for _ in range(self.n_clusters)]
      for i in range(len(X)):
        syuukei[self.labels_[i]].append(X[i])
      for i,l in enumerate(syuukei):
        if l:
          x, y = sum(x for x,y in l)//len(l), sum(y for x,y in l)//len(l)
        else:
          x, y = random.randint(0, 1000), random.randint(0, 1000)
        self.cluster_centers_[i] = (x, y)

      labels_prev = self.labels_[:]
      for i in range(len(X)):
        dist = 10**18
        for j in range(self.n_clusters):
          tmp = (X[i][0] - self.cluster_centers_[j][0])**2 + (X[i][1] - self.cluster_centers_[j][1])**2
          if tmp < dist:
            dist = tmp
            self.labels_[i] = j
      count += 1


def main():
  model = KMeans(8)
  model.fit(AB)
  labels = model.labels_
  centers = model.cluster_centers_

  def dist(a, b):
    return (a[0] - b[0])**2 + (a[1] - b[1])**2

  def calc(ans):
    score = 0
    for i in range(len(ans)-1):
      type_pre, indx_pre = ans[i]
      type_now, indx_now = ans[i+1]
      if type_pre == 1 and type_now == 1:
        score += 25*dist(AB[indx_pre], AB[indx_now])
      if type_pre == 1 and type_now == 2:
        score +=  5*dist(AB[indx_pre], centers[indx_now])
      if type_pre == 2 and type_now == 1:
        score +=  5*dist(centers[indx_pre], AB[indx_now])
      if type_pre == 2 and type_now == 2:
        score +=  1*dist(centers[indx_pre], centers[indx_now])
    return score

  vest = 1<<30
  ANS = []
  L = [[] for _ in range(8)]
  for i, cluster_number in enumerate(labels):
    L[cluster_number].append(i)
  for p in itertools.permutations(range(8)):
    if time.time() - start > 0.9:
      return 1<<30, -1, -1
    ans = [(1, 0)]
    for i in p:
      if i >= len(L): continue
      ans.append((2, i))
      lim = len(L[i])
      j = 0
      while j < lim:
        d1 = 5 * dist(AB[L[i][j]], centers[i])
        d2 = 1<<30
        if j+1 < len(L[i]):
          d2 = 25* dist(AB[L[i][j]], AB[L[i][j+1]])
        if d2 < d1:
          ans.append((1, L[i][j]))
        else:
          ans.append((1, L[i][j]))
          ans.append((2, i))
        j += 1
    ans.append((1, 0))
    score = calc(ans)
    if score < vest:
      vest = score
      ANS = ans[:]
  return vest, ANS, centers

ANS = []
CENTERS = []
vest = 1<<30
while time.time() - start < 0.4:
  tmp, ans, centers = main()
  if tmp < vest:
      vest = tmp
      ANS = ans[:]
      CENTERS = centers

for c,d in CENTERS:
  print(c, d)
print(len(ANS))
for a,b in ANS:
  print(a, b+1)
0