问题描述
众所周知,moreD的宠物已经被moreD奴役得体无完肤。这只宠物实在忍无可忍,把自己每天走魔法树的经历告诉了自己的宠物。同时他还说明了自己爬树是多么地慢,以至于moreD每天都残酷地训练他爬树。幸运的是moreD的宠物的宠物不是moreD的宠物,moreD的宠物深知”宠物是用来宠的而不是用来奴役的”这一点,所以moreD的宠物对待自己的宠物很有爱。所以moreD的宠物与其宠物商量着要推翻moreD的暴政,方法是把moreD告上法庭,就以自己每天被迫爬树来做证据。
由于魔法树是树,训练树当然也是树啦。
moreD的训练有着GX的文化,每天moreD会把自己的宠物通灵到树的一个端点上,这个通灵点可能与之前的通灵点相同。然后moreD命令他的宠物从这个点开始走,让这只宠物随便访问自己该天之前没有访问过的节点,一直走到该天无路可走,训练才会结束,宠物才可以休息。
moreD的宠物每天都会在这棵树上训练,幸运的是他每天走过的训练路径都不同,直到若干天后,所有可能的训练路径都被走遍了。
你,作为moreD宠物的宠物,一只被moreD的宠物宠着的宠物,当然想帮moreD的宠物算出他总共走过的路径长度啦。
输入格式
第一行包含两个正整数N,M,表示树的点数与边数。接下来M行,每行三个正整数表示Li,bi,ci分别表示树上有一条长度为Li的连接bi,ci两个结点的边。
输出格式
仅一行表示答案。输入样例
5 41 2 1
1 3 1
2 4 2
2 5 2
输出样例
37数据范围
对于30%的数据N≤300对于70%的数据N≤3,000
对于100%的数据对于所有输入的整数均不大于100,000,输入的树保证连通,无重边,无自环。
样例解释
可能的训练路径有(1-2-4),(1-2-5),(1-3),(2-4),(2-5),(2-1-3),(3-1-2-4),(3-1-2-5)(4-2-5),(4-2-1-3),(5-2-4),(5-2-1-3)
长度依次为3,3,1,2,2,2,4,4,4,4,4,4。和为37.
题解
怪怪的题目,其实就是要求出所有“两个端点至少有一个为叶子节点”的路径长度和。只要考虑每个边对答案的贡献次数即可。注意:long long强转要放在每个变量前,否则是乘完再转。
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<cmath>
#include<algorithm>
#define ll long long
#define MAXN 100002
using namespace std;
int n,m,zz,head[MAXN],con[MAXN],root;
struct bian{int frm,to,nx,v;} e[MAXN*2];
ll son[MAXN],leaf[MAXN],fa[MAXN];
ll ans;
//------------------------------------------------------------------------------
void insert(int x,int y,int z)
{
zz++; e[zz].to=y; e[zz].v=z; e[zz].nx=head[x]; head[x]=zz;
zz++; e[zz].to=x; e[zz].v=z; e[zz].nx=head[y]; head[y]=zz;
}
void init()
{
scanf("%d%d",&n,&m);
int i,x,y,z;
for(i=1;i<=m;i++)
{scanf("%d%d%d",&z,&x,&y);
insert(x,y,z);
//con[x]++; con[y]++;
}
//for(i=1;i<=n;i++)
//{if(con[i]>1) {root=i; break;}}
}
void dfs(int x)
{
int tag=0,i,p;
son[x]=1;
for(i=head[x];i;i=e[i].nx)
{p=e[i].to;
if(p==fa[x]) continue;
tag=1;
fa[p]=x; dfs(p);
son[x]+=son[p]; leaf[x]+=leaf[p];
}
if(!tag) leaf[x]=1;
}
void dp(int x)
{
int i,p;
for(i=head[x];i;i=e[i].nx)
{p=e[i].to;
if(p==fa[x]) continue;
ans+=(ll) (leaf[p]*(son[1]-son[p])+(leaf[1]-leaf[p])*son[p])*e[i].v;
dp(p);
}
}
int main()
{
freopen("senso.in","r",stdin);
freopen("senso.out","w",stdout);
init(); dfs(1);
//for(int i=1;i<=n;i++) printf("%d: %d %d\n",i,son[i],leaf[i]);
dp(1);
printf("%I64d\n",ans);
return 0;
}