POJ2152 Fire 树形DP

题目大意:n个城市组成一棵树,要在其中若干城市修建消防站。其中,在城市i修建消防站需要花费w[i]。最后想达到这样的状态,对于每个城市j,使得离其最近的消防站的距离不超过d[j]。问,达到这种状态的最小花费。


一读完,就觉得好难,仔细思考了一天,发现确实好难 = =

然后,...,然后就去看陈启峰的论文《一张一弛,解题之道》


状态设计为dp[i][j],表示对i为根的子树,满足每个点,都找到了消防站的最小花费。其中,城市i一定找的是建在城市j的消防站。而其它的,找的是城市j或i子树内的城市的消防站。

这个状态为什么是对的?因为对于i为根的子树,里面的两个城市i1,i2,若这两个城市都是依赖于i为根的子树外的消防站,两个消防站是同一个,这一定是最优解之一。(假设i1距离i,比i2更近。i1依赖于子树外的消防站jj,那么i2也一定可以依赖于jj)


看完论文,我大概懂了,然后一敲,发现...咦,论文中的第三种情况............O(n*n)能做?然后又不会了。


于是又去看别人代码,然后发现若dp[i][j]的花费包括建立城市j的消防站,顿时好些不少。

p.s,我最开始还用欧拉序列去判断子孙关系,果然我还是太弱 ~~o(>_<)o ~~


dp[i][j],i为根的子树,每个点都有依赖的消防站,其中i一定依赖j。而城市j和i子树内的消防站已经建好。最小花费。

best[i],i为根的子树,每个点都有依赖的消防站,所依赖的消防站都建立好了的最小花费。


//#pragma comment(linker, "/STACK:102400000,102400000")
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
#include<cctype>
#include<string>
#include<algorithm>
#include<iostream>
#include<ctime>
#include<map>
#include<set>
using namespace std;
#define MP(x,y) make_pair((x),(y))
#define PB(x) push_back(x)
typedef long long LL;
//typedef unsigned __int64 ULL;
/* ****************** */
const int INF=1000111222;
const double INFF=1e200;
const double eps=1e-8;
const LL mod=1000000007;
const int NN=1010;
const int MM=100010;
/* ****************** */

int dp[NN][NN],best[NN];
int w[NN],d[NN];

struct G
{
    int v,len,next;
}E[NN*2];
int dis[NN],p[NN],T;
bool inq[NN];
void add(int u,int v,int len)
{
    E[T].v=v;
    E[T].len=len;
    E[T].next=p[u];
    p[u]=T++;
}

void spfa(int st,int n)
{
    int i,ix,v;
    queue<int>q;

    for(i=1;i<=n;i++)
    {
        inq[i]=false;
        dis[i]=INF;
    }
    dis[st]=0;
    inq[st]=true;
    q.push(st);
    while(!q.empty())
    {
        ix=q.front();
        q.pop();
        inq[ix]=false;
        for(i=p[ix];i+1;i=E[i].next)
        {
            v=E[i].v;
            if(dis[v]>dis[ix]+E[i].len)
            {
                dis[v]=dis[ix]+E[i].len;
                if(!inq[v])
                {
                    inq[v]=true;
                    q.push(v);
                }
            }
        }
    }

}

void dfs(int u,int fa,int n)
{
    int i,j,v;
    for(i=p[u];i+1;i=E[i].next)
    {
        v=E[i].v;
        if(v==fa)
            continue;
        dfs(v,u,n);
    }

    spfa(u,n);
    best[u]=INF;

    for(i=1;i<=n;i++)
    {
        if(dis[i]>d[u])
        {
            dp[u][i]=INF;
            continue;
        }

        dp[u][i]=w[i];
        for(j=p[u];j+1;j=E[j].next)
        {
            v=E[j].v;
            if(v==fa)
                continue;
            dp[u][i]+=min(dp[v][i]-w[i],best[v]);
        }
        best[u]=min(best[u],dp[u][i]);
    }
}

int main()
{
    int cas;
    int n,i,u,v,len;
    scanf("%d",&cas);
    while(cas--)
    {
        scanf("%d",&n);
        for(i=1;i<=n;i++)
            scanf("%d",&w[i]);
        for(i=1;i<=n;i++)
            scanf("%d",&d[i]);

        memset(p,-1,sizeof(p));
        T=0;

        for(i=1;i<n;i++)
        {
            scanf("%d%d%d",&u,&v,&len);
            add(u,v,len);
            add(v,u,len);
        }

        dfs(1,-1,n);

        printf("%d\n",best[1]);
    }
    return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值