poj1741 Tree

树的路径统计

统计有多少条长度不大于k的路径

树的点分治

做法：将路径分为经过重心和不经过重心的路径。1）经过重心，以重心为根进行统计，需要知道每点对于重心的深度。所有两个深度之和<=k的路径数-属于同一子树两个深度之和<=k的路径数即为经过该重心的长度不大于k的路径数。2）不经过重心，将重心除去，分为x个子树，分别寻找每个子树的重心，重复上述步骤即可。

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#define maxn 20010
#define lowbit(x) x&(-x)
#define mem(a,b) memset(a,b,sizeof(a))
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;

int n;
int k;
struct node{
    int to,next,w;
}edge[maxn*2];
int head[maxn],tot;
int sum,f[maxn],vis[maxn],son[maxn],core,deep[maxn],d[maxn];//getcore
int ans;

void init(){
    mem(head,-1);
    mem(vis,0);
    tot=0;
    ans=0;
}

void add(int a,int b,int c){
    edge[tot].to=b,edge[tot].next=head[a],edge[tot].w=c;
    head[a]=tot++;
}

void getcore(int u,int fa){
    son[u]=1,f[u]=0;//每次更新前必须初始化！！！---dfs可以在搜索子树前在内部对点初始化
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(v==fa||vis[v]) continue;//除去已找到的重心后求子树重心
        getcore(v,u);
        son[u]+=son[v],f[u]=max(f[u],son[v]);
    }
    f[u]=max(f[u],sum-son[u]);
    if(f[u]<f[core]) core=u;//core记录该树重心
}

void getdeep(int u,int fa){//得到u子树每个节点的深度
    deep[++deep[0]]=d[u];
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to,w=edge[i].w;
        if(v==fa||vis[v]) continue;
        d[v]=d[u]+w;
        getdeep(v,u);
    }
}

int cal(int u,int w){//1.该点是重心：求该点子树每点到该点深度2.该点父节点是重心：求该点子树到父节点的深度
    deep[0]=0,d[u]=w,getdeep(u,0);
    int tmp=0;
    sort(deep+1,deep+deep[0]+1);//<strong><span style="color:#ff0000;">技巧</span></strong>
    for(int l=1,r=deep[0];l<r;)
        if(deep[l]+deep[r]<=k) tmp+=(r-l),l++;
        else r--;
    return tmp;
}

void work(int u){
    ans+=cal(u,0);//经过原树重心的k路径条数
    vis[u]=1;//原树重心已算，标记
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to,w=edge[i].w;
        if(vis[v]) continue;//如果v已经作为重心枚举过，u不会遍历到v，因此不必除去
        ans-=cal(v,w);//除去经过重心同一子树的路径
        sum=son[v];//更改当前树节点总和
        core=0;
        getcore(v,core);
        work(core);
    }
}

int main(){
    //freopen("a.txt","r",stdin);
    while(scanf("%d%d",&n,&k)!=EOF){
        if(!n&&!k) break;
        init();
        for(int i=0;i<n-1;i++){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c),add(b,a,c);
        }
        core=0,sum=n,f[0]=INF;
        getcore(1,0);
        work(core);
        printf("%d\n",ans);
    }
    return 0;
}

易错点：1、sum、core、vis[u]的更新 2、边集开两倍数组大小

技巧：统计一列数和<=k的数对数量

扫描法。1）先将数从小到达排序2）每次统计左端点作为较小数时的数对对数。详见代码