#include <cstdio> #include <vector> #include <algorithm> #define FOR(i,n) for(int i = 0; i < (n); i++) #define REP(i,n) for(int i = 1; i <= (n); i++) typedef long long LL; using namespace std; const int MAX = 2014; struct Edge { int u,v; // u <= v int c; Edge(){} Edge(int u, int v, int c): u(u), v(v), c(c){} }; bool cmd(const Edge &a, const Edge &b) { return a.c < b.c; } int n; vector<Edge> E; int R[MAX], P[MAX]; int get_par(int x) { if(x == P[x]) return x; P[x] = get_par(P[x]); return P[x]; } void link(int a, int b) { a = get_par(a); b = get_par(b); if(R[a] == R[b]) R[a] ++; if(R[a] < R[b]) swap(a,b); P[b] = a; } int main() { int c; long long ans = 0; scanf("%d", &n); REP(i,n) { for(int j = i; j <= n; j++) { scanf("%d", &c); E.push_back(Edge(i - 1,j,c)); } } sort(E.begin(), E.end(), cmd); FOR(i,n+1) P[i] = i; for(auto e: E) { if(get_par(e.u) == get_par(e.v)) continue; link(e.u, e.v); ans += (LL)e.c; } printf("%lld\n", ans); }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | #include <cstdio> #include <vector> #include <algorithm> #define FOR(i,n) for(int i = 0; i < (n); i++) #define REP(i,n) for(int i = 1; i <= (n); i++) typedef long long LL; using namespace std; const int MAX = 2014; struct Edge { int u,v; // u <= v int c; Edge(){} Edge(int u, int v, int c): u(u), v(v), c(c){} }; bool cmd(const Edge &a, const Edge &b) { return a.c < b.c; } int n; vector<Edge> E; int R[MAX], P[MAX]; int get_par(int x) { if(x == P[x]) return x; P[x] = get_par(P[x]); return P[x]; } void link(int a, int b) { a = get_par(a); b = get_par(b); if(R[a] == R[b]) R[a] ++; if(R[a] < R[b]) swap(a,b); P[b] = a; } int main() { int c; long long ans = 0; scanf("%d", &n); REP(i,n) { for(int j = i; j <= n; j++) { scanf("%d", &c); E.push_back(Edge(i - 1,j,c)); } } sort(E.begin(), E.end(), cmd); FOR(i,n+1) P[i] = i; for(auto e: E) { if(get_par(e.u) == get_par(e.v)) continue; link(e.u, e.v); ans += (LL)e.c; } printf("%lld\n", ans); } |