题目:给定一棵N(1<=N<=10000)个结点的带权树,定义dist(u,v)为u,v两点间的最短路径长度,路径的长度定义为路径上所有边的权和。再给定一个K(1<=K<=10^9),如果对于不同的两个结点a,b,如果满足dist(a,b)<=K,则称(a,b)为合法点对。求合法点对个数。
算法:树上分治基础题
第一步:枚举中间节点(x是头结点也按照中间节点算):假设一部分合法点对会经过点x(当然x可以是N个节点中的任意一个)。当然这个x在树形结构中可以按根节点树形递归枚举。
第二步:计算以x为根节点的子树中所有节点到x节点的路径长度放在dep[top]数组中,具体方法是DFS搜索x的所有子树
第三步:计算点对:这一步最为关键 首先将dep[]数组从小到大排序,O(n)复杂度求出dep[]中任意两个和<=K的数量(这里请仔细想想是如何做到的),我们记作sum1。这样算下来会有不合法的点对,因为在同一棵子树中的两个dep[i]+dep[j]<=K 会共用公共边,即有重边。我们在递归的过程中把x为根的子树中会在sum1中重复计算的点对个数都统计在sum2中 最后sum1-sum2即为以x节点为中间节点的点对的个数。依次递归即可得到答案。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
#define N 11000
#define inf 0x0f0f0f0f
struct Edge{
int nxt,val;
Edge(){}
Edge(int _n,int _v){
nxt = _n; val = _v;
}
};
vector<Edge> edg[N];
int siz[N],f[N],vis[N],dep[N];
int n,ans,k,root;
int max(int a,int b){ return a>b?a:b; }
int min(int a,int b){ return a<b?a:b; }
void getRoot(int cur,int fa,int sum){
int tmp = 0; siz[cur] = 1;
for(int i = 0;i<(int)edg[cur].size();i++){
int nxt = edg[cur][i].nxt;
if(vis[nxt]||nxt==fa) continue;
getRoot(nxt,cur,sum);
siz[cur] += siz[nxt];
tmp = max(tmp,siz[nxt]);
}
f[cur] = max(tmp,sum-siz[cur]);
if(f[root]>f[cur]) root = cur;
}
void getDep(int cur,int fa,int len,int &top){
dep[++top] = len;
for(int i = 0;i<(int)edg[cur].size();i++){
int nxt = edg[cur][i].nxt;
int val = edg[cur][i].val;
if(vis[nxt]||nxt==fa) continue;
getDep(nxt,cur,len+val,top);
}
}
int getTimes(int t_root,int len){
int top = 0,tmp = 0;
getDep(t_root,-1,len,top);
sort(dep+1,dep+top+1);
for(int l = 1,r = top;l<r;){
if(dep[l]+dep[r]<=k){
tmp += r-l;
l++;
}
else r--;
}
return tmp;
}
int getRootID(int cur,int sum){
root = 0; f[root] = inf;
getRoot(cur,-1,sum);
getRoot(root,-1,sum);
return root;
}
void CDQ(int t_root){
vis[t_root] = 1;
ans += getTimes(t_root,0);
for(int i = 0;i<(int)edg[t_root].size();i++){
int nxt = edg[t_root][i].nxt;
int val = edg[t_root][i].val;
if(vis[nxt]==1) continue;
ans -= getTimes(nxt,val);
nxt = getRootID(nxt,siz[nxt]);
CDQ(nxt);
}
vis[t_root] = 0;
}
void solve(int n){
ans = 0;
int t_root = getRootID(1,n);
CDQ(t_root);
printf("%d\n",ans);
}
void init(){
for(int i = 1;i<=n;i++) edg[i].clear();
memset(vis,0,sizeof(vis));
}
int main(){
int a,b,v;
//freopen("Test.txt","r",stdin);
while(~scanf("%d%d",&n,&k)){
init();
if(n==0&&k==0) break;
for(int i = 1;i<=n-1;i++){
scanf("%d%d%d",&a,&b,&v);
edg[a].push_back(Edge(b,v));
edg[b].push_back(Edge(a,v));
}
solve(n);
}
return 0;
}