POJ 3249 Test for Job

题目链接~~>

做题感悟：开始做时没处理好，还是按照深搜的思路写了，果断超时，后来改为记忆化才过。

解题思路：从起点开始深搜，在搜每个点的时候要记录最优解，如果最优解已经存在就直接返回dp [ i ] = w[ i ] + dfs ( j )  ; ( i 与j 相连)  。

代码：

#include<stdio.h>
#include<iostream>
#include<map>
#include<stack>
#include<string>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std ;
#define pret(a,b) memset(a,b,sizeof(a))
const double PI = 3.1415926 ;
const double esp = 1e-4 ;
const int  md= 2810778 ;
const int INF = 99999999 ;
const int MX = 1000005 ;
int n,m,top,ans ;
int w[MX],dp[MX] ;
bool vix[MX] ;
struct node // 统计入度和出度
{
int r,c ;
}T[MX] ;
struct vert //  顶点
{
int head ;
}V[MX] ;
struct Edge
{
int v,next ;
}E[MX] ;
void init() // 初始化
{
pret(dp,0) ;
pret(T,0) ;
pret(w,0) ;
pret(vix,false) ;
pret(V,-1) ;
for(int i=0 ;i<n ;i++)
dp[i]=-INF ;
top=0 ;
ans=-INF ;
}
void add_edge(int u,int v) // 邻接表建图
{
E[top].v=v ;
E[top].next=V[u].head ;
V[u].head=top++ ;
}
int dfs(int x)
{
if(dp[x]!=-INF)
return dp[x] ;
int vx,max=-INF ;
for(int e=V[x].head ; e!=-1 ;e=E[e].next)
{
vx=E[e].v ;
int mx=w[x]+dfs(vx) ;
max= max < mx ? mx : max ;
}
return  dp[x]=max ;
}
int main()
{
int x,y ;
while(~scanf("%d%d",&n,&m))
{
init() ;
for(int i=0 ;i<n ;i++) // w[] 存每点的花费
scanf("%d",&w[i]) ;
for(int i=0 ;i<m ;i++)
{
scanf("%d%d",&x,&y) ;
x-- ;     y-- ;
add_edge(x,y) ;
T[x].c++ ;  // 出度加一
T[y].r++ ;  // 入度加一
}
for(int i=0 ;i<n ;i++) // 处理终点
if(!T[i].c)
dp[i]=w[i] ;
for(int i=0 ;i<n ;i++)  // 从入读为零的点开始
if(!T[i].r)
{
int mx=dfs(i) ;
ans= ans < mx ? mx : ans ;
}
printf("%d\n",ans) ;
}
return 0 ;
}