Cod sursa(job #1708660)

Utilizator iulianrotaruRotaru Gheorghe-Iulian iulianrotaru Data 27 mai 2016 19:09:05
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.89 kb
#include <fstream>
#include <algorithm>
using namespace std;
class InputReader {
    public:
        InputReader() {}
        InputReader(const char *file_name) {
            input_file = fopen(file_name, "r");
            cursor = 0;
            fread(buffer, SIZE, 1, input_file);
        }
        inline InputReader &operator >>(int &n) {
            while((buffer[cursor] < '0' || buffer[cursor] > '9')&&buffer[cursor]!='-') {
                advance();
            }
            n =0;
            semn= 1;
            if(buffer[cursor]=='-') semn=-1,advance();
            while('0' <= buffer[cursor] && buffer[cursor] <= '9') {
                n = n * 10 + buffer[cursor] - '0';
                advance();
            }
            n*=semn;
            return *this;
        }
    private:
        FILE *input_file;
        static const int SIZE = 1 << 17;
        int cursor,semn;
        char buffer[SIZE];
        inline void advance() {
            ++ cursor;
            if(cursor == SIZE) {
                cursor = 0;
                fread(buffer, SIZE, 1, input_file);
            }
        }
}f("apm.in");
ofstream g("apm.out");
struct el{int x,y,c;}v[1<<18];
int i,n,m,t,S,sol[1<<18],GR[1<<18];
struct cmp
{
    bool operator () (el a,el b)
    {
        return a.c<b.c;
    }
};
inline int gr(int i)
{
    if(GR[i]==i) return i;
    GR[i]=gr(GR[i]);
    return GR[i];
}
inline void unite(int i,int j)
{
    GR[gr(i)]=gr(j);
}
int main ()
{
    f>>n>>m;
    for(i=1;i<=m;++i)
        f>>v[i].x>>v[i].y>>v[i].c;
    sort(v+1,v+m+1,cmp());
    for(i=1;i<=n;++i) GR[i]=i;
    for(i=1;i<=m;++i)
        if(gr(v[i].x)!=gr(v[i].y))
        {
            S+=v[i].c;
            unite(v[i].x,v[i].y);
            sol[++t]=i;
        }
    g<<S<<'\n'<<n-1<<'\n';
    for(i=1;i<n;++i)
        g<<v[sol[i]].x<<' '<<v[sol[i]].y<<'\n';
    return 0;
}