Official
D - Tying Rope Editorial by cn449
\(N\) 本のロープをグラフの \(N\) 個の頂点とし、ロープ \(a\) と \(b\) を結ぶことを 頂点 \(a\) と頂点 \(b\) を結ぶ辺に対応させると、元の問題は次のようなグラフ上の問題に言い換えることができます。
- \(N\) 頂点 \(M\) 辺のグラフがあり、\(i\) 番目の辺は頂点 \(A_i\) と頂点 \(C_i\) を結んでいる。また、各頂点の次数は \(2\) 以下である。グラフの各連結成分はサイクルまたはパスとなるが、これらの個数をそれぞれ求めよ。
連結成分がサイクルをなすことは連結成分内のすべての頂点の次数が \(2\) であることと同値であるため、あらかじめ各頂点の次数を記録しておき、各連結成分に対してその連結成分がサイクルとなるかを BFS などを用いて判定することで \(O(N+M)\) でこの問題を解くことができました。
実装例
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
int main() {
int n, m;
cin >> n >> m;
vector<vector<int>> graph(n, vector<int>());
vector<int> deg(n);
for (int i = 0; i < m; i++) {
int a, c;
char b, d;
cin >> a >> b >> c >> d;
a--; c--;
graph[a].push_back(c);
graph[c].push_back(a);
deg[a]++; deg[c]++;
}
int x = 0, y = 0;
vector<bool> used(n);
for (int i = 0; i < n; i++) {
if (!used[i]) {
queue<int> que;
used[i] = true;
que.push(i);
bool f = true;
while (!que.empty()) {
int q = que.front(); que.pop();
if (deg[q] != 2) f = false;
for (int v : graph[q]) {
if (!used[v]) {
que.push(v);
used[v] = true;
}
}
}
if (f) x++;
else y++;
}
}
cout << x << ' ' << y << '\n';
}
posted:
last update: