bzoj2599: [IOI2011]Race（点分治）

写了四五道点分治的题目了，算是比较理解点分治是什么东西了吧= =

点分治主要用来解决点对之间的问题的，比如距离为不大于K的点有多少对。

这道题要求距离等于K的点对中连接两点的最小边数。

那么其实道理是一样的。先找重心，然后先从重心开始求距离dis和边数num，更新ans，再从重心的儿子开始求得dis和num，减去这部分答案

因为这部分的答案中，从重心开始的两条链有重叠部分，所以要剪掉

基本算是模板题，但是减去儿子的答案的那部分还有双指针那里调了好久，所以还不算特别熟练。。

PS跑了27秒慢到飞起，不过代码短一点，看起来比较清晰

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 200010;
struct node{
    int to,next,cost;
}e[maxn*2];
struct data{
    int l,e;
}dis[maxn];
int n,m,K,tot,head[maxn],size[maxn],sz,total,vis[maxn],root,p,ans[maxn*2];

void insert(int u, int v, int w){
    e[++tot].to=v; e[tot].next=head[u]; head[u]=tot; e[tot].cost=w;
}

void getroot(int u, int f){
    int mx=0; size[u]=1;
    for (int i=head[u],v; i; i=e[i].next){
        if (vis[v=e[i].to] || v==f) continue;
        getroot(v,u);
        size[u]+=size[v];
        mx=max(mx,size[v]);
    }
    mx=max(mx,total-size[u]);
    if (mx<sz) sz=mx,root=u;
}

void getdis(int u, int f, int len, int num){
    dis[++p].l=len; dis[p].e=num;
    for (int i=head[u],v; i; i=e[i].next){
        if (vis[v=e[i].to] || v==f) continue;
        getdis(v,u,len+e[i].cost,num+1);
    }
}

bool cmp(data a, data b){
    if (a.l==b.l) return a.e<b.e; return a.l<b.l;  /////
}

void count(int u, int len, int num, int f){
    p=0; getdis(u,0,len,num); //这里要将len和num传下去。。
    int l=1,r=p;
    sort(dis+1,dis+1+p,cmp);
    while (l<=r){  //注意这里要用<=，WA了几发
        while (l<r && dis[l].l+dis[r].l>K) r--;
        for (int k=r; dis[l].l+dis[k].l==K; k--) ans[dis[l].e+dis[k].e]+=f;
        l++;
    }
}

void work(int u){
    total=size[u]?size[u]:n;
    sz=INF;
    getroot(u,0); u=root;
    vis[u]=1; count(u,0,0,1);
    for (int i=head[u],v; i; i=e[i].next){
        if (vis[v=e[i].to]) continue;
        count(v,e[i].cost,1,-1);
        work(v);
    }
}

int main(){
    scanf("%d%d", &n, &K);
    for (int i=1,u,v,w; i<n; i++) scanf("%d%d%d", &u, &v, &w),u++,v++,insert(u,v,w),insert(v,u,w);
    work(1);
    for (int i=1; i<n; i++) if (ans[i]){printf("%d
", i); return 0;} puts("-1");
    return 0;
}