【bzoj2599】[IOI2011]Race 点分治

点分治
1、求树的重心
2、计算以当前重心为根的子树的答案
3、去掉以当前重心儿子为根的子树的答案
4、枚举每个儿子，分治

考虑计算过程如何实现
我们不妨记一个ans数组，ans[i]表示使用i条边权值为k的有多少对
每次实现2的时候，权值设为+1
每次实现3的时候，权值设为-1

把子树内所有的dis排序，计算有多少对权值和为k的，两个指针扫一遍就可以了。

#include<cstdio>#include<cstring>#include<cstdlib>#include<cmath>#include<algorithm>#include<iostream>#define maxn 200100 using namespace std;struct yts{int d1,d2;}st[maxn];int head[maxn],to[2*maxn],next[2*maxn],len[2*maxn],dis[maxn],dep[maxn];int size[maxn],ans[maxn],fa[maxn];bool vis[maxn];int n,m,num,k,top;bool cmp(yts x,yts y){return x.d1<y.d1;}void addedge(int x,int y,int z){num++;to[num]=y;len[num]=z;next[num]=head[x];head[x]=num;}void get_root(int x,int Size,int &cg){bool flag=1;size[x]=1;for (int p=head[x];p;p=next[p])  if (!vis[to[p]] && to[p]!=fa[x])  {  fa[to[p]]=x;  get_root(to[p],Size,cg);  size[x]+=size[to[p]];  if (size[to[p]]>Size/2) flag=0;  }if (Size-size[x]>Size/2) flag=0;if (flag) cg=x;}void dfs1(int x,int d1,int d2){dis[x]=d1;dep[x]=d2;for (int p=head[x];p;p=next[p])  if (!vis[to[p]] && to[p]!=fa[x]) fa[to[p]]=x,dfs1(to[p],d1+len[p],d2+1);}void dfs2(int x){st[++top].d1=dis[x];st[top].d2=dep[x];for (int p=head[x];p;p=next[p])  if (!vis[to[p]] && to[p]!=fa[x]) dfs2(to[p]);}void calc(int x,int f){top=0;dfs2(x);sort(st+1,st+top+1,cmp);for (int i=1,j=top;i<=j;i++){while (j>i && st[i].d1+st[j].d1>k) j--;for (int p=j;st[i].d1+st[p].d1==k;p--) ans[st[i].d2+st[p].d2]+=f;}}void solve(int x,int Size){int cg;fa[x]=0;get_root(x,Size,cg);size[fa[cg]]=Size-size[cg];fa[cg]=0;vis[cg]=1;dfs1(cg,0,0);calc(cg,1);for (int p=head[cg];p;p=next[p]) if (!vis[to[p]]) calc(to[p],-1);for (int p=head[cg];p;p=next[p]) if (!vis[to[p]]) solve(to[p],size[to[p]]);}int main(){scanf("%d%d",&n,&k);for (int i=1;i<n;i++){int x,y,z;scanf("%d%d%d",&x,&y,&z);x++;y++;addedge(x,y,z);addedge(y,x,z);}solve(1,n);for (int i=1;i<n;i++) if (ans[i]) {printf("%d\n",i);return 0;}printf("-1\n");return 0;}