【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;
}