codeup24697 生成输入数据

题目描述

生成输入数据 input

【问题描述】

首先看到题目别太开心，这题可不是让你出数据～^_*

背景神马的就忽略了。这题就是给你一棵带边权的树，然后这棵树是某个完全图唯一的最小生成树。问原来的完全图中所有边可能的最小边权和是多少。

完全图是任意两个点之间都有边相连的图。

【输入格式】

第一行包含一个整数T表示数据组数。

每组数据第一行一个整数N表示点数。

接下来N-1行每行三个整数ai，bi，wi表示最小生成树上ai和bi之间有一条权值为wi的边。

【输出格式】

输出应有T行，每行表示一组数据的答案。

【样例输入输出】

input.in	input.out
2 3 1 2 4 2 3 7 4 1 2 1 1 3 1 1 4 2	19 12

【数据说明】

20%的数据满足：T≤5，n≤5，wi≤5

另外30%的数据满足：n≤1000，给定的树是一条链

100%的数据满足：T≤10，n≤20000，wi≤10000

 

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <map>
#include <set>
#include <vector>
#include <queue>
#include <cmath>

using namespace std;

const int N = 20020;

int T, n, fa[N];
long long ans, size[N];

struct E {
    int u, v, w;
} e[N];

bool operator < (E a, E b) {
    return a.w < b.w;
}

int find(int x) {
    return x == fa[x] ? x : fa[x] = find(fa[x]);
}

int main() {
    scanf("%d", &T);
    while(T --) {
        ans = 0;
        scanf("%d", &n);
        for(int i = 1 ; i < n ; i ++) {
            scanf("%d%d%d", &e[i].u, &e[i].v, &e[i].w);
            fa[i] = i;
            size[i] = 1;
        }
        fa[n] = n;
        size[n] = 1;
        sort(e + 1, e + n);
        for(int i = 1 ; i < n ; i ++) {
            int u = find(e[i].u), v = find(e[i].v);
            if(u != v) {
                ans += (size[u] * size[v] - 1) * (e[i].w + 1);
                size[u] += size[v];
                fa[v] = u;
                ans += e[i].w;
            }
        }
        printf("%lld\n", ans);
    }
}