bzoj1468
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define Maxn 80010
using namespace std;
struct edge{
int to,v,next;
}p[Maxn];
int tot,root,sum,ans,k;
int head[Maxn],vis[Maxn],son[Maxn],bal[Maxn],st[Maxn],dist[Maxn];
void addedge(int a,int b,int c){
p[tot].to=b;
p[tot].v=c;
p[tot].next=head[a];
head[a]=tot++;
}
void getroot(int u,int fa){
son[u]=1,bal[u]=0;
for(int i=head[u];i!=-1;i=p[i].next){
int v=p[i].to;
if(fa==v||vis[v]) continue;
son[u]+=son[v];
bal[u]=max(bal[u],son[v]);
}
bal[u]=max(bal[u],sum-son[u]);
if(bal[u]<bal[root]) root=u;
}
void dfs(int u,int fa){
st[++st[0]]=dist[u];
for(int i=head[u];i!=-1;i=p[i].next){
int v=p[i].to;
if(v==fa||vis[v]) continue;
dist[v]=dist[u]+p[i].v;
dfs(v,u);
}
}
int cal(int u,int val){
dist[u]=val,st[0]=0;
dfs(u,-1);
sort(st+1,st+1+st[0]);
int res=0;
for(int i=1,j=st[0];i<j;){
if(st[i]+st[j]<=k){res+=j-i;i++;}
else j--;
}
return res;
}
void work(int u){
ans+=cal(u,0);
vis[u]=1;
for(int i=head[u];i!=-1;i=p[i].next){
int v=p[i].to;
if(vis[v]) continue;
ans-=cal(v,p[i].v);
sum=son[v],root=0;
getroot(v,-1);
work(root);
}
}
int main()
{
int n,a,b,c;
cin>>n;
memset(head,-1,sizeof head);
tot=0;
for(int i=1;i<n;i++){
scanf("%d%d%d",&a,&b,&c);
addedge(a,b,c);
addedge(b,a,c);
}
cin>>k;
ans=0;
sum=n,bal[root=0]=1<<30;
getroot(1,-1);
work(root);
cout<<ans<<endl;
return 0;
}
bzoj2152
#include<iostream>
#include<cstdio>
#include<cstring>
#define Maxn 40010
using namespace std;
struct edge{
int to,v,next;
}p[Maxn];
int head[Maxn],vis[Maxn],bal[Maxn],mod[Maxn],dist[Maxn],son[Maxn];
int tot,sum,root,ans;
const int inf=0x3f3f3f3f;
void addedge(int a,int b,int c){
p[tot].to=b;
p[tot].v=c;
p[tot].next=head[a];
head[a]=tot++;
}
void getroot(int u,int fa){ //sum,root
son[u]=1,bal[u]=0;
for(int i=head[u];i!=-1;i=p[i].next){
int v=p[i].to;
if(v==fa||vis[v]) continue;
getroot(v,u);
son[u]+=son[v];
bal[u]=max(bal[u],son[v]);
}
bal[u]=max(bal[u],sum-son[u]);
if(bal[u]<bal[root]) root=u;
}
void dfs(int u,int fa){
mod[dist[u]]++;
for(int i=head[u];i!=-1;i=p[i].next){
int v=p[i].to;
if(v==fa||vis[v]) continue;
dist[v]=(dist[u]+p[i].v)%3;
dfs(v,u);
}
}
int cal(int u,int val){
mod[0]=mod[1]=mod[2]=0;
dist[u]=val%3;
dfs(u,-1);
return mod[0]*mod[0]+mod[1]*mod[2]*2;
}
void work(int u){
ans+=cal(u,0);
vis[u]=1;
for(int i=head[u];i!=-1;i=p[i].next){
int v=p[i].to;
if(vis[v]) continue;
ans-=cal(v,p[i].v);
sum=son[v],root=0;
getroot(v,-1);
work(root);
}
}
int gcd(int a,int b){
return b==0?a:gcd(b,a%b);
}
int main()
{
int n,a,b,c;
cin>>n;
tot=ans=0;
memset(head,-1,sizeof head);
memset(vis,0,sizeof vis);
for(int i=1;i<n;i++){
scanf("%d%d%d",&a,&b,&c);
addedge(a,b,c);
addedge(b,a,c);
}
sum=n,bal[root=0]=inf;
getroot(1,-1);
work(root);
int t=gcd(n*n,ans);
printf("%d/%d\n",ans/t,n*n/t);
return 0;
}