Cod sursa(job #2554936)

Utilizator marius004scarlat marius marius004 Data 23 februarie 2020 15:24:01
Problema Balanta Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.82 kb
#include <fstream>

std::ifstream f("balanta.in");
std::ofstream g("balanta.out");

const int NMAX = 1025;
int n,m,k,x,r,stanga[NMAX],dreapta[NMAX],usor[NMAX],greu[NMAX];

int main(){

    f >> n >> m;

    /// Consideram doi vectori usor si greu.
    /// usor[i] = 1 daca elementul i nu poate sa fie mai usor decat celelate elemente
    /// greu[i] = 1 daca elementul i nu poate sa fie mai greu dacat celelate elemente

    /// Query:
    /// 0) parte stanga este egala cu partea dreapta.insemana ca toate elementele din partea stanga si partea dreapta nu pot sa fie mai usoare,nici mai grele
    /// 1) partea stanga este mai grea decat partea dreapta.inseamna ca exista un element e care se afla in parte stanga care e mai greu sau exista un element
    ///     in parte dreapta care e mai usor decat toate elementele din partea stanga.
    /// primul caz:exista un element mai greu in stanga => elementele din stanga nu au cum sa fie usoare
    /// al doilea caz : exista un element mai usor in dreapta => elementele din dreapta nu au cum sa fie mai grele
    /// 2) analog cu 1) doar ca e invers

    /// La afisare:
    /// daca exista numai un element in usor care are valoarea 0 atunci afisem indexul elementului
    /// daca exista numai un element in greu care are valoarea 0 atunci afisem indexul elementului
    /// astfel afisem -1. Nu am reusit sa ne dam seama de elementul mai greu sau mai usor

    while(m--){

        f >> k;

        for(int i = 1;i <= n;++i)
            dreapta[i] = stanga[i] = 0;

        for(int i = 1;i <= k;++i){
            f >> x;
            stanga[x] = 1;
        }

        for(int i = 1;i <= k;++i){
            f >> x;
            dreapta[x] = 1;
        }

        f >> r;

        if(r == 0){

            for(int i = 1;i <= n;++i)
                if(stanga[i] || dreapta[i])
                    usor[i] = greu[i] = 1;

        }else if(r == 1){// stanga > dreapta

            for(int i = 1;i <= n;++i){
                if(stanga[i])usor[i] = 1;
                else if(dreapta[i])greu[i] = 1;
                else usor[i] = greu[i] = 1;
            }
        }else{// stanga < dreapta

            for(int i = 1;i <= n;++i){
                if(stanga[i])greu[i] = 1;
                else if(dreapta[i])usor[i] = 1;
                else greu[i] = usor[i] = 1;
            }
        }
    }

    int cnt = 0;
    int pos = 0;

    for(int i = 1;i <= n && cnt <= 1;++i)
        if(usor[i] == 0){
            cnt++;
            pos = i;
        }

    if(cnt == 1){
        g << pos;
        return 0;
    }

    cnt = pos = 0;

    for(int i = 1;i <= n && cnt <= 1;++i)
        if(greu[i] == 0){
            cnt++;
            pos = i;
        }

    if(cnt == 1){
        g << pos;
        return 0;
    }

    g << 0;

    return 0;
}