传送门
题解:裸的点分治,统计距离不超过k的点对数量。好像题目没说是一棵树,但是应该是默认的,不然两点的距离就不唯一了。
注意:点分治中getdis( )要再求一次size,以前本人写bzoj 1468的题解时扯的那些*话就当没听见,什么整体偏大不影响都是扯淡的,这个操作不执行就会导致重心找偏,有些题数据水一样能A,但是另外的就会出问题。
这次真的是稳稳地自己调出来的。<( ̄︶ ̄)>
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=4e4+8;
int n,m,K,siz[MAXN],mx[MAXN],sum,root,ans=0,tot;
ll a[MAXN],dis[MAXN];
bool vis[MAXN];
int head[MAXN],edge=0;
struct EDGE {
int v,nxt,w;
}e[MAXN<<1];
void getroot(int p,int fa) {
siz[p]=1,mx[p]=0;
for (int i=head[p];~i;i=e[i].nxt) {
int v=e[i].v;
if (v^fa&&!vis[v]) {
getroot(v,p);
siz[p]+=siz[v];
mx[p]=max(mx[p],siz[v]);
}
}
mx[p]=max(mx[p],sum-siz[p]);
if (mx[p]<mx[root]) root=p;
}
void getdis(int p,int fa) {
siz[p]=1,a[++tot]=dis[p];
for (int i=head[p];~i;i=e[i].nxt) {
int v=e[i].v;
if (v^fa&&!vis[v]) {
dis[v]=dis[p]+e[i].w;
getdis(v,p);
siz[p]+=siz[v];//!!!
}
}
}
inline int cal(int p,int val) {
int ret=0;
tot=0;
dis[p]=val;
getdis(p,0);
sort(a+1,a+tot+1);
for (int l=1,r=tot;l<r;++l) {
while (a[r]+a[l]>K&&l<r)
--r;
ret+=r-l;
}
return ret;
}
void work(int p) {
ans+=cal(p,0);
vis[p]=true;
for (int i=head[p];~i;i=e[i].nxt) {
int v=e[i].v;
if (!vis[v]) {
ans-=cal(v,e[i].w);
sum=siz[v],root=0;
getroot(v,0);
work(root);
}
}
}
inline int read() {
int x=0,f=1;char c=getchar();
while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
return x*f;
}
inline void adde(int u,int v,int w) {
e[edge].nxt=head[u],e[edge].v=v,e[edge].w=w,head[u]=edge++;
e[edge].nxt=head[v],e[edge].v=u,e[edge].w=w,head[v]=edge++;
}
int main() {
// freopen("bzoj 3365.in","r",stdin);
memset(head,-1,sizeof(head));
memset(vis,false,sizeof(vis));
n=read(),m=read();
for (register int i=1;i<=m;++i) {
int u=read(),v=read(),w=read();
char st[2];scanf("%s",st);
adde(u,v,w);
}
K=read();
sum=n,root=0,mx[0]=MAXN;
getroot(1,0);
work(root);
printf("%d\n",ans);
return 0;
}