結果

問題 No.5007 Steiner Space Travel
ユーザー ks2mks2m
提出日時 2022-07-30 16:15:31
言語 Java21
(openjdk 21)
結果
AC  
実行時間 996 ms / 1,000 ms
コード長 6,185 bytes
コンパイル時間 2,304 ms
実行使用メモリ 62,496 KB
スコア 7,845,070
最終ジャッジ日時 2022-07-30 16:16:11
合計ジャッジ時間 34,825 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge10
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 991 ms
53,076 KB
testcase_01 AC 996 ms
52,948 KB
testcase_02 AC 992 ms
54,424 KB
testcase_03 AC 990 ms
54,148 KB
testcase_04 AC 992 ms
54,536 KB
testcase_05 AC 991 ms
52,504 KB
testcase_06 AC 994 ms
52,844 KB
testcase_07 AC 991 ms
53,548 KB
testcase_08 AC 992 ms
54,080 KB
testcase_09 AC 993 ms
51,228 KB
testcase_10 AC 990 ms
52,992 KB
testcase_11 AC 992 ms
54,612 KB
testcase_12 AC 994 ms
53,812 KB
testcase_13 AC 993 ms
52,844 KB
testcase_14 AC 992 ms
52,392 KB
testcase_15 AC 995 ms
52,980 KB
testcase_16 AC 991 ms
54,376 KB
testcase_17 AC 992 ms
52,844 KB
testcase_18 AC 996 ms
53,136 KB
testcase_19 AC 995 ms
52,892 KB
testcase_20 AC 994 ms
53,696 KB
testcase_21 AC 994 ms
62,496 KB
testcase_22 AC 996 ms
51,304 KB
testcase_23 AC 992 ms
52,868 KB
testcase_24 AC 992 ms
52,952 KB
testcase_25 AC 991 ms
52,480 KB
testcase_26 AC 993 ms
52,732 KB
testcase_27 AC 993 ms
52,076 KB
testcase_28 AC 994 ms
53,932 KB
testcase_29 AC 996 ms
53,060 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class Main {
	static boolean batch = false;
	static boolean readFile = false;
	static int num = 0;
	static long limitTime = 900;
	static long startTime;

	static int a = 5;
	static int N, M;
	static Obj[] arr, pln;

	public static void main(String[] args) throws Exception {
		if (batch) {
			readFile = true;
		}
		if (readFile) {
			if (batch) {
				long tl = 1000;
				long tw = (long) (tl * 0.9);
				long total = 0;
				int bidx = 0;
				int best = 0;
				int widx = 0;
				int worst = 1000000000;
				int re = 0;
				int tle = 0;
				for (int z = 0; z < 100; z++) {
					try {
						long st = System.currentTimeMillis();
						int score = solve(z);
						long time = System.currentTimeMillis() - st;
						if (time > tw) {
							System.out.println(z + ":\t" + score + "\t" + time + "ms");
							if (time > tl) {
								tle++;
							}
						} else {
							System.out.println(z + ":\t" + score);
						}
						total += score;
						if (score > best) {
							best = score;
							bidx = z;
						}
						if (score < worst) {
							worst = score;
							widx = z;
						}
					} catch (Exception e) {
						System.out.println(z + ":\t" + e.getMessage());
						re++;
					}
				}
				System.out.println("total: " + total);
				System.out.println("best: " + bidx + ": " + best);
				System.out.println("worst: " + widx + ": " + worst);
				System.out.println("RE: " + re);
				System.out.println("TLE: " + tle);
			} else {
				int score = solve(num);
				System.out.println(score);
			}

		} else {
			BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
			init(br);
			solve();
		}
	}

	static int solve(int z) throws Exception {
		StringBuilder sb = new StringBuilder();
		sb.append(z);
		while (sb.length() < 4) {
			sb.insert(0, '0');
		}
		File file = new File("G:\\input\\" + sb.toString() + ".txt");
		BufferedReader br = new BufferedReader(new FileReader(file));
		init(br);

		return solve();
	}

	static void init(BufferedReader br) throws Exception {
		startTime = System.currentTimeMillis();

		String[] sa = br.readLine().split(" ");
		N = Integer.parseInt(sa[0]);
		M = Integer.parseInt(sa[1]);
		arr = new Obj[N + M];
		pln = new Obj[N];
		for (int i = 0; i < N; i++) {
			sa = br.readLine().split(" ");
			Obj o = new Obj();
			o.t = 1;
			o.i = i;
			o.x = Integer.parseInt(sa[0]);
			o.y = Integer.parseInt(sa[1]);
			pln[i] = o;
			arr[i] = o;
		}
	}

	static int solve() throws Exception {
		Ans ans = new Ans();
		int cnt = 0;
		while (System.currentTimeMillis() - startTime < limitTime) {
			Ans res = solve2();
			if (res.score > ans.score) {
				ans = res;
			}
			cnt++;
		}

		if (!batch) {
			PrintWriter pw = new PrintWriter(System.out);
			for (Obj o : ans.sta) {
				pw.println(o.x + " " + o.y);
			}
			pw.println(ans.his.size());
			for (int cur : ans.his) {
				int ct = cur / N;
				int ci = cur % N;
				pw.println(ct + " " +  (ci + 1));
			}
			if (readFile) {
				pw.println("cnt: " + cnt);
			}
			pw.flush();
		}
		return ans.score;
	}

	static Ans solve2() throws Exception {
		// 宇宙ステーションの配置
		Obj[] sta = new Obj[M];
		for (int i = 0; i < M; i++) {
			Obj o = new Obj();
			o.t = 2;
			o.i = i;
			o.x = rand(100, 900);
			o.y = rand(100, 900);
			while (true) {
				boolean flg = true;
				for (int j = 0; j < i; j++) {
					Obj o2 = sta[j];
					if (dist(o, o2) < 200) {
						flg = false;
						break;
					}
				}
				if (flg) {
					break;
				}
				o.x = rand(100, 900);
				o.y = rand(100, 900);
			}
			sta[i] = o;
			arr[N + i] = o;
		}

		// 最も近い宇宙ステーションを設定
		for (Obj o : pln) {
			o.costs = Integer.MAX_VALUE;
			for (Obj o2 : sta) {
				int e = cost(o, o2);
				if (e < o.costs) {
					o.costs = e;
					o.sta = o2;
				}
			}
		}

		Set<Integer> rem = new HashSet<>();
		for (int i = 1; i < N; i++) {
			rem.add(i);
		}

		List<Integer> his = new ArrayList<>();
		his.add(toInt(pln[0]));
		Obj now = pln[0];
		while (!rem.isEmpty()) {
			int min = Integer.MAX_VALUE;
			List<Integer> nexts = new ArrayList<>(3);
			for (int i : rem) {
				min = minCost(now, pln[i], min, nexts);
			}
			his.addAll(nexts);
			int next = his.get(his.size() - 1) % N;
			now = pln[next];
			rem.remove(next);
		}
		List<Integer> nexts = new ArrayList<>(3);
		minCost(now, pln[0], Integer.MAX_VALUE, nexts);
		his.addAll(nexts);

		// 得点計算
		int s = 0;
		int v = his.size();
		for (int i = 1; i < v; i++) {
			int cur = his.get(i);
			int pre = his.get(i - 1);
			int e = cost(arr[pre - N], arr[cur - N]);
			s += e;
		}
		int score = (int) Math.round(1000000000 / (1000 + Math.sqrt(s)));

		Ans ans = new Ans();
		ans.sta = sta;
		ans.his = his;
		ans.score = score;
		return ans;
	}

	static int minCost(Obj now, Obj o, int min, List<Integer> nexts) {
		int d = cost(now, o);
		if (d < min) {
			min = d;
			nexts.clear();
			nexts.add(toInt(o));
		}
		d = cost(now, now.sta) + cost(now.sta, o);
		if (d < min) {
			min = d;
			nexts.clear();
			nexts.add(toInt(now.sta));
			nexts.add(toInt(o));
		}
		d = cost(now, o.sta) + cost(o.sta, o);
		if (d < min) {
			min = d;
			nexts.clear();
			nexts.add(toInt(o.sta));
			nexts.add(toInt(o));
		}
		d = cost(now, now.sta) + cost(now.sta, o.sta) + cost(now.sta, o);
		if (d < min) {
			min = d;
			nexts.clear();
			nexts.add(toInt(now.sta));
			nexts.add(toInt(o.sta));
			nexts.add(toInt(o));
		}
		return min;
	}

	static int dist(Obj o1, Obj o2) {
		int dx = o1.x - o2.x;
		int dy = o1.y - o2.y;
		return dx * dx + dy * dy;
	}

	static int cost(Obj o1, Obj o2) {
		int e = dist(o1, o2);
		if (o1.t == 1) e *= a;
		if (o2.t == 1) e *= a;
		return e;
	}

	static int toInt(Obj o) {
		return o.t * N + o.i;
	}

	static int rand(int l, int u) {
		int d = u - l + 1;
		int x = (int) (Math.random() * d);
		return x + l;
	}

	static class Obj {
		int t, i, x, y;
		Obj sta;
		int costs;
	}

	static class Ans {
		Obj[] sta;
		List<Integer> his;
		int score;
	}
}
0