题目
题意
给定一棵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);
}
}