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是一种比较一般化的套路…?
因为拓扑保证了更新顺序
1/* ***********************************************
2Author :111qqz
3Created Time :2017年11月07日 星期二 13时52分27秒
4File Name :3249.cpp
5************************************************ */
6
7#include <iostream>
8#include <cmath>
9#include <queue>
10#include <cstdio>
11#include <cstring>
12#define PB push_back
13#define fst first
14#define sec second
15#define lson l,m,rt<<1
16#define rson m+1,r,rt<<1|1
17#define ms(a,x) memset(a,x,sizeof(a))
18typedef long long LL;
19#define pi pair < int ,int >
20#define MP make_pair
21
22using namespace std;
23const double eps = 1E-8;
24const int dx4[4]={1,0,0,-1};
25const int dy4[4]={0,-1,1,0};
26const int inf = 0x3f3f3f3f;
27const int N=1E5+7;
28vector <int>edge[N];
29int in[N],out[N];
30int n,m;
31int dp[N];
32int val[N];
33void init()
34{
35 for ( int i = 0 ; i < N ; i++) edge[i].clear();
36 ms(in,0);
37 ms(out,0);
38 ms(dp,0xca);
39 // cout<<"dp:"<<dp[1]<<endl;
40}
41void topo()
42{
43 queue<int>Q;
44 for ( int i = 1 ; i <= n ; i++)
45 if (in[i]==0) Q.push(i),dp[i] = val[i];
46
47 while (!Q.empty())
48 {
49 int cur = Q.front();
50// cout<<"cur:"<<cur<<endl;
51 Q.pop();
52 int siz = edge[cur].size();
53 for ( int i = 0 ; i < siz ; i++)
54 {
55 int v = edge[cur][i];
56 dp[v] = max(dp[v],dp[cur]+val[v]);
57 in[v]--;
58 if (in[v]==0) Q.push(v);
59 }
60 }
61}
62int main()
63{
64 #ifndef ONLINE_JUDGE
65 freopen("./in.txt","r",stdin);
66 #endif
67 while (~scanf("%d %d",&n,&m))
68 {
69 init();
70 for ( int i = 1 ; i <= n ; i++) scanf("%d",&val[i]);
71 while (m--)
72 {
73 int x,y;
74 scanf("%d %d",&x,&y);
75// cout<<"x:"<<x<<" y:"<<y<<endl;
76 edge[x].PB(y);
77 out[x]++;
78 in[y]++;
79 }
80 topo();
81 int ans = -inf;
82 for ( int i = 1 ; i <= n ; i++) if (!out[i]&&dp[i]>ans) ans = dp[i];
83 printf("%d\n",ans);
84 }
85
86
87
88 #ifndef ONLINE_JUDGE
89 fclose(stdin);
90 #endif
91 return 0;
92}