結果
問題 | No.2494 Sum within Components |
ユーザー | risujiroh |
提出日時 | 2023-10-06 21:30:30 |
言語 | C++23 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 125 ms / 2,000 ms |
コード長 | 3,336 bytes |
コンパイル時間 | 3,330 ms |
コンパイル使用メモリ | 261,712 KB |
実行使用メモリ | 14,008 KB |
最終ジャッジ日時 | 2023-10-06 21:30:35 |
合計ジャッジ時間 | 5,055 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,376 KB |
testcase_01 | AC | 2 ms
4,376 KB |
testcase_02 | AC | 2 ms
4,376 KB |
testcase_03 | AC | 2 ms
4,376 KB |
testcase_04 | AC | 1 ms
4,376 KB |
testcase_05 | AC | 2 ms
4,376 KB |
testcase_06 | AC | 1 ms
4,376 KB |
testcase_07 | AC | 2 ms
4,376 KB |
testcase_08 | AC | 1 ms
4,376 KB |
testcase_09 | AC | 6 ms
4,380 KB |
testcase_10 | AC | 8 ms
4,380 KB |
testcase_11 | AC | 3 ms
4,380 KB |
testcase_12 | AC | 11 ms
4,380 KB |
testcase_13 | AC | 8 ms
4,376 KB |
testcase_14 | AC | 99 ms
11,284 KB |
testcase_15 | AC | 107 ms
10,040 KB |
testcase_16 | AC | 41 ms
8,128 KB |
testcase_17 | AC | 31 ms
8,512 KB |
testcase_18 | AC | 39 ms
9,312 KB |
testcase_19 | AC | 125 ms
14,008 KB |
ソースコード
#if __INCLUDE_LEVEL__ == 0 #include __BASE_FILE__ namespace { using Fp = atcoder::modint998244353; void solve() { int n, m; cin >> tie(n, m); vector<Fp> a(n); cin >> a; vector<vector<int>> g(n); while (m--) { int i, j; cin >> tie(i, j); --i, --j; g[i].push_back(j); g[j].push_back(i); } vector<bool> vis(n); Fp ans = 1; for (int s : rep(n)) { if (vis[s]) { continue; } vis[s] = true; vector<int> q{s}; Fp sum = 0; for (int qi = 0; qi < len(q); ++qi) { int i = q[qi]; sum += a[i]; for (int j : g[i]) { if (!vis[j]) { vis[j] = true; q.push_back(j); } } } for (int _ = len(q); _--;) { ans *= sum; } } print(ans); } } // namespace int main() { ios::sync_with_stdio(false); cin.tie(nullptr); solve(); } #else // __INCLUDE_LEVEL__ #include <bits/stdc++.h> using namespace std; #include <atcoder/modint> template <class T, class U = T> bool chmin(T& x, U&& y) { return y < x && (x = forward<U>(y), true); } template <class T, class U = T> bool chmax(T& x, U&& y) { return x < y && (x = forward<U>(y), true); } namespace std { template <class T1, class T2> istream& operator>>(istream& is, pair<T1, T2>& p) { return is >> p.first >> p.second; } template <class... Ts> istream& operator>>(istream& is, tuple<Ts...>& t) { return apply([&is](auto&... xs) -> istream& { return (is >> ... >> xs); }, t); } template <class... Ts> istream& operator>>(istream& is, tuple<Ts&...>&& t) { return is >> t; } template <class R, enable_if_t<!is_convertible_v<R, string>>* = nullptr> auto operator>>(istream& is, R&& r) -> decltype(is >> *begin(r)) { for (auto&& e : r) { is >> e; } return is; } template <class T1, class T2> ostream& operator<<(ostream& os, const pair<T1, T2>& p) { return os << p.first << ' ' << p.second; } template <class... Ts> ostream& operator<<(ostream& os, const tuple<Ts...>& t) { auto f = [&os](const auto&... xs) -> ostream& { [[maybe_unused]] auto sep = ""; ((os << exchange(sep, " ") << xs), ...); return os; }; return apply(f, t); } template <class R, enable_if_t<!is_convertible_v<R, string_view>>* = nullptr> auto operator<<(ostream& os, R&& r) -> decltype(os << *begin(r)) { auto sep = ""; for (auto&& e : r) { os << exchange(sep, " ") << e; } return os; } } // namespace std namespace atcoder { template <class T, internal::is_modint_t<T>* = nullptr> istream& operator>>(istream& is, T& x) { int v; is >> v; x = T::raw(v); return is; } template <class T, internal::is_modint_t<T>* = nullptr> ostream& operator<<(ostream& os, const T& x) { return os << x.val(); } } // namespace atcoder template <class... Ts> void print(Ts&&... xs) { cout << tie(xs...) << '\n'; } inline auto rep(int l, int r) { return views::iota(min(l, r), r); } inline auto rep(int n) { return rep(0, n); } inline auto rep1(int l, int r) { return rep(l, r + 1); } inline auto rep1(int n) { return rep(1, n + 1); } inline auto per(int l, int r) { return rep(l, r) | views::reverse; } inline auto per(int n) { return per(0, n); } inline auto per1(int l, int r) { return per(l, r + 1); } inline auto per1(int n) { return per(1, n + 1); } inline auto len = ranges::ssize; #endif // __INCLUDE_LEVEL__