結果
問題 | No.2494 Sum within Components |
ユーザー |
|
提出日時 | 2023-10-06 21:55:58 |
言語 | C#(csc) (csc 3.9.0) |
結果 |
AC
|
実行時間 | 288 ms / 2,000 ms |
コード長 | 1,933 bytes |
コンパイル時間 | 2,483 ms |
コンパイル使用メモリ | 113,876 KB |
実行使用メモリ | 52,136 KB |
最終ジャッジ日時 | 2024-07-26 16:03:52 |
合計ジャッジ時間 | 3,382 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc) Copyright (C) Microsoft Corporation. All rights reserved.
ソースコード
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, m) = (c[0], c[1]); var a = NList; var map = NMap(m); var uf = new UnionFindTree(n); foreach (var edge in map) { uf.Unite(edge[0], edge[1]); } var sum = new long[n]; for (var i = 0; i < n; ++i) sum[uf.GetRoot(i)] += a[i]; var ans = 1L; var mod = 998_244_353; for (var i = 0; i < n; ++i) ans = ans * (sum[uf.GetRoot(i)] % mod) % mod; 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)]; } } }