ACM: hdu 2647 Reward -拓扑排序

#include"algorithm"
#include"iostream"
#include"cstring"
#include"cstdlib"
#include"string"
#include"cstdio"
#include"vector"
#include"cmath"
#include"queue"
using namespace std;
#define memset(x,y) memset(x,y,sizeof(x))
#define memcpy(x,y) memcpy(x,y,sizeof(x))
#define MX 10005

int indegree[MX];
vector<int>next_v[MX];

void init() {
	for(int i=0; i<MX; i++)
		next_v[i].clear();
	memset(indegree,0);
}

int toposort(int n) {
	int p[MX];
	int cnt=0,money=0,num,summoney=n*888 ;
	while(cnt!=n) {
		num=0;
		for(int i=1; i<=n; i++) {
			if(!indegree[i]) {
				indegree[i]=-1;
				p[num]=i;
				num++;
			}
		}
		if(!num) {//没有员工，工资最高，不能排序，返回错误；
			return -1;
		} else {
			summoney+=money*num;
			money++;
			cnt+=num; //表示已经历遍了这么多个员工
			for(int i=0; i<num; i++) {
				for(int  j=0; j<next_v[p[i]].size(); j++) {
					indegree[next_v[p[i]][j]]--;
				}
			}
		}
	}
	return summoney;
}

int main() {
	int n,m,a,b;
	while(~scanf("%d%d",&n,&m)) {
		init();
		for(int qq=0; qq<m; qq++) {
			scanf("%d%d",&a,&b);
			int flag=0;
			for(int i=0; i<next_v[b].size(); i++) {
				if(next_v[b][i]==a)
					flag=1;
			}
			if(!flag) {
				indegree[a]++;
				next_v[b].push_back(a);
			}
		}
		printf("%d\n",toposort(n));
	}
	return 0;
}