题目大意：同POJ1741
链接：http://blog.csdn.net/popoqqq/article/details/39959803
和POJ1741一样的题，土豪题，POJ有道一样的题，做完1741可以水水
我会告诉你我直接改了下上题的代码么0.0
数据范围4W 距离最大1000 不会爆INT 放心写吧
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 40400
using namespace std;
struct abcd{
int to,f,next;
bool ban;
}table[M<<1];
int head[M],tot=1;
int n,m,k,ans;
int siz[M],fa[M],dist[M];
int stack[M],top;
void Add(int x,int y,int z)
{
table[++tot].to=y;
table[tot].f=z;
table[tot].next=head[x];
head[x]=tot;
table[tot].ban=0;
}
void Get_Centre_Of_Gravity(int x,int size,int &cg)
{
int i;
bool flag=1;
siz[x]=1;
for(i=head[x];i;i=table[i].next)
if(!table[i].ban)
if(table[i].to!=fa[x])
{
fa[table[i].to]=x;
Get_Centre_Of_Gravity(table[i].to,size,cg);
if(siz[table[i].to]>size>>1)
flag=0;
siz[x]+=siz[table[i].to];
}
if(size-siz[x]>size>>1)
flag=0;
if(flag)
cg=x;
}
void DFS1(int x,int dis)
{
int i;
dist[x]=dis;
for(i=head[x];i;i=table[i].next)
{
if(table[i].ban)
continue;
if(table[i].to==fa[x])
continue;
fa[table[i].to]=x;
DFS1(table[i].to,dis+table[i].f);
}
}
void DFS2(int x)
{
int i;
stack[++top]=dist[x];
for(i=head[x];i;i=table[i].next)
{
if(table[i].ban)
continue;
if(table[i].to==fa[x])
continue;
DFS2(table[i].to);
}
}
int Calculate(int root)
{
int i,j,re=0;
top=0;
DFS2(root);
sort(stack+1,stack+top+1);
for(i=1,j=top;i<=top;i++)
{
for(;j&&stack[j]+stack[i]>k;j--);
re+=j;
}
return re;
}
void Tree_Divide_And_Conquer(int root,int size)
{
int i,cg;
fa[root]=0;
Get_Centre_Of_Gravity(root,size,cg);
siz[fa[cg]]=size-siz[cg];
fa[cg]=0;
DFS1(cg,0);
ans+=Calculate(cg);
for(i=head[cg];i;i=table[i].next)
if(!table[i].ban)
{
table[i].ban=table[i^1].ban=1;
ans-=Calculate(table[i].to);
Tree_Divide_And_Conquer(table[i].to,siz[table[i].to]);
}
}
int main()
{
int i,x,y,z;
char p[10];
cin>>n>>m;
for(i=1;i<n;i++)
scanf("%d%d%d%s",&x,&y,&z,p),Add(x,y,z),Add(y,x,z);
cin>>k;
Tree_Divide_And_Conquer(1,n);
cout<<(ans-n>>1)<<endl;
}
BZOJ 3365 Distance Statistics 树的点分治
原文地址：http://blog.csdn.net/popoqqq/article/details/39960413
