結果

問題 No.1690 Power Grid
ユーザー ks2mks2m
提出日時 2021-09-24 22:43:45
言語 Java21
(openjdk 21)
結果
AC  
実行時間 556 ms / 3,000 ms
コード長 2,682 bytes
コンパイル時間 3,501 ms
コンパイル使用メモリ 88,104 KB
実行使用メモリ 62,420 KB
最終ジャッジ日時 2024-07-05 11:03:26
合計ジャッジ時間 11,070 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 129 ms
54,084 KB
testcase_01 AC 116 ms
52,800 KB
testcase_02 AC 131 ms
54,440 KB
testcase_03 AC 127 ms
54,296 KB
testcase_04 AC 125 ms
54,124 KB
testcase_05 AC 128 ms
54,048 KB
testcase_06 AC 453 ms
60,792 KB
testcase_07 AC 466 ms
60,820 KB
testcase_08 AC 504 ms
60,388 KB
testcase_09 AC 450 ms
60,628 KB
testcase_10 AC 147 ms
54,240 KB
testcase_11 AC 165 ms
54,528 KB
testcase_12 AC 131 ms
54,068 KB
testcase_13 AC 140 ms
53,924 KB
testcase_14 AC 255 ms
59,616 KB
testcase_15 AC 191 ms
54,656 KB
testcase_16 AC 556 ms
62,420 KB
testcase_17 AC 375 ms
61,464 KB
testcase_18 AC 337 ms
61,372 KB
testcase_19 AC 511 ms
61,472 KB
testcase_20 AC 513 ms
61,232 KB
testcase_21 AC 162 ms
54,320 KB
testcase_22 AC 336 ms
61,536 KB
testcase_23 AC 161 ms
54,728 KB
testcase_24 AC 204 ms
55,760 KB
testcase_25 AC 201 ms
55,560 KB
testcase_26 AC 203 ms
54,960 KB
testcase_27 AC 131 ms
53,828 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.PriorityQueue;
import java.util.Scanner;

public class Main {
	public static void main(String[] args) throws Exception {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int m = sc.nextInt();
		int k = sc.nextInt();
		int[] a = new int[n];
		for (int i = 0; i < n; i++) {
			a[i] = sc.nextInt();
		}

		long inf = 1000000000000L;
		long[][] d = new long[n][n];
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < n; j++) {
				if (i != j) {
					d[i][j] = inf;
				}
			}
		}
		for (int i = 0; i < m; i++) {
			int x = sc.nextInt() - 1;
			int y = sc.nextInt() - 1;
			int z = sc.nextInt();
			d[x][y] = z;
			d[y][x] = z;
		}
		for (int p = 0; p < n; p++) {
			for (int i = 0; i < n; i++) {
				for (int j = 0; j < n; j++) {
					if (d[i][p] != inf && d[p][j] != inf) {
						d[i][j] = Math.min(d[i][j], d[i][p] + d[p][j]);
					}
				}
			}
		}
		sc.close();

		List<Obj> list = new ArrayList<>();
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < n; j++) {
				if (i != j) {
					Obj o = new Obj();
					o.x = i;
					o.y = j;
					o.z = d[i][j];
					list.add(o);
				}
			}
		}
		Collections.sort(list, (o1, o2) -> Long.compare(o1.z, o2.z));

		long ans = Long.MAX_VALUE;
		int end = 1 << n;
		for (int i = 0; i < end; i++) {
			if (Integer.bitCount(i) != k) {
				continue;
			}
			PriorityQueue<Integer> que = new PriorityQueue<>();
			for (int j = 0; j < n; j++) {
				if ((i >> j & 1) == 1) {
					que.add(a[j]);
				}
			}
			long val = 0;
			for (int j = 0; j < k; j++) {
				val += que.poll();
			}

			UnionFind uf = new UnionFind(n);
			for (Obj o : list) {
				if ((i >> o.x & 1) == 1 && (i >> o.y & 1) == 1
						&& !uf.same(o.x, o.y)) {
					uf.union(o.x, o.y);
					val += o.z;
				}
			}
			ans = Math.min(ans, val);
		}
		System.out.println(ans);
	}

	static class Obj {
		int x, y;
		long z;
	}

	static class UnionFind {
		int[] parent, size;
		int num = 0; // 連結成分の数

		UnionFind(int n) {
			parent = new int[n];
			size = new int[n];
			num = n;
			for (int i = 0; i < n; i++) {
				parent[i] = i;
				size[i] = 1;
			}
		}

		void union(int x, int y) {
			int px = find(x);
			int py = find(y);
			if (px != py) {
				parent[px] = py;
				size[py] += size[px];
				num--;
			}
		}

		int find(int x) {
			if (parent[x] == x) {
				return x;
			}
			parent[x] = find(parent[x]);
			return parent[x];
		}

		/**
		 * xとyが同一連結成分か
		 */
		boolean same(int x, int y) {
			return find(x) == find(y);
		}

		/**
		 * xを含む連結成分のサイズ
		 */
		int size(int x) {
			return size[find(x)];
		}
	}
}
0