結果
問題 |
No.1605 Matrix Shape
|
ユーザー |
![]() |
提出日時 | 2021-07-17 00:19:45 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 417 ms / 2,000 ms |
コード長 | 2,210 bytes |
コンパイル時間 | 1,493 ms |
コンパイル使用メモリ | 179,588 KB |
実行使用メモリ | 17,408 KB |
最終ジャッジ日時 | 2024-07-06 11:52:22 |
合計ジャッジ時間 | 7,392 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
#include <bits/stdc++.h> using namespace std; typedef long long ll; const int INF = 1000000000; const ll INFL = 2e18; template <class T>bool chmin(T &a, T b){ if (a > b) { a = b; return true;} else return false;} template <class T>bool chmax(T &a, T b){ if (a < b) { a = b; return true;} else return false;} class UnionFind { public: vector<int> par; vector<int> siz; UnionFind() { } UnionFind(int sz_): par(sz_), siz(sz_, 1) { for(int i = 0;i < sz_;i++)par[i] = i; } void init(int sz_) { par.resize(sz_); siz.resize(sz_,1); for(int i = 0;i < sz_;i++)par[i] = i; } int root(int x) { while (par[x] != x) { x = par[x] = par[par[x]]; } return x; } bool merge(int x,int y) { x = root(x); y = root(y); if(x == y)return false; if(siz[x] < siz[y])swap(x, y); siz[x] += siz[y]; par[y] = x; return true; } bool issame(int x, int y) { return root(x) == root(y); } int size(int x) { return siz[root(x)]; } }; int main() { int N; cin >> N; vector<int> H(N); vector<int> W(N); map<int,int> mp; for (int i = 0;i < N;i++){ cin >> H[i] >> W[i]; mp[H[i]] = mp[W[i]] = 1; } int idx = 0; for (auto& p:mp) { p.second = idx; idx++; } UnionFind uf(mp.size()); vector<int> in_deg(mp.size(), 0); vector<int> out_deg(mp.size(), 0); for (int i = 0;i < N;i++) { H[i] = mp[H[i]], W[i] = mp[W[i]]; uf.merge(H[i], W[i]); in_deg[W[i]]++; out_deg[H[i]]++; } if (uf.size(0) != mp.size()) { cout << 0 << endl; return 0; } int cnt_0 = 0; int cnt_p1 = 0; int cnt_m1 = 0; for (int i = 0;i < mp.size();i++) { int deg = in_deg[i] - out_deg[i]; if (deg == 0)cnt_0++; if (deg == -1)cnt_m1++; if (deg == 1)cnt_p1++; } if (cnt_0 == mp.size()) { cout << mp.size() << endl; } else if(cnt_m1 == 1 && cnt_p1 == 1 && cnt_0 + cnt_m1 + cnt_p1 == mp.size()) { cout << 1 << endl; } else cout << 0 << endl; }