Kortit II

input/code.cpp: In function 'int main()':
input/code.cpp:124:17: warning: unused variable 'nytMahdollisa' [-Wunused-variable]
  124 |             int nytMahdollisa = 0;
      |                 ^~~~~~~~~~~~~

#include <bits/stdc++.h>
using namespace std;

int mod = 1e9 + 7;

long long binomikerroin(int n, int k)
{
    if (k > (n >> 1))
        return binomikerroin(n, n - k);

    long long summa = 1;
    while (k > 0)
    {
        summa = (summa * n) % mod;
        n--;
        k--;
    }

    return summa;
}

set<vector<int>> luoLista(vector<int> perusta, int n, int lisayksia, int eka)
{
    set<vector<int>> listat;

    if (eka == 0 && lisayksia == 1)
    {
        vector<int> v = vector<int>(perusta);
        v[0] = 1;
        v[2 * n - 1] = 1;
        listat.insert(v);
        return listat;
    }

    for (int i = eka; i < (eka == 0 ? 1 : n - lisayksia + 1); i++)
    {
        for (int j = i + 1; j < n; j++)
        {
            if (perusta[n + j] == 1)
                continue;

            vector<int> alku = vector<int>(perusta);
            alku[i] = 1;
            alku[n + j] = 1;

            if (lisayksia == 1)
            {
                listat.insert(alku);
                continue;
            }

            for (vector<int> v : luoLista(alku, n, lisayksia - 1, i + 1))
            {
                if (i > 0 || (v[2 * n - 1] == 1))
                    listat.insert(v);
            }
        }
    }

    return listat;
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int t, n, a, b, c;
    cin >> t;

    int kertomat[2001];
    kertomat[1] = 1;

    for (long long i = 2; i <= 2000; i++)
    {
        kertomat[i] = (kertomat[i - 1] * i) % mod;
    }

    for (int i = 0; i < t; i++)
    {
        cin >> n >> a >> b;
        int voittoja = a + b;

        if (voittoja < 0 || voittoja > n || (a == 0) ^ (b == 0))
        {
            cout << "0\n";
            continue;
        }

        if (voittoja == 0)
        {
            cout << kertomat[n] << "\n";
            continue;
        }

        int tasapeleja = binomikerroin(n, n - voittoja);

        if (a > b)
        {
            c = b;
            b = a;
            a = c;
        }

        vector<int> alku;
        for (int j = -voittoja; j < voittoja; j++)
        {
            alku.push_back(0);
        }

        set<vector<int>> luvut = luoLista(alku, voittoja, a, 0);

        long long mahdollisia = 0LL;

        for (vector<int> v : luvut)
        {
            int voitotMaara = 1;
            int haviotMaara = 1;

            int voitotAlku = 0;
            int haviotAlku = 0;

            bool mahdollinen = true;
            int nytMahdollisa = 0;

            for (int j = 0; j < voittoja; j++)
            {

                if (v[voittoja + j] == 1)
                {
                    if (voitotAlku <= 0)
                    {
                        mahdollinen = false;
                        break;
                    }
                    voitotMaara *= voitotAlku;
                    voitotAlku--;
                }

                if (v[j] == 1)
                {
                    voitotAlku++;
                }
                else
                {
                    if (haviotAlku <= 0)
                    {
                        mahdollinen = false;
                        break;
                    }
                    haviotMaara *= haviotAlku;
                    haviotAlku--;
                }

                if (v[voittoja + j] != 1)
                {
                    haviotAlku++;
                }
            }
            if (mahdollinen)
            {
                /*for (int k : v)
                {
                    cout << k << " ";
                }
                cout << "\nMahdollisia: "
                     << (haviotMaara - b + 1) * (voitotMaara - a + 1) << "\n";*/
                mahdollisia += (haviotMaara) * (voitotMaara);
            }
        }

        cout << (mahdollisia * tasapeleja * kertomat[n]) % mod << "\n";
    }
}

Test 1

Group: 1, 2, 3, 4, 5

Verdict: WRONG ANSWER

Test 2

Group: 2, 3, 4, 5

Verdict: WRONG ANSWER

Test 3

Group: 3, 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 4

Group: 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 5

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 5

Verdict: TIME LIMIT EXCEEDED