fzu2038

http://acm.fzu.edu.cn/problem.php?pid=2038

题意:告诉一个图的权值,求从任一点到所有点的权值之和

dfs按个点搜肯定t,可以计算每条边经过的次数(次数=边的左边的点数)

注意最后的和很大,long long 

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <map>
#include <queue>
#include <set>
#include <vector>
#include <stack>
using namespace std;
const int M=100005;
vector<pair<int,int>  >vt[M];
int vis[M];
int n;
long long sum;
long long  dfs(int u)
{
    long long  t=0;
    vis[u]=1;
    for(int i=0;i<vt[u].size();i++)
    {
        int x=vt[u][i].first;
        int y=vt[u][i].second;
        if(vis[x])
            continue;

        long long cnt=dfs(x);
        t+=cnt;//cout<<t<<endl;
        sum+=cnt*(n-cnt)*y;
    }
    return t+1;
}
int main()
{
    int t,i,j,k;
    scanf("%d",&t);
    for(int cas=1;cas<=t;cas++)
    {
        scanf("%d",&n);
        for(i=0;i<n;i++)
            vt[i].clear();
        int a,b,c;
        k=n-1;
        while(k--)
        {
            scanf("%d%d%d",&a,&b,&c);
            vt[a].push_back( make_pair(b,c) );
            vt[b].push_back( make_pair(a,c) );

        }//cout<<"ddddddd"<<endl;
        memset(vis,0,sizeof(vis));

        sum=0;
        dfs(0);
        printf("Case %d: ",cas);
        cout<<2*sum<<endl;
    }
    return 0;
}