Cod sursa(job #2055062)

Utilizator BogdanisarBurcea Bogdan Madalin Bogdanisar Data 2 noiembrie 2017 19:56:00
Problema Sortare Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.25 kb
#include <iostream>
#include <fstream>
#include <algorithm>

#if 1
#define pv(x) cout<<#x<<" = "<<x<<"; "
#define pn cout<<"\n"
#else
#define pv(x)
#define pn
#endif

using namespace std;
ifstream in("sortare.in");
ofstream out("sortare.out");

using ll = long long;
using ull = unsigned long long;
using ui = unsigned int;
#define pb push_back
#define mp make_pair
const int NMax = 5e3 + 5;
const int inf = 1e9 + 5;
using zint = int;

int N;
int v[NMax],param[NMax][3];
// in v se genereaza permutarea;
// param[i] - retine cele trei indexuri in ordine crescatoare;

void moveRight(int,int);
int solve(int);

int main() {
    cin.sync_with_stdio(false);
    cin.tie(0);

    in>>N;
    for (int i=2;i <= N;++i) {
        in>>param[i][0]>>param[i][1]>>param[i][2];
        sort(param[i],param[i]+3);
    }

    int height = solve(N);

    out<<height<<'\n';
    for (int i=1;i <= N;++i) {
        out<<v[i]<<' ';
    }

    return 0;
}

// solve(nr):
// genereaza o permutare de lungime nr (si care contine primele nr numere)
// care duce la numar maxim de recursii;
int solve(int nr) {
    if (nr == 1) {
        v[1] = 1;
        return 1;
    }

    int idx;
    // daca avem doua indexuri identice, atunci
    // putem pune cel mai mare element disponibil
    // pe una din pozitiile care se repeta
    // astfel incat partionarea sa fie in jurul lui (worst-case);
    if (param[nr][0] == param[nr][1] || param[nr][1] == param[nr][2]) {
        int ret = solve(nr-1) + 1;

        idx = param[nr][1];
        moveRight(idx,nr);
        v[idx] = nr;

        return ret;
    }

    // daca avem toate indexurile diferite, atunci
    // suntem nevoiti sa punem cele mai mari doua elemente
    // pe doua dintre cele trei pozitii
    // si se obtine o partitionare dupa
    // al doilea cel mai mare element (aproape worst-case);
    int ret = solve(nr-2) + 1;

    idx = param[nr][1];
    moveRight(idx,nr);
    v[idx] = nr-1;

    idx = param[nr][2];
    moveRight(idx,nr);
    v[idx] = nr;

    return ret;
}

// moveRight(st,dr) - muta toate elementele de pe pozitiile [st,dr-1] cu o pozitie la dreapta;
void moveRight(int st,int dr) {

    for (int i=dr;i >= st;--i) {
        v[i] = v[i-1];
    }
}