BZOJ2599 [IOI2011]Race

传送门

点分治，黄学长的选根方法会T掉，换了这个人的选根方法就可以了。

当然，你也可以选择黄学长的奇淫优化

 //BZOJ 2599
 //by Cydiater
 //2016.9.23
 #include <iostream>
 #include <cstring>
 #include <string>
 #include <algorithm>
 #include <queue>
 #include <map>
 #include <ctime>
 #include <cmath>
 #include <cstdio>
 #include <cstdlib>
 #include <iomanip>
 using namespace std;
 #define ll long long
 #define up(i,j,n)        for(int i=j;i<=n;i++)
 #define down(i,j,n)        for(int i=j;i>=n;i--)
 ;
 const int oo=0x3f3f3f3f;
 inline int read(){
     ,f=;
     ;ch=getchar();}
     +ch-';ch=getchar();}
     return x*f;
 }
 ,N,K,sum,LINK[MAXN],dis[MAXN],eum[MAXN],root=,siz[MAXN],Count[MAXN],head,tail,f[MAXN];
 bool vis[MAXN];
 struct edge{
     int y,next,v;
 }e[MAXN];
 struct Queue{
     int v,id;
 }q[MAXN];
 namespace solution{
     inline void insert(int x,int y,int v){e[++len].next=LINK[x];LINK[x]=len;e[len].y=y;e[len].v=v;}
     inline bool cmp(Queue x,Queue y){return x.v<y.v;}
     void init(){
         N=read();K=read();
         memset(vis,,sizeof(vis));
         up(i,,N){
             ,y=read()+,v=read();
             insert(x,y,v);
             insert(y,x,v);
         }
     }
     void make_root(int node,int fa){
         siz[node]=;;
         for(int i=LINK[node];i;i=e[i].next)if(!vis[e[i].y]&&e[i].y!=fa){
             make_root(e[i].y,node);
             siz[node]+=siz[e[i].y];
             f[e[i].y]=node;
             )ff=;
         }
         )ff=;
         if(ff)root=node;
     }
     void get_deep(int node,int fa){
         q[++tail].v=dis[node];q[tail].id=node;
         for(int i=LINK[node];i;i=e[i].next)if(!vis[e[i].y]&&e[i].y!=fa){
             f[e[i].y]=node;
             dis[e[i].y]=dis[node]+e[i].v;
             eum[e[i].y]=eum[node]+;
             get_deep(e[i].y,node);
         }
     }
     void col(int node,int dist,int flag){
         dis[node]=dist;head=;tail=;
         eum[node]=flag==-?:;
         get_deep(node,);
         sort(q+,q+tail+,cmp);
         while(head<tail){
             while(head<tail&&q[head].v+q[tail].v>K)tail--;
             int tmp=tail;
             while(head<tmp&&q[tmp].v+q[head].v==K){
                 int x=q[tmp].id,y=q[head].id;
                 Count[eum[x]+eum[y]]+=flag;
                 tmp--;
             }
             head++;
         }
     }
     void work(int node){
         col(node,,);
         vis[node]=;siz[f[root]]=sum-siz[root];f[root]=;
         );
         for(int i=LINK[node];i;i=e[i].next)if(!vis[e[i].y]){
             root=;sum=siz[e[i].y];
             make_root(e[i].y,node);
             work(root);
         }
     }
     void slove(){
         sum=N;root=;
         make_root(,);
         work(root);
     }
     void output(){
         up(i,,N)if(Count[i]){
             printf("%d\n",i);
             return;
         }
         puts("-1");
     }
 }
 int main(){
     //freopen("input.in","r",stdin);
     using namespace solution;
     init();
     slove();
     output();
     ;
 }