POJ 3249 Test for a job dp最长路

题目链接

http://poj.org/problem?id=3249

题意

某人从一个“源”城市，沿着可以走的路，一直走到“汇”城市，每个城市都没有返回上一个城市的路，且每个城市有个权值，可正可负，求从“源”城市到达“汇”城市所能获得的最大权值。

思路

题目的情景很奇怪，城市居然单向连通，还是个dag图。题目意思是，从入度为0的点走到出度为0的点，能获得的最大权值。图是DAG图，dp就可以了。

#include<cstdio> // dag 上跑最长路 
#include<queue>
#include<iostream>
#include<vector>
#include<map>
#include<cstring>
#include<string>
#include<set>
#include<stack>
#include<algorithm>
#define cle(a) memset(a,0,sizeof(a))
#define inf(a) memset(a,0x3f,sizeof(a))
#define ll long long
#define Rep(i,a,n) for(int i=a;i<=n;i++)
using namespace std;
const int INF = ( 2e9 ) + 2;
const ll maxn = 1e5+10;
int head[maxn],In[maxn],Out[maxn];
int p[maxn];
int dp[maxn];
struct edge
{
    int to,next;
}e[20*maxn];
int tot;
void addedge(int u,int v)
{
    e[tot].to=v;
    e[tot].next=head[u];
    head[u]=tot++;
}
int find(int u)
{
    int &ans = dp[u];
    if(ans!=-INF)return ans;
    for(int i=head[u];i!=-1;i=e[i].next)
    {
        int v=e[i].to;
        ans = max(ans,p[u]+find(v));
    }
    if(ans==-INF)return dp[u]=p[u];     // 说明这个点就是出度为0的点 
    return ans;
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        memset(head,-1,sizeof(head));
        memset(In,0,sizeof(In));
        memset(Out,0,sizeof(Out));
        for(int i=1;i<=n;i++)dp[i]=-INF;
        tot=0;
        for(int i=1;i<=n;i++)
        scanf("%d",&p[i]);

        for(int i=0;i<m;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            addedge(u,v);
            In[v]++;
        }
        int ans=-INF;
        for(int i=1;i<=n;i++)
        if(In[i]==0)
        ans=max(ans,find(i));

        printf("%d\n",ans); 
    }
}