Cod sursa(job #2709076)

Utilizator PetyAlexandru Peticaru Pety Data 19 februarie 2021 18:23:51
Problema Sate2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva ICPC Marime 1.43 kb
#include <bits/stdc++.h>

using namespace std;

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

const int bulaneala = 10000000;
int t, n, m, k, a[3002];
vector<int>G[4];
int sum[4];

void reset () {
  for (int i = 0; i < 4; i++) {
    G[i].clear();
    sum[i] = 0;
  }
}

int main()
{
    srand(time(NULL));
  fin >> t;
  while (t--) {
    fin >> n >> m >> k;
    int mx = 0;
    for (int i = 1; i <= n; i++) {
      fin >> a[i];
      mx = max(mx, a[i]);
      int x = rand() % k;
      G[x].push_back(a[i]);
      sum[x] += a[i];
    }
    if (m % k != 0 || mx > m / k) {
      fout << "NU\n";
      reset();
      continue;
    }
    bool ok = 0;
    for (int soarta = 1; soarta <= bulaneala; soarta++) {
      int mx = 0, poz1 = 0;
      for (int j = 0; j < k; j++)
        if (sum[j] > mx) {
          poz1 = j;
          mx = sum[j];
        }
      int mn = 1e9, poz2 = 0;
      for (int j = 0; j < k; j++)
        if (sum[j] < mn) {
          poz2 = j;
          mn = sum[j];
        }
      if (mn == m / k) {
        fout << "DA\n";
        ok = 1;
        break;
      }
      int lol = rand() % G[poz1].size();
      sum[poz1] -= G[poz1][lol];
      sum[poz2] += G[poz1][lol];
      G[poz2].push_back(G[poz1][lol]);
      swap(G[poz1][lol], G[poz1][G[poz1].size() - 1]);
      G[poz1].pop_back();
    }
    reset();
    if (!ok)
      fout << "NU\n";
  }
  return 0;
}