Cod sursa(job #2761240)

Utilizator VanillaSoltan Marian Vanilla Data 1 iulie 2021 11:35:17
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.69 kb
#include <bits/stdc++.h>
#include <fstream>
using namespace std;
typedef long long int64;
typedef vector<int> vec;
typedef vector<int64> vec64;
string __fname = "apm";
ifstream in (__fname + ".in");
ofstream out (__fname + ".out");
#define cin in
#define cout out
#define pii pair <int, int> 
#define pii64 pair <int64, int64>
#define ss cout << " ";
#define nn cout << "\n";
#define ct(x) cout << x;
#define cts(x) cout << x << " ";
#define ctn(x) cout << x << "\n";
#define db(x) cout << "> " << #x << ": " << x << "\n";
#define qr queries();
void solve(int);
void queries(){int n;cin >> n;for (int i = 1; i <= n; i++) solve(i);}
int64 ceildiv(int64 a, int64 b) {return a / b + !!(a % b);}
template<class T>T gcd (T a, T b){return (b ? gcd(b, a % b): a);}
template<class T>T lcm (T a, T b){return a * b / gcd(a, b);}
// // // // // // // // // // // // // // // // // // // // // // 
/*                  TEMPLATE - VANILLA                         */
// // // // // // // // // // // // // // // // // // // // // //
const int maxn = 200200;
const int64 mod = 1000000007;
const double pi = 3.14159265359;
const int ddx[] = {-1, -1, 0, 1, 1, 1, 0, -1};
const int ddy[] = {0, 1, 1, 1, 0, -1, -1, -1};
const int dx[] = {-1, 0, 1, 0};
const int dy[] = {0, 1, 0, -1};

struct node {
    int x,y,val;
};


void solve(int id){
    
    return;
}

template <class T> class dsu {
    private:
        vector <T> parent;

    public:
        dsu(int size) {
            parent.resize(size);
        }
        
        void make_set(T val) {
            parent[val] = val;
        }

        void union_sets(T a, T b) {
            a = find_set(a);
            b = find_set(b);
            if (a != b) {
                parent[b] = a;
            }
        }

        T find_set (T node) {
            if (node == parent[node]) return node;
            return parent[node] = find_set(parent[node]);
        }
};

bool comp (node a, node b) {
    return a.val < b.val;
}


int main(){
    ios_base::sync_with_stdio(0);cin.tie(0);
    int n,m;
    cin >> n >> m;
    vector <node> a (m);
    for (int i = 0; i < m; i++){
        cin >> a[i].x >> a[i].y >> a[i].val;
    }
    sort(a.begin(), a.end(), comp);
    dsu <int> dsu (n + 1);
    vector <pii> mst;
    for (int i = 1; i <= n; i++){
        dsu.make_set(i);
    }
    int rs = 0;
    rs+=a[0].val;
    dsu.union_sets(a[0].x, a[0].y);
    mst.push_back({a[0].x, a[0].y});
    for (int i = 1; i < m; i++){
        if (dsu.find_set(a[i].x) != dsu.find_set(a[i].y)) {
            dsu.union_sets(a[i].x, a[i].y);
            rs+=a[i].val;
            mst.push_back({a[i].x, a[i].y});
        }
    }
    ctn(rs);
    ctn(mst.size());
    for (auto i: mst) {
        cts(i.first); ctn(i.second);
    }
    return 0;
}