結果
問題 | No.2494 Sum within Components |
ユーザー | ks2m |
提出日時 | 2023-10-06 22:07:55 |
言語 | Java21 (openjdk 21) |
結果 |
AC
|
実行時間 | 852 ms / 2,000 ms |
コード長 | 3,599 bytes |
コンパイル時間 | 2,484 ms |
コンパイル使用メモリ | 78,384 KB |
実行使用メモリ | 82,900 KB |
最終ジャッジ日時 | 2023-10-06 22:08:04 |
合計ジャッジ時間 | 8,909 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 54 ms
50,036 KB |
testcase_01 | AC | 54 ms
50,360 KB |
testcase_02 | AC | 54 ms
50,092 KB |
testcase_03 | AC | 54 ms
49,856 KB |
testcase_04 | AC | 54 ms
50,156 KB |
testcase_05 | AC | 54 ms
50,060 KB |
testcase_06 | AC | 55 ms
50,080 KB |
testcase_07 | AC | 55 ms
50,060 KB |
testcase_08 | AC | 55 ms
50,048 KB |
testcase_09 | AC | 192 ms
56,148 KB |
testcase_10 | AC | 196 ms
56,568 KB |
testcase_11 | AC | 139 ms
55,696 KB |
testcase_12 | AC | 236 ms
56,412 KB |
testcase_13 | AC | 208 ms
57,208 KB |
testcase_14 | AC | 707 ms
75,944 KB |
testcase_15 | AC | 611 ms
66,940 KB |
testcase_16 | AC | 481 ms
76,044 KB |
testcase_17 | AC | 499 ms
80,376 KB |
testcase_18 | AC | 557 ms
82,900 KB |
testcase_19 | AC | 852 ms
75,276 KB |
ソースコード
import java.io.BufferedReader; import java.io.InputStreamReader; import java.util.ArrayList; import java.util.Arrays; import java.util.List; public class Main { public static void main(String[] args) throws Exception { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); String[] sa = br.readLine().split(" "); int n = Integer.parseInt(sa[0]); int m = Integer.parseInt(sa[1]); sa = br.readLine().split(" "); int[] a = new int[n]; for (int i = 0; i < n; i++) { a[i] = Integer.parseInt(sa[i]); } DSU uf = new DSU(n); for (int i = 0; i < m; i++) { sa = br.readLine().split(" "); int u = Integer.parseInt(sa[0]) - 1; int v = Integer.parseInt(sa[1]) - 1; uf.merge(u, v); } br.close(); int mod = 998244353; List<List<Integer>> gs = uf.groups(); long[] s = new long[n]; for (List<Integer> g : gs) { long sum = 0; for (int i : g) { sum += a[i]; } sum %= mod; for (int i : g) { s[i] = sum; } } long ans = 1; for (int i = 0; i < n; i++) { ans *= s[i]; ans %= mod; } System.out.println(ans); } } class DSU { private int n; private int[] parentOrSize; private int num; /** * n頂点0辺の無向グラフを作る。<br> * O(n) * * @param n 頂点数 */ public DSU(int n) { this.n = n; this.parentOrSize = new int[n]; Arrays.fill(parentOrSize, -1); num = n; } /** * 辺(a, b)を足す。<br> * ならしO(α(n)) * * @param a 頂点番号(0≦a<n) * @param b 頂点番号(0≦b<n) * @return 代表元 */ int merge(int a, int b) { assert 0 <= a && a < n : "a=" + a; assert 0 <= b && b < n : "b=" + b; int x = leader(a); int y = leader(b); if (x == y) { return x; } if (-parentOrSize[x] < -parentOrSize[y]) { int tmp = x; x = y; y = tmp; } parentOrSize[x] += parentOrSize[y]; parentOrSize[y] = x; num--; return x; } /** * 頂点a, bが連結かどうか。<br> * ならしO(α(n)) * * @param a 頂点番号(0≦a<n) * @param b 頂点番号(0≦b<n) * @return true:連結 false:非連結 */ boolean same(int a, int b) { assert 0 <= a && a < n : "a=" + a; assert 0 <= b && b < n : "b=" + b; return leader(a) == leader(b); } /** * 頂点aの属する連結成分の代表元を返す。<br> * ならしO(α(n)) * * @param a 頂点番号(0≦a<n) * @return 代表元 */ int leader(int a) { assert 0 <= a && a < n : "a=" + a; if (parentOrSize[a] < 0) { return a; } else { return parentOrSize[a] = leader(parentOrSize[a]); } } /** * 頂点aの属する連結成分の要素数を返す。<br> * ならしO(α(n)) * * @param a 頂点番号(0≦a<n) * @return 要素数 */ int size(int a) { assert 0 <= a && a < n : "a=" + a; return -parentOrSize[leader(a)]; } /** * 連結成分の数を返す。<br> * O(1) * * @return 連結成分数 */ int num() { return num; } /** * グラフを連結成分に分けた情報を返す。<br> * O(n) * * @return 「1つの連結成分の頂点番号のリスト」のリスト */ List<List<Integer>> groups() { int[] leaderBuf = new int[n]; int[] groupSize = new int[n]; for (int i = 0; i < n; i++) { leaderBuf[i] = leader(i); groupSize[leaderBuf[i]]++; } List<List<Integer>> result = new ArrayList<>(n); for (int i = 0; i < n; i++) { result.add(new ArrayList<>(groupSize[i])); } for (int i = 0; i < n; i++) { result.get(leaderBuf[i]).add(i); } result.removeIf(List::isEmpty); return result; } }