bzoj2599 [IOI2011]Race（点分治）

求边权和为K，且边数最少的路径。我们点分治就好了，不过据说这题卡时。。。所以我们需要一些小trick。比如说不容斥，提前记录bel，用来保证只更新合法情况。类似之前的做法，我们可以O(n)扫来更新答案，然后因为相等时我们要找最优的答案，比较麻烦。。。复杂度可能也就退化了。我们排序时如果dis相等，要把dep小的排在后面，这样我们从后往前扫遇到的第一个合法的就是最优得了。其实应该用类似dp的方法来维护这个东西，保证复杂度，具体做法可以参见Elijahqi神犇的blog：传送门。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 200010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,K,rt,h[N],num=0,sz[N],fa[N],sumsz,ans=inf,dep[N],dis[N],tot=0,top,f[N];
bool vis[N];
struct edge{
    int to,next,val;
}data[N<<1];
struct node{
    int len,dep,bel;
    friend bool operator<(node a,node b){return a.len==b.len?a.dep>b.dep:a.len<b.len;}
}a[N];
inline void dfs1(int x){
    sz[x]=1;a[++tot].len=dis[x];a[tot].dep=dep[x];a[tot].bel=top;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]||vis[y]) continue;
        fa[y]=x;dep[y]=dep[x]+1;dis[y]=dis[x]+data[i].val;
        dfs1(y);sz[x]+=sz[y];
    }
}
inline void dfs2(int x){
    f[x]=0;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]||vis[y]) continue;
        dfs2(y);f[x]=max(f[x],sz[y]);
    }f[x]=max(f[x],sumsz-sz[x]);if(f[x]<f[rt]) rt=x;
}
inline void calc(int x){
    a[1].len=0;a[1].dep=0;a[1].bel=0;tot=1;
    sz[x]=1;dep[x]=0;dis[x]=0;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]) continue;
        fa[y]=x;dep[y]=1;dis[y]=data[i].val;top=y;dfs1(y);sz[x]+=sz[y];
    }sort(a+1,a+tot+1);int l=1,r=tot;
    while(l<r){
        if(a[l].len+a[r].len>K) r--;
        else{
            int now=r;
            while(now>l&&a[l].bel==a[now].bel&&a[l].len+a[now].len==K) now--;
            if(now>l&&a[l].len+a[now].len==K) ans=min(ans,a[l].dep+a[now].dep);l++; 
        }
    }
}
inline void solve(int x){
    vis[x]=1;fa[x]=0;calc(x);
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]) continue;
        rt=0;sumsz=sz[y];dfs2(y);solve(rt);
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();K=read();f[0]=inf;
    if(!K){puts("0");return 0;}
    for(int i=1;i<n;++i){
        int x=read()+1,y=read()+1,val=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=val;
        data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].val=val;
    }dfs1(1);rt=0;sumsz=n;dfs2(1);solve(rt);
    if(ans==inf) puts("-1");
    else printf("%d\n",ans);
    return 0;
}