图上DP POJ3249

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

int val[100100];
int head[100100];
int dp[100100];
int num;
bool in[100010];
bool en[100010];

struct edge
{
int fro,to,next;
}e[10000100];

void addEdge(int _fro,int _to)
{
num++;
e[num].fro=_fro;
e[num].to=_to;
e[num].next=head[_fro];
head[_fro]=num;
}

int dfs(int x)
{
if(dp[x]!=0)
return dp[x];
dp[x]=-(1<<29);
for(int i=head[x];i!=-1;i=e[i].next)
{
int v=e[i].to;
dfs(v);
dp[x]=max(dp[x],dp[v]+val[x]);
}
return dp[x];
}

int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
num=0;
memset(head,-1,sizeof(head));
for(int i=0;i<=n;i++)
{
in[i]=true;
en[i]=true;
}
for(int i=1;i<=n;i++)
scanf("%d",&val[i]);
int a,b;
for(int i=1;i<=m;i++)
{
scanf("%d%d",&a,&b);
en[b]=false;
in[a]=false;
addEdge(b,a);
}
int ans=-(1<<29);
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
{
if(en[i])
dp[i]=val[i];
}
for(int i=1;i<=n;i++)
{
if(in[i])
{
ans=max(ans,dfs(i));
}
}
cout<<ans<<endl;
}
return 0;
}