#include <cstdio>
#include <algorithm>
#include <set>
using namespace std;
int c[2014][2014];
int parent[2014], tsize[2014];
bool cmp(const pair<int, int> &a, const pair<int, int> &b) { return c[a.first][a.second] < c[b.first][b.second]; }
int par(int a) {
if (parent[a] == a) return a;
return parent[a] = par(parent[a]);
}
void join(int a, int b) {
if (tsize[a] >= tsize[b]) {
tsize[a] += tsize[b];
parent[b] = a;
} else join(b,a);
}
int main() {
int n, s=0;
scanf("%d", &n);
pair<int, int> t[n*n];
for (int i=0; i<n; i++) {
for (int j=i; j<n; j++) {
scanf("%d", &c[i][j+1]);
t[s++] = make_pair(i,j+1);
}
}
for (int i=0; i<=n; i++){
parent[i] = i;
tsize[i] = 1;
}
sort(t,t+s,cmp);
long long res = 0;
for (int i=0; i<s; i++) {
int x = par(t[i].first), y = par(t[i].second);
if (x != y) {
res += c[t[i].first][t[i].second];
join(x, y);
}
}
printf("%lld\n", res);
return 0;
}
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 | #include <cstdio> #include <algorithm> #include <set> using namespace std; int c[2014][2014]; int parent[2014], tsize[2014]; bool cmp(const pair<int, int> &a, const pair<int, int> &b) { return c[a.first][a.second] < c[b.first][b.second]; } int par(int a) { if (parent[a] == a) return a; return parent[a] = par(parent[a]); } void join(int a, int b) { if (tsize[a] >= tsize[b]) { tsize[a] += tsize[b]; parent[b] = a; } else join(b,a); } int main() { int n, s=0; scanf("%d", &n); pair<int, int> t[n*n]; for (int i=0; i<n; i++) { for (int j=i; j<n; j++) { scanf("%d", &c[i][j+1]); t[s++] = make_pair(i,j+1); } } for (int i=0; i<=n; i++){ parent[i] = i; tsize[i] = 1; } sort(t,t+s,cmp); long long res = 0; for (int i=0; i<s; i++) { int x = par(t[i].first), y = par(t[i].second); if (x != y) { res += c[t[i].first][t[i].second]; join(x, y); } } printf("%lld\n", res); return 0; } |
English