[POJ1741][BZOJ1468]经典点分题

题目

POJ1741传送门
BZOJ1468传送门

题意

给定一棵n各节点的无根树，节点之间有边权。再给定一个整数K，求出书中路径长度小于等于K的边数。题目有多组数据，输入以0 0结尾。数据范围：POJ上N小于等于10000，BZOJ上N小于等于40000

---

解法

对于每个子结构（子树），dfs1找出重心和siz，并把重心vis打成1（避免以后再访问，本来是要删除重心的，但是删除麻烦就用标记代替），以后访问到vis过的就continue，dfs2找出过当前重心的路径长度（dis数组）
再用队列维护（一个head++，tail–的队列，理解看代码）所有长度少于K的过重心的路径，条数加起来。
再在进入下一棵子树前减掉来自同一棵子树的路径，递归。

话说写的时候没有看到POJ的多组数据。比较傻的WA了好几次

---

自带大常数的代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
int n,k,tw=0,dis[10005],rt,minsiz,nowsiz;
long long ans=0;
int siz[10005],vis[10005],head[10005];
vector <int> d;
struct node{
    int pre,to,len;
}w[20005];

void In(int t1,int t2,int t3){
    w[++tw].pre=head[t1];
    w[tw].to=t2;
    w[tw].len=t3;
    head[t1]=tw;
}

void dfs(int u,int f){
    siz[u]=1;
    int maxnow=-1;
    for(int i=head[u];i;i=w[i].pre){
        int v=w[i].to;
        if(vis[v]||v==f)    continue;
        dfs(v,u);
        siz[u]+=siz[v];
        maxnow=max(maxnow,siz[v]);
    }
    if(minsiz>max(maxnow,nowsiz-siz[u])){
        minsiz=max(maxnow,nowsiz-siz[u]);
        rt=u;
    }
    return;
}

void getdis(int u,int f){
    d.push_back(dis[u]);
    for(int i=head[u];i;i=w[i].pre){
        int v=w[i].to;
        if(vis[v]||v==f)    continue;
        dis[v]=dis[u]+w[i].len;
        getdis(v,u);
    }
}

int cnt(int u,int x){
    int ans1=0;
    d.clear();
    dis[u]=x;
    getdis(u,0);
    sort(d.begin(),d.end());
    int l=0,r=d.size()-1;
    while(l<r){
        while(d[l]+d[r]>k&&l<r) r--;
        ans1+=r-l;
        l++;
    }
    return ans1;
}

int work(int u){
    vis[u]=true;
    ans+=cnt(u,0);
    for(int i=head[u];i;i=w[i].pre){
        int v=w[i].to;
        if(vis[v])  continue;
        minsiz=0x3f3f3f3f;
        ans-=cnt(v,w[i].len);
        nowsiz=siz[v];
        dfs(v,u);
        work(rt);
    }
    return ans;
}

int main(){
    while(1){
        scanf("%d",&n);scanf("%d",&k);
        if(n==0&&k==0)  return 0;
        int t1,t2,t3;
        memset(vis,false,sizeof(vis));
        memset(head,0,sizeof(head));
        tw=0;ans=0;
        for(int i=1;i<=n-1;i++){
            scanf("%d%d%d",&t1,&t2,&t3);
            In(t1,t2,t3);
            In(t2,t1,t3);
        }
        nowsiz=n;
        minsiz=0x3f3f3f3f;
        dfs(1,0);
        work(rt);
        printf("%lld\n",ans);
    }
}