现在的位置: 首页 > 综合 > 正文

hdu5176—The Experience of Love

2019年02月14日 ⁄ 综合 ⁄ 共 1681字 ⁄ 字号 评论关闭

依次考虑每一条边,对边排序
考虑每条边作为路径最大值,对边从小到大排序,用并查集记录边的两个端点所在集合的点的数目a,b 因为目前集合里不存在比当前边更大的边,所以a×b就是它贡献的了
考虑每条边作为路径最小值的做法同上,注意用long long 不够,要用 usigned long long

/*************************************************************************
    > File Name: bc-30c.cpp
    > Author: ALex
    > Mail: zchao1995@gmail.com 
    > Created Time: 2015年02月14日 星期六 21时07分25秒
 ************************************************************************/

#include <map>
#include <set>
#include <queue>
#include <stack>
#include <vector>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#
const double pi = acos(-1);
const int inf = 0x3f3f3f3f;
const double eps = 1e-15;
typedef unsigned long long LL;
typedef pair <int, int> PLL;

const int N = 150010;
int father[N];
LL num[N];

struct node
{
    int u, v;
    LL w;
}edge[N];

int cmp1 (node a, node b)
{
    return a.w < b.w;
}

int cmp2 (node a, node b)
{
    return a.w > b.w;
}

int find (int x)
{
    if (father[x] == -1)
    {
        return x;
    }
    return father[x] = find (father[x]);
}

int main ()
{
    int n;
    int icase = 1;
    while (~scanf("%d", &n))
    {
        for (int i = 1; i <= n - 1; ++i)
        {
            father[i] = -1;
            num[i] = 1;
            scanf("%d%d%llu", &edge[i].u, &edge[i].v, &edge[i].w);
        }
        father[n] = -1;
        num[n] = 1;
        sort (edge + 1, edge + n, cmp1);
        LL ans = 0;
        for (int i = 1; i < n; ++i)
        {
            int u = find (edge[i].u);
            int v = find (edge[i].v);
            ans += edge[i].w * num[u] * num[v];
            num[u] += num[v];
            father[v] = u;
        }
        for (int i = 1; i <= n; ++i)
        {
            father[i] = -1;
            num[i] = 1;
        }
        sort (edge + 1, edge + n, cmp2);
        for (int i = 1; i < n; ++i)
        {
            int u = find (edge[i].u);
            int v = find (edge[i].v);
            ans -= edge[i].w * num[u] * num[v];
            num[u] += num[v];
            father[v] = u;
        }
        printf("Case #%d: %llu\n", icase++, ans);
    }
    return 0;
}

抱歉!评论已关闭.