2018.07.31 POJ1741Tree

传送门
只是来贴一个点分治的板子(年轻时候写的丑别介意)。
代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 10010
#define INF 1e9+10
struct node{int v,c,next;}e[N*2];
int first[N],m;
int son[N],f[N];
bool in[N];
int d[N],dep[N];
int n,sum,root,k,ans;
inline void add(int u,int v,int cost){//邻接表存储边
    e[++m].next=first[u];
    first[u]=m;
    e[m].v=v; 
    e[m].c=cost;
}
void getroot(int v,int fa){//找出当前树的重心
    son[v]=1; 
    f[v]=0;
    for(int i=first[v];i;i=e[i].next)
        if(e[i].v!=fa&&!in[e[i].v]){
            getroot(e[i].v,v);
            son[v]+=son[e[i].v];
            f[v]=max(f[v],son[e[i].v]);
        }
    f[v]=max(f[v],sum-son[v]);
    if(f[v]<f[root])root=v;//更新根节点
}
void getdeep(int v,int fa){//预处理到当前根节点的距离
    dep[++dep[0]]=d[v];
    for(int i=first[v];i;i=e[i].next)
        if(e[i].v!=fa&&!in[e[i].v]){
            d[e[i].v]=d[v]+e[i].c;
            getdeep(e[i].v,v);
        }
}
int cal(int v,int cost){//用双指针法求出当前子树信息
    d[v]=cost; 
    dep[0]=0;
    getdeep(v,0);
    sort(dep+1,dep+dep[0]+1);
    int l=1,r=dep[0],sum=0;
    while(l<r){//双指针
        if(dep[l]+dep[r]<=k){
            sum+=r-l;
            ++l;
        }
        else --r;
    }
    return sum;
}
void solve(int v){//求出子问题答案
    ans+=cal(v,0);
    in[v]=1;
    for(int i=first[v];i;i=e[i].next)
        if(!in[e[i].v]){
            ans-=cal(e[i].v,e[i].c);//去重
            sum=son[e[i].v];
            root=0;
            getroot(e[i].v,0);
            solve(root);
        }
}
int main(){
    int u,v,w;
    while(scanf("%d%d",&n,&k)&&n&&k){
        ans=root=m=0;
        memset(in,0,sizeof(in));
        memset(first,0,sizeof(first));
        for(int i=1;i<n;++i){
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
            add(v,u,w);
        }
        f[0]=INF;//赋极大初值来找重心
        sum=n;
        getroot(1,0);//找重心
        solve(root);//点分治求解
        printf("%d\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/dreaming__ldx/article/details/81297433