1468: Tree|树的点分治

楼教主男人八题之一(好可怕)
似乎不是很难的样子。。。
点分治大致是这样:先选出一个根(一般是重心),然后可以把两个点之间的路径分为经过根的和不经过根的,经过根的直接处理记录,不经过根的递归处理计算。似乎还用到了容斥原理的方法。
记录经过当前根的方案数的方法是:
先以当前的根为起点遍历一遍记下 dis 数组即到根的距离,然后排序直接统计距离之和 K 的点的对数,这样会把在同一颗子树里的点也计算进去,所以在找每一颗子树的时候再减去那些无效的点对的数量。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<iostream>
#include<algorithm>
#define N 80022
using namespace std;
int sc()
{
    int i=0,f=1; char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
    return i*f;
}
int vis[N],mx[N],size[N],a[N],dis[N];
int head[N],nxt[N],lst[N],v[N];
int m,tot,n,K,sum,root,ans,top,TI;
void insert(int x,int y,int z)
{
    lst[++tot]=y;nxt[tot]=head[x];head[x]=tot;v[tot]=z;
    lst[++tot]=x;nxt[tot]=head[y];head[y]=tot;v[tot]=z;
}
void get_root(int x,int f)
{
    size[x]=1;mx[x]=0;
    for(int i=head[x];i;i=nxt[i])
        if(lst[i]!=f&&!vis[lst[i]])
        {
            get_root(lst[i],x);
            size[x]+=size[lst[i]];
            mx[x]=max(mx[x],size[lst[i]]);
        }
    mx[x]=max(mx[x],sum-size[x]);
    if(mx[x]<mx[root])root=x;
}
void get_dis(int x,int f)
{
    a[++top]=dis[x];
    for(int i=head[x];i;i=nxt[i])
        if(lst[i]!=f&&!vis[lst[i]])
        {
            dis[lst[i]]=dis[x]+v[i];
            get_dis(lst[i],x);
        }
}
int cal(int x,int now)
{
    dis[x]=now;top=0;
    get_dis(x,0);
    sort(a+1,a+top+1);
    int ans=0,l=1,r=top;
    while(l<r)
        if(a[l]+a[r]<=K)ans+=r-l,l++;else r--;
    return ans;
}   
void solve(int x)
{
    ans+=cal(x,0);
    vis[x]=1;
    for(int i=head[x];i;i=nxt[i])
        if(!vis[lst[i]])
        {
            ans-=cal(lst[i],v[i]);
            sum=size[lst[i]];
            root=0;
            get_root(lst[i],0);
            solve(root);
        }

}
int main()
{
    mx[0]=N;
    n=sc();
    for(int i=1;i<n;i++)
    {
        int x=sc(),y=sc(),z=sc();
        insert(x,y,z);
    }
    K=sc();
    sum=n;
    root=0;
    get_root(1,0);
    solve(root);
    cout<<ans;
    return 0;
}

你可能感兴趣的:(树的点分治)