之前一段时间做的,但是因为懒所以没有写题解。有点忘了现在重新做一下,还是乖乖写题解吧~
树分治主要的在于要找重心。对于那些需要统计满足条件的路径条数的题,我们就可以想到树分治。
对于某一个点,我们可以将答案分成经过这个点的和不经过这个点的,那么经过这个点的所有答案可以快速统计出来,剩下的就只用递归到子树中去找就好了。
还是多做题吧~
下面是代码:
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 10005;
struct edge{
int to,next;
int c;
}e[N << 1];
int n,k;
int head[N],sz;
int size[N],mx[N];
int nsize,root;
int res;
bool done[N];
vector<int> dis;
void init(){
memset(head,-1,sizeof(head));
memset(done,0,sizeof(done));
sz = 0;
}
void addedge(int u,int v,int c){
e[sz].to = v;
e[sz].c = c;
e[sz].next = head[u];
head[u] = sz ++;
}
void getroot(int u,int f){
size[u] = 1;mx[u] = 0;
for(int i = head[u] ; i != -1 ; i = e[i].next){
int v = e[i].to;
if(v == f || done[v]) continue;
getroot(v,u);
size[u] += size[v];
mx[u] = max(mx[u],size[v]);
}
mx[u] = max(mx[u],nsize - size[u]);
if(mx[u] < mx[root]) root = u;
}
void getdis(int u,int f,int dist){
dis.push_back(dist);
for(int i = head[u] ; i != -1 ; i = e[i].next){
int v = e[i].to;
if(v == f || done[v]) continue;
getdis(v,u,dist + e[i].c);
}
}
int cal(int u,int c){
dis.clear();
getdis(u,0,c);
sort(dis.begin(),dis.end());
int res = 0;
int l = 0,r = dis.size() - 1;
while(l < r){
if(dis[l] + dis[r] <= k) res += r - l++;
else r --;
}
return res;
}
void deal(int u){
res += cal(u,0);
done[u] = 1;
for(int i = head[u] ; i != -1 ; i = e[i].next){
int v = e[i].to;
if(done[v]) continue;
res -= cal(v,e[i].c);
mx[0] = nsize = size[v];
getroot(v,root = 0);
deal(root);
}
}
void solve(){
init();
int u,v,c;
for(int i = 1 ; i < n ; i ++){
scanf("%d%d%d",&u,&v,&c);
addedge(u,v,c);
addedge(v,u,c);
}
res = 0;
mx[0] = nsize = n;
getroot(1,root = 0);
deal(root);
printf("%d\n",res);
}
int main()
{
int u,v;
while(~scanf("%d%d",&n,&k)){
if(!n && !k) break;
solve();
}
return 0;
}