Pagini recente » Cod sursa (job #302258) | Cod sursa (job #111040) | Cod sursa (job #1601890) | Cod sursa (job #2479069) | Cod sursa (job #2709080)
#include <bits/stdc++.h>
using namespace std;
ifstream fin ("sate2.in");
ofstream fout ("sate2.out");
const int bulaneala = 20000000;
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;
int ans = 1e9;
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];
}
ans = min(mx - mn, ans);
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();
}
if (ans <= 100) {
cout << "DA\n";
continue;
}
reset();
if (!ok)
fout << "NU\n";
}
return 0;
}