Problem Solving
Algorithm in A..Z - 백준 3648 아이돌
강태종
2021. 7. 9. 18:16
https://www.acmicpc.net/problem/3648
3648번: 아이돌
각 테스트 케이스에 대해서, 상근이를 포함해, 다음 라운드 진출 목록을 심사위원의 의심 없이 만들 수 있으면 'yes'를, 없으면 'no'를 출력한다.
www.acmicpc.net
접근
Tarjan 알고리즘으로 SCC를 구현하면 특성상 하위의 SCC가 먼저 결정된다. 그렇기 때문에 x < ¬x 를 만족하면 x가 참이된다.
상근이가 합격하려면 ¬x가 False면 합격이기 때문에 그래프를 구현할 때 역방향으로 그래프를 구현하여 ¬x가 False가 될 수 있는지 확인한다. (명제의 대우는 동치이기 때문에 가능하다.)
코드
#include <bits/stdc++.h>
using namespace std;
constexpr int NONE = -1;
int n, m;
vector<vector<int>> graph;
inline int convert(int value) {
return value <= n ? value + n : value - n;
}
int order;
stack<int> st;
vector<int> sccNumber, isVisited;
vector<vector<int>> scc;
int dfs(int index) {
st.push(index);
isVisited[index] = order++;
int value = isVisited[index];
for (auto &next : graph[index]) {
if (isVisited[next] == NONE) {
value = min(value, dfs(next));
} else if (sccNumber[next] == NONE) {
value = min(value, isVisited[next]);
}
}
if (value == isVisited[index]) {
vector<int> cycle;
while (!st.empty()) {
int i = st.top(); st.pop();
cycle.push_back(i);
sccNumber[i] = (int)scc.size();
if (i == index) {
break;
}
}
scc.push_back(cycle);
}
return value;
}
inline void init() {
graph = vector<vector<int>>(2*n + 1);
order = 0;
st = stack<int>();
sccNumber = isVisited = vector<int>(2*n + 1, NONE);
scc = vector<vector<int>>();
}
int main() {
ios::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr);
while (cin >> n >> m) {
init();
while (m--) {
int s, e;
cin >> s >> e;
s = s < 0 ? convert(-s) : s;
e = e < 0 ? convert(-e) : e;
graph[convert(s)].push_back(e);
graph[convert(e)].push_back(s);
}
for (int i = 1;i <= 2*n;++i) {
if (isVisited[i] == NONE) {
dfs(i);
}
}
bool flag = true;
for (int i = 1;i <= n;++i) {
if (sccNumber[i] == sccNumber[convert(i)]) {
flag = false;
break;
}
}
if (sccNumber[convert(1)] < sccNumber[1]) {
flag = false;
}
cout << (flag ? "yes" : "no") << "\n";
}
}