結果
問題 | No.1293 2種類の道路 |
ユーザー |
|
提出日時 | 2024-12-11 00:11:23 |
言語 | C# (.NET 8.0.404) |
結果 |
AC
|
実行時間 | 278 ms / 2,000 ms |
コード長 | 2,777 bytes |
コンパイル時間 | 10,038 ms |
コンパイル使用メモリ | 167,776 KB |
実行使用メモリ | 215,684 KB |
最終ジャッジ日時 | 2024-12-11 00:11:46 |
合計ジャッジ時間 | 15,983 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
コンパイルメッセージ
復元対象のプロジェクトを決定しています... /home/judge/data/code/main.csproj を復元しました (95 ms)。 MSBuild のバージョン 17.9.6+a4ecab324 (.NET) main -> /home/judge/data/code/bin/Release/net8.0/main.dll main -> /home/judge/data/code/bin/Release/net8.0/publish/
ソースコード
using System;using static System.Console;using System.Linq;using System.Collections.Generic;class Program{static int NN => int.Parse(ReadLine());static int[] NList => ReadLine().Split().Select(int.Parse).ToArray();static int[] NMi => ReadLine().Split().Select(c => int.Parse(c) - 1).ToArray();static int[][] NMap(int n) => Enumerable.Repeat(0, n).Select(_ => NMi).ToArray();public static void Main(){Solve();}static void Solve(){var c = NList;var (n, d, w) = (c[0], c[1], c[2]);var dlist = NMap(d);var wlist = NMap(w);var dur = new UnionFindTree(n);foreach (var edge in dlist) dur.Unite(edge[0], edge[1]);var wur = new UnionFindTree(n);foreach (var edge in wlist) wur.Unite(edge[0], edge[1]);var wgroups = new HashSet<int>[n];for (var i = 0; i < n; ++i) wgroups[i] = new HashSet<int>();for (var i = 0; i < n; ++i) wgroups[dur.GetRoot(i)].Add(wur.GetRoot(i));var wcontains = new Dictionary<int, int>[n];for (var i = 0; i < n; ++i) wcontains[i] = new Dictionary<int, int>();for (var i = 0; i < n; ++i){var dr = dur.GetRoot(i);var wr = wur.GetRoot(i);if (wcontains[wr].ContainsKey(dr)) ++wcontains[wr][dr];else wcontains[wr][dr] = 1;}var ans = 0L;for (var i = 0; i < n; ++i){if (i != dur.GetRoot(i)) continue;ans += (dur.GetSize(i) - 1L) * dur.GetSize(i);foreach (var wr in wgroups[i]) ans += ((long)wur.GetSize(wr) - wcontains[wr][i]) * dur.GetSize(i);}for (var i = 0; i < n; ++i){if (i != wur.GetRoot(i)) continue;}WriteLine(ans);}class UnionFindTree{int[] roots;public UnionFindTree(int size){roots = new int[size];for (var i = 0; i < size; ++i) roots[i] = -1;}public int GetRoot(int a){if (roots[a] < 0) return a;return roots[a] = GetRoot(roots[a]);}public bool IsSameTree(int a, int b){return GetRoot(a) == GetRoot(b);}public bool Unite(int a, int b){var x = GetRoot(a);var y = GetRoot(b);if (x == y) return false;if (-roots[x] < -roots[y]) { var tmp = x; x = y; y = tmp; }roots[x] += roots[y];roots[y] = x;return true;}public int GetSize(int a){return -roots[GetRoot(a)];}}}