poj 3249 Test for Job (拓扑排序+dp)

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

题意:

给一个DAG,现要从一条入度为0的点到一个出度为0的点,问最大点权和。

思路:

其实比较容易想到搜...不过复杂度会炸?

由于到一个点的最大点权和,需要更新完所有到达它的路线之后才能确定。

容易联想到拓扑排序,我们可以在拓扑排序的同时做dp

dp[v] = max(dp[v],dp[u]+a[v]),初始化对于入度为0的点,dp[i] = val[i].

其实拓扑+dp是一种比较一般化的套路...?

因为拓扑保证了更新顺序

/* ***********************************************
Author :111qqz
Created Time :2017年11月07日 星期二 13时52分27秒
File Name :3249.cpp
************************************************ */
 1#include <iostream>
 2#include <cmath>
 3#include <queue>
 4#include <cstdio>
 5#include <cstring>
 6#define PB push_back
 7#define fst first
 8#define sec second
 9#define lson l,m,rt<<1
10#define rson m+1,r,rt<<1|1
11#define ms(a,x) memset(a,x,sizeof(a))
12typedef long long LL;
13#define pi pair < int ,int >
14#define MP make_pair
 1using namespace std;
 2const double eps = 1E-8;
 3const int dx4[4]={1,0,0,-1};
 4const int dy4[4]={0,-1,1,0};
 5const int inf = 0x3f3f3f3f;
 6const int N=1E5+7;
 7vector <int>edge[N];
 8int in[N],out[N];
 9int n,m;
10int dp[N];
11int val[N];
12void init()
13{
14    for ( int i = 0 ; i < N ; i++) edge[i].clear();
15    ms(in,0);
16    ms(out,0);
17    ms(dp,0xca);
18 //   cout<<"dp:"<<dp[1]<<endl;
19}
20void topo()
21{
22    queue<int>Q;
23    for ( int i = 1 ; i <= n ; i++)
24    if (in[i]==0) Q.push(i),dp[i] = val[i];
 1    while (!Q.empty())
 2    {
 3    int cur = Q.front();
 4//  cout<<"cur:"<<cur<<endl;
 5    Q.pop();
 6    int siz = edge[cur].size();
 7    for ( int i = 0 ; i < siz ; i++)
 8    {
 9        int v = edge[cur][i];
10        dp[v] = max(dp[v],dp[cur]+val[v]);
11        in[v]--;
12        if (in[v]==0) Q.push(v);
13    }
14    }
15}
16int main()
17{
18    #ifndef  ONLINE_JUDGE 
19    freopen("./in.txt","r",stdin);
20  #endif
21    while (~scanf("%d %d",&n,&m))
22    {
23        init();
24        for ( int i = 1 ; i <= n ; i++) scanf("%d",&val[i]);
25        while (m--)
26        {
27        int x,y;
28        scanf("%d %d",&x,&y);
29//      cout<<"x:"<<x<<" y:"<<y<<endl;
30        edge[x].PB(y);
31        out[x]++;
32        in[y]++;
33        }
34        topo();
35        int ans = -inf;
36        for ( int i = 1 ; i <= n ; i++) if (!out[i]&&dp[i]>ans) ans = dp[i];
37        printf("%d\n",ans);
38    }
1  #ifndef ONLINE_JUDGE  
2  fclose(stdin);
3  #endif
4    return 0;
5}