백준 17455 kdh9949

https://www.acmicpc.net/problem/17455

연결이 되어 있더라도 갈 수 없는 간선은 모두 지워줍시다. 우리는 이제 단방향 그래프를 얻을 수 있습니다.

이 그래프를 k로 시작하는 DAG로 만들고 싶습니다. k가 아닌데 indegree가 0인 노드들을 모두 날려줍시다. 최악의 경우 d,h가 한번씩 떨어져 나간다고 해도 3*(V+E)의 복잡도로 k에서 시작하는 DAG를 만들어 줄 수 있습니다.

위상정렬된 결과를 이용하여 “가장 먼 리프 노드에서의 거리”를 구할 수 있습니다.

#include<bits/stdc++.h>
#define pb push_back
#define ss second
#define ff first
using namespace std;
const int SZ=200005;
vector<int> graph[SZ];
int nog[SZ];
int arr[SZ];
int kdh[SZ];
int ind[SZ];
int ch2int(char x){
	return x=='K'?0:(x=='D'?1:2);
}
bool cycle(int n){
	for(int i=1;i<=n;i++){
		if(nog[i]) continue;
		if(kdh[i]==0) return true;
	}
	return false;
}
int main(void){
	ios::sync_with_stdio(false); cin.tie(NULL);
	int n,m; cin>>n>>m;
	string s; cin>>s;
	for(int i=1;i<=n;i++) arr[i]=ch2int(s[i-1]);
	for(int i=0;i<m;i++){
		int x,y; cin>>x>>y;
		if((arr[x]+1)%3==arr[y]){
			graph[x].pb(y);
			ind[y]++;
		}
		if(arr[x]==(arr[y]+1)%3){
			graph[y].pb(x);
			ind[x]++;
		}
	}
	for(int k=0;k<3;k++){
		for(int i=1;i<=n;i++){
			if(nog[i]) continue;
			if(ind[i]==0 && arr[i]!=0){
				nog[i]=1;
				for(int nxt:graph[i]) ind[nxt]--;
			}
		}
	}
	queue<int> q;
	for(int i=1;i<=n;i++){
		if(nog[i]) continue;
		if(ind[i]==0){
			q.push(i);
			kdh[i]=1;
		}
	}
	while(!q.empty()){
		int f=q.front(); q.pop();
		for(int nxt:graph[f]){
			if(nog[nxt]) continue;
			kdh[nxt]=max(kdh[nxt],kdh[f]+1);
			ind[nxt]--;
			if(ind[nxt]==0) q.push(nxt);
		}
	}
	if(cycle(n)){
		cout<<-1;
		return 0;
	}
	int ans=0;
	for(int i=1;i<=n;i++) ans=max(ans,kdh[i]/3);
	cout<<(ans==0?-1:3*ans);
	return 0;
}