POJ2762 Going from u to v or from v to u? 强连通分量缩点+拓扑排序

#include<stdio.h>
#include<string.h>
#include<stack>
#include<queue>
#include<algorithm>
using namespace std;

int n, m, cnt, cnt1;
int head[1010], head1[1010];
int dfn[1010], low[1010], vis[1010], deep, k_color, color[1010];
stack<int>S;
int in[1010], flag;

struct Edge
{
	int to, next;
}edge[6010], edge1[6010];

void add(int a, int b)
{
	edge[++ cnt].to = b;
	edge[cnt].next = head[a];
	head[a] = cnt;
}

void add1(int a, int b)
{
	edge[++ cnt1].to = b;
	edge[cnt1].next = head1[a];
	head1[a] = cnt1;
}

void tarjan(int now)
{
	dfn[now] = low[now] = ++ deep;
	vis[now] = 1;
	S.push(now);
	for(int i = head[now]; i != -1; i = edge[i].next)
	{
		int to = edge[i].to;
		if(!dfn[to])
		{
			tarjan(to);
			low[now] = min(low[now], low[to]);
		}
		else if(vis[to])
		{
			low[now] = min(low[now], dfn[to]);
		}
	}
	if(dfn[now] == low[now])
	{
		k_color ++;
		while(1)
		{
			int temp = S.top();
			S.pop();
			color[temp] = k_color;
			vis[temp] = 0;
			if(temp == now)
				break;
		}
	}
}

void topo_sort()
{
	queue<int>Q;
	while(!Q.empty())
		Q.pop();
	for(int i = 1; i <= k_color; i ++)
		if(!in[i])
			Q.push(i);
	if(Q.size() > 1)
	{
		flag = 0;
		return ;
	}
	while(!Q.empty())
	{
		int temp = Q.front();
		Q.pop();
		for(int i = head1[temp]; i != -1; i = edge[i].next)
		{
			int to = edge[i].to;
			in[to] --;
			if(!in[to])
				Q.push(to);
			if(Q.size() > 1)
			{
				flag = 0;
				return ;
			}
		}
	}
}

int main()
{
	int T;
	scanf("%d", &T);
	while(T --)
	{
		flag = 1;
		cnt = deep = k_color = cnt1 = 0;
		memset(head, -1, sizeof(head));
		memset(in, 0, sizeof(in));
		memset(head1, -1, sizeof(head1));
		memset(dfn, 0, sizeof(dfn));
		memset(low, 0, sizeof(low));
		memset(vis, 0, sizeof(vis));
		scanf("%d%d", &n, &m);
		for(int i = 1; i <= m; i ++)
		{
			int a, b;
			scanf("%d%d", &a, &b);
			add(a, b);
		}
		for(int i = 1; i <= n; i ++)
			if(!dfn[i])
				tarjan(i);
		for(int i = 1; i <= n; i ++)//缩点 
		{
			for(int j = head[i]; j != -1; j = edge[j].next)
			{
				int to = edge[j].to;
				int x = color[i], y = color[to];
				if(x != y)
				{
					add1(x, y);
					in[y] ++;
				}
			}
		}
		topo_sort();
		if(flag)
			printf("Yes\n");
		else
			printf("No\n");
	}
	return 0;
}