#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; } |