POJ1741：tree（点分治）

 

题目大意：

给你一颗树，树上每条边有自己的权值，问你两点间权值和小于等于k的点对有多少个。

 

解题思路：

点分治模板题。

今天学了一下点分治这个东西，看了整整一天，还是感觉没怎么搞懂，大概原理明白了，但总是觉得还有点欠缺= =，题目稍微一变就不会做的样子。。。还需继续学习。。。

点分治个人认为也是一种比较重要的算法，在解决树上路径问题的时候有很大的作用，推荐学习一下，直接看论文即可。

论文地址：树分治论文

 

Ac代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#include <algorithm>
#include <set>
#include <functional>
#define size si
using namespace std;
const int maxn=1e5+5;
const int INF=1e9+7;
int n,k;
struct Edge
{
    int to,next,w;
}e[maxn];
int head[maxn],tot;
bool vis[maxn];
void add(int u,int v,int w)
{
    e[tot].to=v;e[tot].w=w;
    e[tot].next=head[u];head[u]=tot++;
}
int res,tmp,root,num,size[maxn],mx[maxn],dep[maxn];
void init() //初始化
{
    tot=0; res=0;
    memset(head,-1,sizeof head);
    memset(vis,0,sizeof vis);
}
void getsize(int x,int fa)  //得到size 以及 mx
{
    size[x]=1;mx[x]=0;
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int to=e[i].to;
        if(to==fa||vis[to]) continue;
        getsize(to,x);
        size[x]+=size[to];
        mx[x]=max(mx[x],size[to]);
    }
}
void getroot(int r,int x,int fa)    //找出重心
{
    mx[x]=max(mx[x],size[r]-size[x]);
    if(mx[x]<tmp) tmp=mx[x],root=x;
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int to=e[i].to;
        if(to==fa||vis[to]) continue;
        getroot(r,to,x);
    }
}
void getdep(int x,int deep,int fa)  //得到每个点的深度
{
    dep[num++]=deep;
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int to=e[i].to;
        if(to==fa||vis[to]) continue;
        getdep(to,deep+e[i].w,x);
    }
}
int cal(int v,int d)    //统计当前子树的答案
{
    int ans=0; num=0;
    getdep(v,d,0);
    sort(dep,dep+num);
    int i=0,j=num-1;
    while(i<j)
    {
        while(dep[i]+dep[j]>k&&i<j) j--;
        ans+= j-i;
        i++;
    }
    return ans;
}
void dfs(int v) //遍历所有子树
{
    tmp=n;
    getsize(v,0);
    getroot(v,v,0);
    res+=cal(root,0);
    vis[root]=1;
    for(int i=head[root];i!=-1;i=e[i].next)
    {
        int to=e[i].to;
        if(!vis[to])
        {
            res-=cal(to,e[i].w);
            dfs(to);
        }
    }
}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        if(n==0&&k==0) break;
        init();
        for(int i=1;i<n;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            add(x,y,z);
            add(y,x,z);
        }
        dfs(1);
        printf("%d\n",res);
    }
}