Cod sursa(job #1746204)

Utilizator akaprosAna Kapros akapros Data 22 august 2016 20:22:46
Problema Nivele Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.1 kb
#include <bits/stdc++.h>
#define maxN 50002
#define maxL 18
using namespace std;
int n, t, a[maxN], p, maxe, pow2[maxL];
void read()
{
    int i;
    scanf("%d", &n);
    for (i = 1; i <= n; ++ i)
        scanf("%d", &a[i]);
}
bool f(int x, int y)
{
    int z = 0;
    while (a[p] == y && p <= n)
    {
        ++ p;
        ++ z;
    }
    if (x < z)
        return 0;
    if (x == z)
        return (p == n + 1);
    if (a[p] - y > maxe || (x - z) * pow2[a[p] - y] > n - p + 1)
        return 0;
    return f((x - z) * pow2[a[p] - y], a[p]);
}
void get_pow2()
{
    int i;
    pow2[0] = 1;
    for (i = 1; (1 << i) < maxN; ++ i)
        pow2[i] = 1 << i;
    maxe = i - 1;
}
void solve()
{
    sort(a + 1, a + n + 1);
    p = 1;
}
void write()
{
    bool ans = f(1, 1);
    if (ans)
        printf("DA\n");
    else
        printf("NU\n");
}
int main()
{
    freopen("nivele.in", "r", stdin);
    freopen("nivele.out", "w", stdout);
    get_pow2();
    scanf("%d", &t);
    while (t --)
    {
        read();
        solve();
        write();
    }
    return 0;
}