洛谷 P4149 [ IOI 2011 ] Race —— 点分治

题目：https://www.luogu.org/problemnew/show/P4149

仍然是点分治；

不过因为是取 min ，所以不能用容斥，那么子树之间就必须分开算，记录桶时注意这个；

每次 memset 桶会很慢，可以用栈记录修改的地方，然后改回来即可；

注意更新 getrt 中 sum 的方式，可以 dfs 时顺便重新算一下 siz，但也可以利用原树求出来的 siz，判断一下当前的儿子在原树中是儿子还是父亲；

那么就要传个参数，是当前的所有点个数，在原树中是父亲的话就用总个数 - siz[to[i]]，这个做法比较快。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int const maxn=2e5+,maxm=1e6+,inf=0x3f3f3f3f;
int n,K,hd[maxn],ct,to[maxn<<],nxt[maxn<<],w[maxn<<],dis[maxn],siz[maxn];
int sum,rt,tmp[maxm],ans=inf,mx,sta[maxn][],f[maxn],top;
bool vis[maxn];
void add(int x,int y,int z){to[++ct]=y; nxt[ct]=hd[x]; w[ct]=z; hd[x]=ct;}
void getrt(int x,int fa)
{
    siz[x]=; int nmx=;//局部变量！
    for(int i=hd[x],u;i;i=nxt[i])
    {
        if((u=to[i])==fa||vis[u])continue;
        getrt(u,x);
        siz[x]+=siz[u]; nmx=max(nmx,siz[u]);
    }
    nmx=max(nmx,sum-siz[x]);
    if(nmx<mx)mx=nmx,rt=x;
}
void dfs(int x,int fa)//siz 不管的话 RE 2个点
{
    siz[x]=;
    for(int i=hd[x],u;i;i=nxt[i])
    {
        if((u=to[i])==fa||vis[u])continue;
        dis[u]=dis[x]+w[i]; f[u]=f[x]+;
        if(dis[u]<=K)
        {
            ans=min(ans,f[u]+tmp[K-dis[u]]);
            sta[++top][]=dis[u]; sta[top][]=f[u];
        }
        dfs(u,x);
        siz[x]+=siz[u];
    }
}
int work(int x,int ss)
{
    vis[x]=; int p=;//局部变量
    for(int i=hd[x],u;i;i=nxt[i])
    {
        if(vis[u=to[i]])continue;
        dis[u]=w[i]; f[u]=;
        if(dis[u]<=K)
        {
            ans=min(ans,f[u]+tmp[K-dis[u]]);
            sta[++top][]=dis[u]; sta[top][]=f[u];
        }
        dfs(u,);
        for(int w;p<=top;p++)tmp[w=sta[p][]]=min(tmp[w],sta[p][]);
    }
    for(int i=;i<=top;i++)tmp[sta[i][]]=inf; top=;
    for(int i=hd[x],u;i;i=nxt[i])
    {
        if(vis[u=to[i]])continue;
        sum=(siz[u]>siz[x]?ss-siz[x]:siz[u]); mx=inf; getrt(u,); work(rt,sum);
        //可以这样更新sum     //u在原树中是x的儿子或父亲
    }
}
int main()
{
    scanf("%d%d",&n,&K);
    for(int i=,x,y,z;i<n;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z); add(y,x,z);
    }
    memset(tmp,0x3f,sizeof tmp); tmp[]=;//
    sum=n; mx=inf; getrt(,);
    work(rt,sum);
    printf("%d\n",ans==inf?-:ans);
    return ;
}