hdu 3409 最短路树+树形dp

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3409

参考博客：http://www.cnblogs.com/woaishizhan/p/3189813.html

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
const int maxn = ;
const int maxe = ;
const int INF = 0x3f3f3f;

struct Edge{
    int u,v,w;
    int next;
    void assign(int u_,int v_,int w_,int next_){
        u = u_;  v = v_;   w = w_;  next = next_;
    }
    bool operator < (const Edge& e) const {
        return w > e.w;    //这个地方害我WA了6次；这错太隐蔽了!!!!。
    }
}edges[maxe];

int head[maxn];
vector<int> G[maxn];
int cnt;
int N,M,P;
double PT[maxn][maxn/];
double dp[maxn][maxn/];  //dp[i][j]表示以i为根、部署j个警察逮到robber的最大概率；

void addedge(int u,int v,int w){
    edges[cnt].assign(u,v,w,head[u]);
    head[u] = cnt++;
    edges[cnt].assign(v,u,w,head[v]);
    head[v] = cnt++;
}
void Dijkstra(){
    priority_queue<Edge>  Q;
    int d[maxn];
    bool vis[maxn];
    memset(d,0x3f,sizeof(d));
    memset(vis,,sizeof(vis));
    for(int i=;i<N;i++) G[i].clear();
    Q.push((Edge){,,});
    d[] = ;
    while(!Q.empty()){
        Edge e = Q.top();  Q.pop();
        int u = e.u;
        if(vis[u]) continue;
        vis[u] = true;
        if(e.u != e.v){
            G[e.v].push_back(e.u);
        }
        for(int i=head[u];i!=-;i=edges[i].next){
            int v = edges[i].v;
            if(d[v] > d[u] + edges[i].w){
                d[v] = d[u] + edges[i].w;
                Q.push((Edge){v,u,d[v]});  //把u存进去是为了方便建最短路图；
            }
        }
    }
}
void dfs(int u){
    int child = G[u].size();
    double son[maxn];                    //son[i]表示u的所有儿子节点总共部署i个人逮住robber的概率；
    for(int i=;i<=P;i++)   son[i] = ;
    if(child == ){
        for(int i=;i<=P;i++){
            dp[u][i] = PT[u][i];
        }
        return;
    }
    for(int i=;i<child;i++){
        int v = G[u][i];
        dfs(v);                     // 求出了dp[v][...]的所有不同概率；
        for(int j=P;j>=;j--)
          for(int k=;k<=j;k++)
            son[j] = max(son[j],dp[v][k]/child+son[j-k]);  //u的v这个儿子及其子节点部署k个人的最大概率；
    }
    for(int i=P;i>=;i--)         //总共i个人
      for(int j=;j<=i;j++){      //u这个节点放j个人
        dp[u][i] = max(dp[u][i],PT[u][j] + (-PT[u][j])* son[i-j]);
    }
}
int main()
{
    //freopen("E:\\acm\\input.txt","r",stdin);

    while(scanf("%d%d",&N,&M) ==  && N+M){
        cnt = ;
        memset(dp,,sizeof(dp));
        memset(head,-,sizeof(head));
        for(int i=;i<=M;i++){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            if(a == b) continue;
            addedge(a,b,c);
        }
        scanf("%d",&P);
        for(int i=;i<N;i++)  PT[i][] = ;
        for(int i=;i<N;i++)
            for(int j=;j<=P;j++){
                scanf("%lf",&PT[i][j]);
            }
        Dijkstra();  //形成以0为根的最短路树；存在G[u]中；

        dfs();
        double ans = ;
        for(int i=;i<=P;i++)
            ans = max(ans,dp[][i]);
        printf("%.2lf\n",ans*);
    }
}