Cod sursa(job #2562187)

Utilizator Dobricean_IoanDobricean Ionut Dobricean_Ioan Data 29 februarie 2020 12:41:17
Problema Congr Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.67 kb
#include <fstream>
#include <stdlib.h>
using namespace std;

ifstream fin("congr.in");
ofstream fout ("congr.out");

const long long MAXN = 600001;
pair <long long,long long> a[MAXN];
long long n,p,sum;

int main() {

    fin >> p;
    n = 2*p-1;
    for(long long i = 1; i <= n; ++i) {
        fin >> a[i].first,a[i].second = i;
        a[i].first %= p;
        if(i <= p)
            sum += a[i].first;
    }
    srand(unsigned(time(0)));
    while( sum % p) {

        int p1 = rand() % p + 1;
        int p2 = rand() % p + p+1;
        sum += -a[p1].first + a[p2].second;
        swap(a[p1],a[p2]);
    }
    for ( int i = 1; i <= p; ++i,fout << " ")
    fout << a[i].second << " ";

}