BC 2015百度之星程序设计大赛 - 热身赛(仅供熟悉比赛环境)(Harry and Magical Computer-topsort)[Template:topsort]

Harry and Magical Computer

Accepts: 402

Submissions: 1588

Time Limit: 2000/1000 MS (Java/Others)

Memory Limit: 32768/32768 K (Java/Others)

问题描述

作为年度优秀魔法学员的奖赏，哈利得到了一台具有魔力的计算机。这台计算机一旦开始处理某个任务，就会一直处理到这个任务结束为止(所以你可以认为它是单线程的)。有一天，这台计算机得到了n个任务要处理，分别标号1到n。这n个任务之间又有一些依赖关系，假如存在依赖关系(a, b)，那么要处理a任务，必须先将b任务完成。现在哈利得到了所有的这些依赖关系，一共m个。他想知道，这台计算机能否完成所有的任务。

输入描述

多组输入数据
每组数据第一行由n m组成，分别代表任务数和依赖关系数。
    
     1≤n≤100,1≤m≤10000 
    


接下来m行，每行两个数a b，表示一组依赖关系。
    
     1≤a,b≤n 
    

输出描述

每组输出一行，"YES"(没有引号)表示能完成所有任务。
"NO"(没有引号)表示不能。

输入样例

3 2
3 1
2 1
3 3
3 2
2 1
1 3

输出样例

YES
NO

拓扑排序

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<functional>
#include<iostream>
#include<cmath>
#include<cctype>
#include<ctime>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define RepD(i,n) for(int i=n;i>=0;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define Forpiter(x) for(int &p=iter[x];p;p=next[p])  
#define Lson (x<<1)
#define Rson ((x<<1)+1)
#define MEM(a) memset(a,0,sizeof(a));
#define MEMI(a) memset(a,127,sizeof(a));
#define MEMi(a) memset(a,128,sizeof(a));
#define INF (2139062143)
#define F (100000007)
#define MAXN (1500+100) 
typedef long long ll;
ll mul(ll a,ll b){return (a*b)%F;}
ll add(ll a,ll b){return (a+b)%F;}
ll sub(ll a,ll b){return (a-b+(a-b)/F*F+F)%F;}
void upd(ll &a,ll b){a=(a%F+b%F)%F;}



int n,indegree[MAXN];
int f[MAXN][MAXN];
bool b[MAXN];
int q[MAXN*4];
void topsort()    
{    
    int head_=1,tail=0;
	int fl=0,flm=0;    
    Fork(i,1,n)  
        if (indegree[i]==0)     
        {    
            q[++tail]=i;b[i]=1; ++fl;    
        }    
    if (fl>1) flm=1;
    
    while (head_<=tail)    
    {    
    	fl=0;
        int now=q[head_];    
        Fork(v,1,n)  
            while (f[now][v])  
            {    
                indegree[v]--;f[now][v]--;    
                if (indegree[v]==0)    
                {    
                    q[++tail]=v;b[v]=1; ++fl;    
                }                                       
            }       
        head_++;    
        
		if (fl>1) flm=1;
            
            
    }         
    if (tail==n)   
    {
    	cout<<"YES\n";
	} 
	else {cout<<"NO\n";
	};  
      
}    

int m;

int main()
{
//	freopen("1001.in","r",stdin);
//	freopen(".out","w",stdout);
	
	
	while (scanf("%d%d",&n,&m)==2)
	{
		MEM(f) MEM(indegree) MEM(q) MEM(b)
		For(i,m)
		{
			int a,b;
			scanf("%d%d",&a,&b);
			indegree[a]++;f[b][a]++;
		}
		topsort();
	}
	
			
	return 0;
}