Nik_Perepelov

6 контест 1 задача

Oct 20th, 2021
777
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. #include<iostream>
  2. #include<vector>
  3. #include<algorithm>
  4.  
  5. using namespace std;
  6. vector<vector<int>> g;
  7. vector<bool> used;
  8. vector<int> min_e;
  9.  
  10. void prim(int n){
  11.     min_e[0] = 0;
  12.     for (int i = 0; i < n; i++) {
  13.         int v = -1;
  14.         for (int j = 0; j < n; j++) {
  15.             if (!used[j] && (v == -1 || min_e[j] < min_e[v])) {
  16.                 v = j;
  17.                 break;
  18.             }
  19.         }
  20.         if (v == -1)
  21.             break;
  22.         used[v] = true;
  23.         for (int to = 0; to < n; to++) {
  24.             if (g[v][to] < min_e[to]) {
  25.                 min_e[to] = g[v][to];
  26.             }
  27.         }
  28.     }
  29. }
  30.  
  31. int main() {
  32.     int n, m;
  33.     cin >> n >> m;
  34.     g = vector<vector<int>>(n, vector<int>(n, 1e9));
  35.     used = vector<bool>(n);
  36.     min_e = vector<int>(n, 1e9);
  37.     for (int i = 0; i < m; i++) {
  38.         int fr, to, value;
  39.         cin >> fr >> to >> value;
  40.         fr--;to--;
  41.         g[fr][to] = value;
  42.         g[to][fr] = value;
  43.     }
  44.     prim(n);
  45.     int sum = 0;
  46.     for (auto &i : min_e) {
  47.         sum += i;
  48.     }
  49.     cout << sum;
  50. }
  51.  
  52.  
RAW Paste Data