POJ3249:Test for Job

传送门

很简单的一道题，被卡了几次，死于答案非法统计。

题意是求图里的一条最长的路径满足起点的入度和终点的出度都是0，而且图是DAG。

既然是DAG求最长路，DP即可。搞出拓扑序，逆序DP，然后统计所有入度为0的点的答案。

//POJ 3249
//by Cydiater
//2016.10.26
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <map>
#include <ctime>
#include <cmath>
#include <cstdlib>
#include <cstdio>
#include <iomanip>
#include <bitset>
using namespace std;
#define ll long long
#define up(i,j,n)		for(int i=j;i<=n;i++)
#define down(i,j,n)		for(int i=j;i>=n;i--)
#define cmax(a,b) a=max(a,b);
const int MAXN=1e6+5;
const int oo=2000000001;
inline int read(){
	char ch=getchar();int x=0,f=1;
	while(ch>'9'||ch<'0'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
int Value[MAXN],LINK[MAXN],len=0,head,tail,indu[MAXN],N,M,outdu[MAXN],q[MAXN],f[MAXN],ans=-oo,Indu[MAXN];
struct edge{
	int y,next;
}e[MAXN];
namespace solution{
	inline void insert(int x,int y){e[++len].next=LINK[x];LINK[x]=len;e[len].y=y;}
	void init(){
		memset(Indu,0,sizeof(Indu));
		memset(indu,0,sizeof(indu));
		memset(outdu,0,sizeof(outdu));
		memset(LINK,0,sizeof(LINK));
		len=0;
		up(i,1,N)Value[i]=read();
		up(i,1,M){
			int x=read(),y=read();
			insert(x,y);
			outdu[x]++;indu[y]++;Indu[y]++;
		}
	}
	void slove(){
		head=1;tail=0;
		memset(f,-10,sizeof(f));
		up(i,1,N)if(!indu[i])q[++tail]=i;
		for(;head<=tail;head++){
			int node=q[head];
			for(int i=LINK[node];i;i=e[i].next)
				if(--indu[e[i].y]==0)q[++tail]=e[i].y;
		}
		ans=-oo;
		up(i,1,N)if(outdu[i]==0)f[i]=Value[i];
		down(i,tail,1){
			int node=q[i];
			for(int j=LINK[node];j;j=e[j].next)cmax(f[node],f[e[j].y]+Value[node]);
			if(Indu[node]==0)cmax(ans,f[node]);
		}
	}
	void output(){
		if(ans==-oo)ans=0;
		printf("%d\n",ans);
	}
}
int main(){
	//freopen("input.in","r",stdin);
	using namespace solution;
	while(scanf("%d %d",&N,&M)!=EOF){
		init();
		slove();
		output();
	}
	return 0;
}