알고리즘 공부/Baekjoon online judge

#24526 전화돌리기

djs100201 2022. 2. 28. 20:05

공식 해설을 봤는데, 나처럼 푼 사람이 없는거 같아서 올려본다.

 

보통 아름다운 풀이는 간선을 뒤집는다는 아이디어 이다. 

이제 무지성 풀이는 scc/위상정렬을 잘 쓰는 풀이가 있다.

 

나는 쉬운 무지성 풀이로 풀었는데, 그냥 실제로 그래프에서 사이클 찾아주면 된다.

#include<bits/stdc++.h>

//#define double long double
//T.erase(unique(T.begin(),T.end()),T.end());
//written by djs100201

#define all(v) v.begin(),v.end()
using namespace std;
using ll = long long;
using P = pair<ll, ll>;
using PP = pair<ll, P>;
const ll n_ = 1e6 + 100, inf = 1e18, mod = 1e9 + 7, sqrtN = 320;
ll dy[4] = { -1,0,1,0 }, dx[4] = { 0,1,0,-1 };
ll n, m, k, tc = 1, a, b, c, d, sum, x, y, z, base = 1, ans, t_c;

vector<ll>v[n_];
ll now[n_], dp[n_];
ll dfs(ll x) {
	if (dp[x] != -1)return dp[x];
	if (now[x] == -1)return 0;
	now[x] = -1;
	ll ret = 1;
	for (auto nxt : v[x]) {
		if (dfs(nxt) != 1) {
			ret = 0;
			break;
		}
	}
	now[x] = 0;
	return dp[x] = ret;
}
void solve() {
	cin >> n >> m;
	while(m--) {
		cin >> a >> b;
		v[a].push_back(b);	
	}
	memset(dp, -1, sizeof(dp));
	for (int i = 1; i <= n; i++) {
		if (dfs(i) == 1)ans++;
	}
	cout << ans;
}

int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0), cout.tie(0);
	//cin >> tc;
	while (tc--)solve();
}

이제 dp랑 탐색을 잘 섞어주면 아주 쉽게 구현할 수 있는 문제이다.