[BZOJ2282][Sdoi2011]消防（二分+贪心）

题目描述

传送门

题解

最大值最小很容易想到二分。
二分出所有点到路径的最大值，然后判断这个最大值是否能满足条件。
一个很重要的性质：一定存在一种方案，这个路径在树的最长链上。证明的话可以用反证法，假设该路径不在最长链上，那么这种方案的最大值一定不比在最长链上的更优。
根据这个性质我们可以确定一个贪心的策略。首先确定树的最长链，然后求出最长链上的每一个点到不是最长链上的点的最远距离。如果最长链上存在一个点它到其他点得最远距离>mid的话，说明它根本不可能走到最长链上，那么就判断无解。
求出了最远距离之后，每一个点剩余的距离记为mid-最远距离。也就是说，离它最远的点走到这个点了之后还能走多少。然后让最长链的左右端点向中间移，直到不能动为止。然后判断这条路径的长度是否<=s，是的话就有解，否则无解。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define N 300005

int n,s,x,y,z,Max,dMax,pt,inf,ans;
int tot,point[N],nxt[N*2],v[N*2],c[N*2];
int h[N],father[N],edge[N],leaf[N],C[N],len[N],goal[N],dis[N];
bool is[N];

void add(int x,int y,int z)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
}
void dfs(int x,int fa,int dep)
{
    father[x]=fa;h[x]=dep;
    if (h[x]>Max)
    {
        Max=h[x],pt=x;
    }
    bool flag=false;
    for (int i=point[x];i;i=nxt[i])
        if (v[i]!=fa)
        {
            dfs(v[i],x,dep+c[i]),flag=true;
            edge[v[i]]=c[i];
        }
    if (!flag) leaf[++leaf[0]]=x;
}
void chain(int x)
{
    while (h[x]!=0)
    {
        is[x]=true;
        C[++C[0]]=x;
        x=father[x];
    }
    C[++C[0]]=x;is[x]=true;
}
void length(int x,int fa,int dep,int st)
{
    dis[st]=max(dis[st],dep);
    for (int i=point[x];i;i=nxt[i])
        if (v[i]!=fa&&!is[v[i]])
            length(v[i],x,dep+c[i],st);
}
bool check(int mid)
{
    for (int i=1;i<=C[0];++i)
        if (dis[C[i]]<=mid) len[C[i]]=dis[C[i]]-mid;
        else return false;

    len[C[1]]=len[C[C[0]]]=mid;
    for (int i=1;i<=C[0];++i)
        if (len[C[i]]>=edge[C[i]])
        {
            len[C[i+1]]=min(len[C[i+1]],len[C[i]]-edge[C[i]]);
            len[C[i]]=inf;
        }
        else break;
    for (int i=C[0];i>=1;--i)
        if (len[C[i]]>=edge[C[i-1]])
        {
            len[C[i-1]]=min(len[C[i-1]],len[C[i]]-edge[C[i-1]]);
            len[C[i]]=inf;
        }
        else break;
    int l=0,a=0,b=0;
    for (int i=1;i<=C[0];++i)
        if (len[C[i]]!=inf) {a=i;break;}
    for (int i=C[0];i>=1;--i)
        if (len[C[i]]!=inf) {b=i;break;}
    for (int i=a;i<b;++i) l+=edge[C[i]];
    if (l<=s) return true;
    else return false;
}
int find()
{
    int l=0,r=dMax,mid,ans;
    while (l<=r)
    {
        mid=(l+r)>>1;
        if (check(mid)) ans=mid,r=mid-1;
        else l=mid+1;
    }
    return ans;
}
int main()
{
    scanf("%d%d",&n,&s);
    for (int i=1;i<n;++i)
    {
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);add(y,x,z);dMax+=z;
    }
    dfs(1,0,0);
    leaf[0]=1,h[pt]=0,Max=0,edge[pt]=0,dfs(pt,0,0);
    chain(pt);
    memset(len,127,sizeof(len));inf=len[0];
    for (int i=1;i<=C[0];++i)
        length(C[i],0,0,C[i]);
    ans=find();
    printf("%d\n",ans);
}

总结

完完全全想出了正解，但是码+调耗费了大半个上午的时间，犯各种傻逼错误，在这里记录一下。
①刚开始不是从最长链上的点到叶子节点dfs的，而是从叶子节点dfs。这不wa就鬼了啊！
②找每一个点到叶子节点的最远距离预处理就行了，可是我刚开始每一次二分都做了一遍。这不t就鬼了啊！
③两个端点往里缩的时候各种乱判。其实走不动了就break掉就行了啊。

老是这样是不行的，写代码思路一定要清晰，要冷静。但是不要磨叽。