結果
問題 | No.3093 Safe Infection |
ユーザー |
|
提出日時 | 2025-03-09 01:30:23 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 55 ms / 2,000 ms |
コード長 | 1,634 bytes |
コンパイル時間 | 3,750 ms |
コンパイル使用メモリ | 286,064 KB |
実行使用メモリ | 11,008 KB |
最終ジャッジ日時 | 2025-03-09 01:30:32 |
合計ジャッジ時間 | 8,303 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 70 |
ソースコード
#include <bits/stdc++.h> using namespace std; struct UnionFind { vector<int> par, rank, siz, level; UnionFind(int n, const vector<int>& v) : par(n, -1), rank(n, 0), siz(n, 1), level(v.begin(), v.end()) {} int root(int a) { if(par[a] == -1) return a; else return par[a] = root(par[a]); } bool same(int a, int b) { return root(a) == root(b); } bool unite(int a, int b) { int ra = root(a), rb = root(b); if(ra == rb) return false; if(rank[ra] < rank[rb]) swap(ra, rb); par[rb] = ra; if(rank[ra] == rank[rb]) rank[ra]++; siz[ra] += siz[rb]; level[ra] = max(level[ra], level[rb]); return true; } int size(int a) { return siz[root(a)]; } int get_level(int a) { return level[root(a)]; } }; int main() { cin.tie(nullptr); ios::sync_with_stdio(false); int N, M, K; cin >> N >> M >> K; vector<int> A(N); for(int i = 0; i < N; i++) { cin >> A[i]; } vector<vector<int>> G(N); for(int i = 0; i < M; i++) { int u, v; cin >> u >> v; u--; v--; G[u].push_back(v); G[v].push_back(u); } UnionFind uf(N, A); vector<int> ord(N); iota(ord.begin(), ord.end(), 0); sort(ord.begin(), ord.end(), [&](int i, int j) { return A[i] < A[j]; }); for(int v : ord) { for(int nv : G[v]) { if(A[v] >= uf.get_level(nv) && (A[v] - uf.get_level(nv)) <= K) uf.unite(v, nv); } } if(uf.size(0) == N) cout << "Yes\n"; else cout << "No\n"; return 0; }