結果
| 問題 | No.2494 Sum within Components |
| ユーザー |
 ゆにぽけ
|
| 提出日時 | 2023-10-07 10:04:32 |
| 言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
AC
|
| 実行時間 | 53 ms / 2,000 ms |
| コード長 | 4,583 bytes |
| コンパイル時間 | 1,177 ms |
| コンパイル使用メモリ | 127,808 KB |
| 最終ジャッジ日時 | 2025-02-17 06:02:34 |
|
ジャッジサーバーID （参考情報） |
judge2 / judge2 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 17 |
ソースコード
#include<iostream>#include<vector>#include<algorithm>#include<cstring>#include<cassert>#include<cmath>#include<ctime>#include<iomanip>#include<numeric>#include<stack>#include<queue>#include<map>#include<unordered_map>#include<set>#include<unordered_set>#include<bitset>#include<random>using namespace std;template<int m> struct modint{private:unsigned int value;static constexpr int mod() {return m;}public:constexpr modint(const long long x = 0) noexcept{long long y = x;if(y < 0 || y >= mod()){y %= mod();if(y < 0) y += mod();}value = (unsigned int)y;}constexpr unsigned int val() noexcept {return value;}constexpr modint &operator+=(const modint &other) noexcept{value += other.value;if(value >= mod()) value -= mod();return *this;}constexpr modint &operator-=(const modint &other) noexcept{unsigned int x = value;if(x < other.value) x += mod();x -= other.value;value = x;return *this;}constexpr modint &operator*=(const modint &other) noexcept{unsigned long long x = value;x *= other.value;value = (unsigned int) (x % mod());return *this;}constexpr modint &operator/=(const modint &other) noexcept{return *this *= other.inverse();}constexpr modint inverse() const noexcept{assert(value);long long a = value,b = mod(),x = 1,y = 0;while(b){long long q = a/b;a -= q*b; swap(a,b);x -= q*y; swap(x,y);}return modint(x);}constexpr modint power(long long N) const noexcept{assert(N >= 0);modint p = *this,ret = 1;while(N){if(N & 1) ret *= p;p *= p;N >>= 1;}return ret;}constexpr modint operator+() {return *this;}constexpr modint operator-() {return modint() - *this;}constexpr modint &operator++(int) noexcept {return *this += 1;}constexpr modint &operator--(int) noexcept {return *this -= 1;}friend modint operator+(const modint& lhs, const modint& rhs) {return modint(lhs) += rhs;}friend modint operator-(const modint& lhs, const modint& rhs) {return modint(lhs) -= rhs;}friend modint operator*(const modint& lhs, const modint& rhs) {return modint(lhs) *= rhs;}friend modint operator/(const modint& lhs, const modint& rhs) {return modint(lhs) /= rhs;}friend ostream &operator<<(ostream &os,const modint &x) {return os << x.value;}};using mint = modint<998244353>;//using mint = modint<1000000007>;template<class S>struct combination{vector<S> f,invf;combination(int N = 0) : f(1,1),invf(1,1){update(N);}void update(int N){if((int)f.size() > N) return;int pi = (int)f.size();N = max(N,pi*2);f.resize(N+1),invf.resize(N+1);for(int i = pi;i <= N;i++) f[i] = f[i-1]*i;invf[N] = S(1)/f[N];for(int i = N-1;i >= pi;i--) invf[i] = invf[i+1]*(i+1);}S factorial(int N){update(N);return f[N];}S invfactorial(int N){update(N);return invf[N];}S P(int N,int K){assert(0 <= K && K <= N);update(N);return f[N]*invf[N-K];}S C(int N,int K){assert(0 <= K && K <= N);update(N);return f[N]*invf[K]*invf[N-K];}S H(int N,int K){if(!N) return K == 0 ? 1:0;return C(N+K-1,K);}};#include<algorithm>#include<vector>#include<cassert>using namespace std;struct UnionFind{private:int n;vector<int> par,siz;public:UnionFind(int n) :n(n),par(n,-1),siz(n,1) {}int root(int u){assert(0 <= u && u < n);return (par[u] < 0 ? u:par[u] = root(par[u]));}bool same(int u,int v){assert(0 <= u && u < n && 0 <= v && v < n);return root(u) == root(v);}bool unite(int u,int v){assert(0 <= u && u < n && 0 <= v && v < n);u = root(u),v = root(v);if(u == v) return false;if(siz[u] < siz[v]) swap(u,v);siz[u] += siz[v];par[v] = u;return true;}int size(int u){assert(0 <= u && u < n);return siz[root(u)];}vector<vector<int>> components(){vector<vector<int>> ret(n);for(int u = 0;u < n;u++) ret[root(u)].push_back(u);ret.erase(remove_if(ret.begin(),ret.end(),[](vector<int> v) { return v.empty();}),ret.end());return ret;}};int N,M,A[2 << 17];long long v[2 << 17];void solve(){cin >> N >> M;for(int i = 0;i < N;i++) cin >> A[i];UnionFind uf(N);for(int i = 0;i < M;i++){int u,v;cin >> u >> v;u--; v--;uf.unite(u,v);}for(int i = 0;i < N;i++) v[uf.root(i)] += A[i];mint ans = 1;for(int i = 0;i < N;i++) ans *= v[uf.root(i)];cout << ans << endl;}int main(){ios::sync_with_stdio(false);cin.tie(nullptr);int T = 1;//cin >> T;while(T--) solve();}