[IOI2011]Race

race

点分治

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

　　在点分治过程中维护一下当层最小值即可，dfs一遍判断 路径权值 为v（v<=k）的路径最小数量g[v];

　　然后求下g[k-v]+g[v]的最小数量,更新下答案即可;

代码

　　

#include<iostream>
#include<cstdio>
using namespace std;
const int M=200009;
const int N=1000009;
int n,k,num=0,tot=0,top=0,ans=1e9;
int size,minn,rt;
int head[M],b[M],g[N],p[M],c[M],siz[M],dep[M];
struct P{int to,ne,w;}e[M<<1];
bool vis[M];
int read(){
    int rex=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){rex=rex*10+ch-'0';ch=getchar();}
    return rex*f;
}
void getrt(int u,int fa){
    int ma=0;siz[u]=1;
    for(int i=head[u];i;i=e[i].ne){
        int v=e[i].to;if(v==fa||vis[v])continue;
        getrt(v,u);
        siz[u]+=siz[v];ma=max(ma,siz[v]);
    }
    ma=max(ma,size-siz[u]);
    if(ma<minn){minn=ma,rt=u;}
}
void dfs(int u,int fa,int w){
    if(w>k)return;
    b[++tot]=w;dep[u]=dep[fa]+1;c[tot]=dep[u];
    for(int i=head[u];i;i=e[i].ne){
        int v=e[i].to;
        if(v!=fa&&!vis[v])dfs(v,u,w+e[i].w);
    }
}
void get(int u,int fa,int w){
    tot=0;
    dep[fa]=0;
    dfs(u,fa,w);
    for(int i=1;i<=tot;++i){
        ans=min(ans,c[i]+g[k-b[i]]);
        p[++top]=b[i];
    }
    for(int i=1;i<=tot;++i)g[b[i]]=min(g[b[i]],c[i]);
}
void work(int u){
    vis[u]=1;top=0;
    for(int i=head[u];i;i=e[i].ne){
        int v=e[i].to;
        if(!vis[v])get(v,u,e[i].w);
    }
    for(int i=1;i<=top;++i)g[p[i]]=1e9;
    for(int i=head[u];i;i=e[i].ne){
        int v=e[i].to;if(vis[v])continue;
        minn=1e9,size=siz[u];
        getrt(v,0);
        work(rt);
    }
}
int main(){
    n=read(),k=read();
    for(int i=1,u,v,w;i<n;++i){
        u=read()+1,v=read()+1,w=read();
        e[++num]=(P){v,head[u],w};head[u]=num;
        e[++num]=(P){u,head[v],w};head[v]=num;
    }
    for(int i=1;i<=k;++i)g[i]=1e9;
    minn=1e9,size=n;
    getrt(1,0);
    work(rt);
    if(ans==1e9)ans=-1;
    printf("%d",ans);
    return 0;
}