C. Online Courses In BSU(dfs+拓扑序)

题目2

题意：

    一共有n门课，每门课都有若干门先导课程。要求输出修习k门指定的课要修习的课程。
    $1\le k\le n\le 10^5$

分析：

    经典拓扑背景，建一张有向图。由于要求满足k个点，显然就是一个超级源点，递归处理即可。处理当前节点x时，先处理它相连的节点，最后再处理本身。用vis表示点的状态，vis[i]=1表示需要被处理且正在处理其先导课程，vis[i]=2表示i这个节点已经被处理完了。当递归到vis[i]=1时会死循环，即有环，输出-1。递归到vis[i]=2直接返回即可。

#include <iostream>
#include <vector>
using namespace std;

vector<int> g[100005],ans;
int vis[100005],flag = 0;

void dfs(int x)
{
	if( vis[x] == 2 ) return;
	if( vis[x] == 1 )
	{
		flag = 1;
		return;
	}
	vis[x] = 1;
	for (int i = 0; i < g[x].size(); i++)
	{
		int t = g[x][i];
		dfs(t);
	}
	if( x != 0 ) 
	{
		ans.push_back(x); 
		vis[x] = 2; 
	}
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int n,k;
	cin >> n >> k;
	for (int i = 1; i <= k; i++)
	{
		int x;
		cin >> x;
		g[0].push_back(x); 
	} 
	for (int i = 1; i <= n; i++)
	{
		int t;
		cin >> t;
		for (int j = 1; j <= t; j++)
		{
			int x;
			cin >> x;
			g[i].push_back(x); 
		}
	}
	dfs(0);
	if( flag ) cout << -1 << '\n';
	else
	{
		cout << ans.size() << '\n';
		for (int i = 0; i < ans.size(); i++)
		{
			cout << ans[i];
			if( i == ans.size() - 1 ) cout << '\n';
			else cout << ' ';
		}
	}
	return 0;
}