Cod sursa(job #3357152)

Utilizator parus_majorParus Major parus_major Data 6 iunie 2026 17:48:38
Problema Barman Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.71 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>
#include <cmath>

using namespace std;

struct Pahar {
    int valoare;
    int id_initial;
};

// Functie ajutatoare pentru a compara paharele dupa valoare
bool comparaPahare(const Pahar& a, const Pahar& b) {
    if (a.valoare == b.valoare)
        return a.id_initial < b.id_initial; // Stabilitate
    return a.valoare < b.valoare;
}

int main() {
    ifstream fin("barman.in");
    ofstream fout("barman.out");

    int N;
    if (!(fin >> N)) return 0;

    vector<Pahar> v(N);
    vector<Pahar> sortat(N);

    for (int i = 0; i < N; ++i) {
        fin >> v[i].valoare;
        v[i].id_initial = i;
        sortat[i] = v[i];
    }

    // Sortam paharele pentru a sti ordinea lor ideala
    sort(sortat.begin(), sortat.end(), comparaPahare);

    long long min_timp_total = -1;

    // Incercam toate cele N deplasari circulare ale solutiei
    for (int start = 0; start < N; ++start) {
        // dest[i] va retine unde trebuie sa ajunga paharul aflat initial pe pozitia i
        vector<int> dest(N);

        for (int i = 0; i < N; ++i) {
            // Pozitia in sirul sortat circular care incepe de la 'start'
            int poz_tinta = (start + i) % N;
            int id_orig = sortat[i].id_initial;
            dest[id_orig] = poz_tinta;
        }

        // Determinam ciclurile si calculam costul pentru configuratia curenta
        vector<bool> vizitat(N, false);
        long long timp_curent = 0;

        for (int i = 0; i < N; ++i) {
            if (!vizitat[i]) {
                int nod = i;
                long long lungime_ciclu = 0;
                long long cost_deplasari = 0;

                // Parcurgem ciclul
                while (!vizitat[nod]) {
                    vizitat[nod] = true;
                    int urmatorul = dest[nod];

                    if (nod != urmatorul) {
                        lungime_ciclu++;
                        cost_deplasari += abs(nod - urmatorul);
                    }

                    nod = urmatorul;
                }

                // Daca ciclul are elemente de mutat
                if (lungime_ciclu > 0) {
                    // Fiecare pahar din ciclu este ridicat si pus (20 secunde)
                    // plus costul deplasarilor efective pe axa
                    timp_curent += (20 * lungime_ciclu) + cost_deplasari;
                }
            }
        }

        // Retinem minimul global
        if (min_timp_total == -1 || timp_curent < min_timp_total) {
            min_timp_total = timp_curent;
        }
    }

    fout << min_timp_total << "\n";

    fin.close();
    fout.close();
    return 0;
}