結果

問題 No.5007 Steiner Space Travel
ユーザー ID 21712
提出日時 2025-01-30 13:58:01
言語 Go
(1.23.4)
結果
AC  
実行時間 789 ms / 1,000 ms
コード長 3,432 bytes
コンパイル時間 13,421 ms
コンパイル使用メモリ 251,040 KB
実行使用メモリ 6,824 KB
スコア 8,356,724
最終ジャッジ日時 2025-01-30 13:58:40
合計ジャッジ時間 38,089 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

package main

import . "fmt"
import "math/rand"

const ALPHA = 5

func dist(a,b []int, i int, c,d []int, j int) int {
	dx := a[i] - c[j]
	dy := b[i] - d[j]
	return dx*dx + dy*dy
}

func main() {
	var n, m int
	Scan(&n, &m)
	a := make([]int, n)
	b := make([]int, n)
	for i := range a {
		Scan(&a[i],&b[i])
	}
	c := make([]int, m)
	d := make([]int, m)
	for i := range c {
		if i < m/2 {
			c[i] = (i/3+1)*250
			d[i] = (i%3+1)*250
		} else {
			c[i] = ((i+1)/3+1)*250
			d[i] = ((i+1)%3+1)*250
		}
		for j, va := range a {
			vb := b[j]
			ok := true
			for k, vc := range c[:i] {
				vd := d[k]
				dx := va - vc
				dy := vb - vd
				ok = ok && (dx*dx+dy*dy>200*200)
			}
			if ok {
				c[i] = a[j]
				d[i] = b[j]
				break
			}
		}
	}

	path := make([][][]int, n+m)
	for i := range path {
		path[i] = make([][]int, n+m)
		for j := range path[i] {
			switch {
				case i == j:
					path[i][j] = []int{ 0, j }
				case i < n && j < n:
					path[i][j] = []int{
						ALPHA*ALPHA*dist(a,b,i,a,b,j),
						j,
					}
				case i < n:
					path[i][j] = []int{
						ALPHA*dist(a,b,i,c,d,j-n),
						j,
					}
				case j < n:
					path[i][j] = []int{
						ALPHA*dist(c,d,i-n,a,b,j),
						j,
					}
				default:
					path[i][j] = []int{
						dist(c,d,i-n,c,d,j-n),
						j,
					}
			}
		}
	}
	
	for i := 0; i < n+m; i++ {
		for j := 0; j < n+m; j++ {
			if i == j {
				continue
			}
			for k := 0; k < n+m; k++ {
				if i==k || j==k {
					continue
				}
				if x := path[i][k][0]+path[k][j][0]; x < path[i][j][0] {
					path[i][j][0] = x
					path[i][j][1] = k
				}
			}
		}
	}
	
	p := []int{0}
	for i := 0; i < n; i++ {
		p = append(p, (i+1)%n)
	}
	
	calc := func(p []int) int {
		sc := 0
		for i := 0; i < n; i++ {
			sc += path[p[i]][p[i+1]][0]
		}
		return sc
	}
	
	bestSc := calc(p)
	bestP := make([]int, len(p))
	copy(bestP, p)

	for kk := 0; kk < 80; kk++ {
	
		for cc, changed := 0, true; changed && cc < 100; cc++ {
			changed = false
			// 2-opt ?
			for i := 1; i < n; i++ {
				for j := i+1; j < n; j++ {
					s0 := path[p[i-1]][p[i]][0] +
						path[p[j]][p[j+1]][0]
					s1 := path[p[i-1]][p[j]][0] +
						path[p[i]][p[j+1]][0]
					if s1 < s0 {
						for f,e := i,j; f < e; f,e=f+1,e-1 {
							p[f],p[e] = p[e],p[f]
						}
						changed = true
					}
				}
			}
			// 3-opt ?
			for i := 1; i < n; i++ {
				for j := i+1; j < n; j++ {
					for k := j+2; k < n; k++ {
						s0 := path[p[i-1]][p[i]][0] +
							path[p[j]][p[j+1]][0] +
							path[p[k]][p[k+1]][0]
						s1 := path[p[i-1]][p[j]][0] +
							path[p[i]][p[k]][0] +
							path[p[j+1]][p[k+1]][0]
						if s1 < s0 {
							for f,e := i,j; f < e; f,e=f+1,e-1 {
								p[f],p[e] = p[e],p[f]
							}
							for f,e := j+1,k; f < e; f,e=f+1,e-1 {
								p[f],p[e] = p[e],p[f]
							}
							changed = true
						}
					}
				}
			}
		}
		
		if sc := calc(p); sc < bestSc {
			bestSc = sc
			copy(bestP, p)
		}

		rand.Shuffle(n-1, func(i, j int) {
			p[i+1],p[j+1] = p[j+1],p[i+1]
		})
	}
	
	p = bestP
		
	t := []int{1}
	r := []int{1}
	var add func(i, j int)
	add = func(i, j int) {
		rt := path[i][j]
		if rt[1] == j {
			if j < n {
				t = append(t, 1)
				r = append(r, j+1)
			} else {
				t = append(t, 2)
				r = append(r, j-n+1)
			}
			return
		}
		add(i, rt[1])
		add(rt[1], j)
	}
	
	for i := 0; i < n; i++ {
		add(p[i], p[i+1])
	}
	
	for i := range c {
		Println(c[i], d[i])
	}
	Println(len(t))
	for i := range t {
		Println(t[i], r[i])
	}
	
}
0