BZOJ 1117: [POI2009]救火站Gas 贪心

最新推荐文章于 2018-09-28 22:32:11 发布

NeighThorn

最新推荐文章于 2018-09-28 22:32:11 发布

阅读量777

点赞数 1

分类专栏： BZOJ 那里是贪心搜索的良辰美景

本文链接：https://blog.csdn.net/neighthorn/article/details/53046157

版权

BZOJ 同时被 2 个专栏收录

45 篇文章 1 订阅

订阅专栏

那里是贪心搜索的良辰美景

16 篇文章 3 订阅

订阅专栏

时空隧道

其实总的来说就是不会做….QAQ
我们对于每一个节点记录两个值，need[i][j]代表需要被i节点覆盖的距离为j的节点数目，cover[i][j]代表可以被i节点覆盖的距离为j的节点数目…
然后如果一个节点的need大于cover那么就贪心的向这里放一些救火站…
相邻节点的need值和cover值可以互相抵消…但是我们贪心的只让距离为i和距离为i-1的两值相消…因为如果相消的两项距离相差太大就是浪费…

代码如下：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
//by NeighThorn
using namespace std;
const int maxn=100000+5,maxk=20+5;
int n,s,k,hd[maxn],to[maxn*2],nxt[maxn*2],cnt;
long long cover[maxn][maxk],need[maxn][maxk],ans;
inline void add(int x,int y){
    to[cnt]=y;nxt[cnt]=hd[x];hd[x]=cnt++;
}
inline void modify(int root,int lala){
    if(lala==0){
        for(int i=0;i<=k;i++)
            for(int j=i;j>=i-1&&cover[root][i];j--){
                long long tmp=min(cover[root][i],need[root][j]);
                cover[root][i]-=tmp,need[root][j]-=tmp;
            }
    }
    else{
        for(int i=0;i<=k;i++)
            for(int j=i;j>=0&&cover[root][i];j--){
                long long tmp=min(cover[root][i],need[root][j]);
                cover[root][i]-=tmp,need[root][j]-=tmp;
            }
    }
}
inline void dfs(int root,int fa){
    need[root][0]=1;
    for(int i=hd[root];i!=-1;i=nxt[i])
        if(to[i]!=fa){
            dfs(to[i],root);
            for(int j=0;j<k;j++)
                need[root][j+1]+=need[to[i]][j];
            for(int j=0;j<k;j++)
                cover[root][j]+=cover[to[i]][j+1];
        }
    long long rema=need[root][k]-cover[root][k];
    if(rema>0)
        cover[root][k]+=(rema+s-1)/s*s,ans+=(rema+s-1)/s;
    modify(root,0);
}
signed main(void){
    scanf("%d%d%d",&n,&s,&k);cnt=0;
    memset(hd,-1,sizeof(hd));
    for(int i=1,x,y;i<n;i++)
        scanf("%d%d",&x,&y),add(x,y),add(y,x);
    dfs(1,0);modify(1,1);long long tmp=0;
    for(int i=0;i<=k;i++)
        tmp+=need[1][i];
    ans+=(tmp+s-1)/s;
    printf("%d\n",ans);
    return 0;
}