点分治 模板 2

自己整理模板，仅作模板保存使用

#include <iostream>
#include<algorithm>
#include<string.h>
#include <stdio.h>

using namespace std;
const int inf=0x3f3f3f3f;
const int maxn=2*1e5+10;
int first[maxn];
int step,sum;
int ssize[maxn],mson[maxn];
bool vis[maxn];
int root,mx,n,m;
struct node
{
    int y,l;
    int next;
}edge[maxn];

void cinn(int x,int y,int l)
{
    edge[step].y=y;
    edge[step].l=l;
    edge[step].next=first[x];
    first[x]=step;
    step++;
}

void getroot(int x,int fa)
{
    ssize[x]=1;mson[x]=0;
    for(int i=first[x];i!=-1;i=edge[i].next)
    {
        int y=edge[i].y;
        if(y!=fa&&vis[y]==0)
        {
            getroot(y,x);
            ssize[x]+=ssize[y];
            if(ssize[y]>mson[x]) mson[x]=ssize[y];
        }
    }
    if(sum-ssize[x]>mson[x]) mson[x]=sum-ssize[x];
    if(mx>mson[x])
    {
        mx=mson[x];
        root=x;
    }
}

int t=0,ans;
int dis[maxn];

void getdis(int x,int len,int fa)
{
    dis[++t]=len;
    for(int i=first[x];i!=-1;i=edge[i].next)
    {
        int y=edge[i].y;
        if(vis[y]==0&&y!=fa) getdis(y,edge[i].l+len,x);
    }
}

int cal(int x,int len)
{
    t=0;
    getdis(x,len,0);
    sort(dis+1,dis+t+1);
    cout<<"root="<<root<<endl;
    for(int i=1;i<=t+1;i++)
    {
        cout<<dis[i]<<" ";
    }
    cout<<endl;
    int i=1,j=t;
    int temp=0;
    while(i<j)
    {
        while(dis[i]+dis[j]>m&&i<j) j--;
        temp+=j-i;
        i++;
    }
    return temp;
}
void fenzhi(int x)
{
    vis[x]=true;
    ans+=cal(x,0);
    for(int i=first[x];i!=-1;i=edge[i].next)
    {
        int y=edge[i].y;
        if(vis[y]==0)
        {
             ans-=cal(y,edge[i].l);
             root=0;
             mx=n;
             sum=ssize[y];
             getroot(y,x);
             fenzhi(root);
        }
    }

}

int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(n==0||m==0) break;
        int x,y,z;
        step=0;
        memset(first,-1,sizeof(first));
        memset(vis,false,sizeof(vis));
        for(int i=1;i<n;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            cinn(x,y,z);
            cinn(y,x,z);
        }
        ans=0;
        sum=n;mx=n;
        getroot(1,0);
        fenzhi(root);
        printf("%d\n",ans);
    }
    return 0;
}