[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);    }}