Cod sursa(job #1981072)

Utilizator BogdanisarBurcea Bogdan Madalin Bogdanisar Data 14 mai 2017 18:52:12
Problema Sortare prin comparare Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.54 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <cstdlib>
#include <ctime>

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

#define ll long long
#define pb push_back
const int NMax = 5e5 + 5;
const int inf = 1e9 + 5;

// algoritmul Quicksort cu schema Lomuto de partitie si pivot aleator

int N;
int v[NMax];

void quickSort(int,int);
int part(int,int);

int main() {
    in>>N;
    for (int i=1;i <= N;++i) {
        in>>v[i];
    }

    srand(time(0));
    quickSort(1,N);

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

    in.close();out.close();
    return 0;
}

// quickSort(st,dr) pune un element in pozitia corecta si
// se autoapeleaza pentru vectorul din stanga si din dreapta
void quickSort(int st,int dr) {
    if (st >= dr) {
        return;
    }

    int idx = part(st,dr);
    quickSort(st,idx-1);
    quickSort(idx+1,dr);
}

// part(st,dr) aduce un element in pozitia corecta si return-eaza aceasta pozitie
int part(int st,int dr) {

    // se alege un pivot random
    int idx = rand() % (dr-st+1) + st;
    swap(v[idx],v[dr]);

    // elementele de la st la i-1 sunt mai mici sau egale ca pivotul
    // elementele de la i la j-1 sunt mai mari ca pivotul
    int i = st;
    for (int j = st;j < dr;++j) {
        if (v[j] <= v[dr]) {
            swap(v[i],v[j]);
            ++i;
        }
    }

    // se pune pivotul intre cele mai mici sau egale si cele mai mari
    swap(v[i],v[dr]);
    return i;
}