点分治练习3——P4149 [IOI2011]Race

蒟蒻的垂死挣扎

这题近似模板题。首先你要确定的是，你的点分治计算贡献是不能再用容斥的那个方法了，我们采取更加优秀的模板，边遍历边计算答案，然后这题就又变成模板了。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<queue>
#include<map>
#include<set>
#define ll long long
#define MOD 998244353
#define N 200100
#define M 10010000
#define RG register

using namespace std;

inline ll read(){
    RG ll x=0,o=1; RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
    if(ch=='-') o=-1,ch=getchar();
    while(ch>='0'&&ch<='9') x=((x<<3)+(x<<1))+ch-'0',ch=getchar();
    return x*o;
}

int n,K,top,first[N],t[M],nt[M],Siz,Max,root,siz[N],vis[N],tot,st[M],ans=1e9;
struct mona { int nxt,en; ll w; } s[N<<1];
inline void Insert(int x,int y,ll w) { s[++top]=(mona) { first[x],y,w },first[x]=top; }

inline void Getroot(int k,int fa){
    siz[k]=1;   int mmax=0;
    for(RG int i=first[k];i;i=s[i].nxt){
        int en=s[i].en; if(en==fa||vis[en]) continue ;
        Getroot(en,k),siz[k]+=siz[en],mmax=max(mmax,siz[en]);
    }   mmax=max(mmax,Siz-siz[k]);
    if(Max>mmax) Max=mmax,root=k; return ;
}
inline void Getdis(int k,int fa,ll w,int num){
    if(w>K) return ;
    ans=min(ans,t[K-w]+num),nt[w]=min(nt[w],num),st[++tot]=w;
    for(RG int i=first[k];i;i=s[i].nxt){
        int en=s[i].en; if(en==fa||vis[en]) continue ;
        Getdis(en,k,w+s[i].w,num+1);
    }
}
inline void Divide(int k){
    vis[k]=1;  int las=0; tot=0; int Copy=Siz;
    for(RG int i=first[k];i;i=s[i].nxt){
        int en=s[i].en; if(vis[en]) continue ;
        Getdis(en,0,s[i].w,1);
        for(RG int j=las+1;j<=tot;++j) t[st[j]]=min(t[st[j]],nt[st[j]]),nt[st[j]]=1e9;
        las=tot;
    }   for(RG int i=1;i<=tot;++i) t[st[i]]=1e9;
    for(RG int i=first[k];i;i=s[i].nxt){
        int en=s[i].en; if(vis[en]) continue ;
        Siz= siz[k]>siz[en]?siz[en]:Copy-siz[k];
        Max=2e9,root=0,Getroot(en,0),Divide(root);
    }
}

int main(){
    n=read(),K=read();
    for(RG int i=1;i<=K;++i) t[i]=nt[i]=1e9;
    for(RG int i=1;i<n;++i){
        int x=read()+1,y=read()+1; ll z=read();
        Insert(x,y,z),Insert(y,x,z);
    }   Siz=n,Max=1e9,Getroot(1,0),Divide(root);
    if(ans>1e9-1) puts("-1"); else cout<<ans; return 0;
}