崆若的水题之--赛跑的乌龟

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<ctime>
using namespace std;
const int maxn=30010;
struct Node
{
    Node* ch[2];
    int r,v;
}nodes[maxn],*null=new Node(),*S=null;
queue<Node*> Q;
Node* newnode()
{
    Node* o=Q.front(); Q.pop();
    return o;
}
void del(Node* &o)
{
    Q.push(o); o=null;
}
void rotate(Node* &o,int d)
{
    Node* k=o->ch[d^1]; o->ch[d^1]=k->ch[d]; k->ch[d]=o;
    o=k;
}
void insert(Node* &o,int v)
{
    if(o==null)
    {
        o=newnode();
        o->r=rand();
        o->v=v;
        o->ch[0]=o->ch[1]=null;
    }
    else
    {
        int d=v<o->v?0:1;
        insert(o->ch[d],v);
        if(o->ch[d]->r>o->r) rotate(o,d^1);
    }
}
int find(Node* &o,int v)
{
    if(o==null) return 0;
    int d=v<o->v?0:1;
    if(o->v==v) return v;
    else if(!d) return find(o->ch[0],v);
    else return max(o->v,find(o->ch[1],v));
}
void clean(Node* &o)
{
    if(o==null) return;
    clean(o->ch[0]);clean(o->ch[1]);
    del(o);
}
void print(Node* &o)
{
    if(o==null) return;
    print(o->ch[0]);
    printf("%d ",o->v);
    print(o->ch[1]);
}
int first[maxn],to[maxn*2],next[maxn*2],d[maxn*2];
int n,m=1,k;
void AddEdge(int a,int b,int c)
{
    to[m]=b;
    d[m]=c;
    next[m]=first[a];
    first[a]=m++;
}
int siz,root,f[maxn],s[maxn],done[maxn];
void findroot(int x,int fa)
{
    int maxs=0;
    s[x]=1;
    for(int i=first[x];i;i=next[i]) if(to[i]!=fa&&!done[to[i]])
    {
        findroot(to[i],x);
        s[x]+=s[to[i]];
        maxs=max(maxs,s[to[i]]);
    }
    f[x]=max(siz-s[x],maxs);
    if(f[x]<f[root]) root=x;
}
int ans,A[maxn],tt=0;
void dfs(int x,int fa,int dis)
{
    s[x]=1;
    A[tt++]=dis;
    for(int i=first[x];i;i=next[i]) 
      if(!done[to[i]]&&to[i]!=fa)
      {
         dfs(to[i],x,dis+d[i]);
         s[x]+=s[to[i]];
      }
}
int tott;
void solve(int x)
{
    done[x]=1;
    insert(S,0);
    for(int i=first[x];i;i=next[i]) if(!done[to[i]])
    {
       tt=0; dfs(to[i],x,d[i]);
       for(int j=0;j<tt&&A[j]<=k;j++) ans=max(ans,find(S,k-A[j])+A[j]);
       for(int j=0;j<tt;j++) insert(S,A[j]);
    }
    clean(S);
    for(int i=first[x];i;i=next[i]) if(!done[to[i]])
    {
       dfs(to[i],x,0);
       siz=f[0]=s[to[i]];
       findroot(to[i],root=0);
       solve(root);
    }
}
int main()
{
    for(int i=0;i<maxn;i++) Q.push(&nodes[i]);
    int a,b,c;
    scanf("%d%d",&n,&k);
    for(int i=1;i<n;i++)
    {
       scanf("%d%d%d",&a,&b,&c);
       AddEdge(a,b,c);
       AddEdge(b,a,c); 
    }
    f[0]=siz=n;
    findroot(1,root=0); 
    solve(root);
    printf("%d\n",ans);
    return 0;
}