Kortit II

input/code.cpp: In function 'void haku(int, int, int)':
input/code.cpp:44:18: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
   44 |     if (v.size() == n)
      |         ~~~~~~~~~^~~~
input/code.cpp:94:13: warning: label 'EHDOT' defined but not used [-Wunused-label]
   94 |             EHDOT :
      |             ^~~~~

#include <iostream>
#include <cmath>
#include <vector>
#include <algorithm>
#include <string>
#include <sstream>

typedef long long ll;
using namespace std;

ll n, a, b;

ll vastaus;

ll kertoma(int x){
    for(int i = x-1; i>0; i--){
        x *= i;
    }
 
    return x;
}

int binomialC(int n, int k)
{
    int ans = 1;
    if (k > n - k)
        k = n - k;

    for (int i = 0; i < k; i++)
    {
        ans = ans * (n - i);
        ans = ans / (i + 1);
    }
    return ans;
}

vector<int> v;
ll p[1000001];

ll iterations;

void haku(int level, int winsA, int winsB)
{
    if (v.size() == n)
    {

        if (v.size() > 0 && level < v[level-1]){ // JOS B VOITTAA, MERKKAA YLÖS
            winsB++;
        }
        if (v.size() > 0 && level > v[level-1]){ // JOS A VOITTAA, MERKKAA YLÖS
            winsA++;
        } 

        ll aTemp = 0;
        ll bTemp = 0;

       for(int i = 0; i<n; i++){
            if (v[i] > i+1){
                aTemp++;
            }
            else if (v[i] < i+1){
                bTemp++;
            }
       }

        // if (aTemp == a && bTemp == b){
        //     vastaus++;
        // }

        if (winsA == a && winsB == b){
            vastaus++;
        }



    }
    else
    {
        for (int i = 1; i <= n; i++)
        {

            iterations++;

            if (p[i])
                continue;

            if (v.size() > 0 && level < v[level-1]){ // JOS B VOITTAA, MERKKAA YLÖS
                winsB++;
            }
            if (v.size() > 0 && level > v[level-1]){ // JOS A VOITTAA, MERKKAA YLÖS
                winsA++;
            }

            EHDOT :
            if (v.size() > 0 && level == v[level-1]) {  // JOS TASAPELI, CONTINUE

                // if (v.size() == 0){
                //     cout << "Level: " << level << " i: " << "0" << "\n";
                // }
                // else{
                //     cout << "Level: " << level << " v: " << v[level-1] << "\n";
                // }
                

                // for (int i2 = 0; i2 < v.size(); i2++)
                // {
                //     cout << i2+1 << " ";
                // }
                // cout << "\n";
                // for (int i2 = 0; i2 < v.size(); i2++)
                // {
                //     cout << v[i2] << " ";
                // }
                // cout << "\n";

                continue;
            }
            if (winsB > b){ // JOS B:LLÄ ON ENEMMÄN VOITTOJA KUIN SAISI, CONTINUE


                // cout << "winsB: "<< winsB << " b: " << b << " Iteration: " << iterations << " Current: " << vastaus << "\n"; 

                // for (int i2 = 0; i2 < v.size(); i2++)
                // {
                //     cout << i2+1 << " ";
                // }
                // cout << "\n";
                // for (int i2 = 0; i2 < v.size(); i2++)
                // {
                //     cout << v[i2] << " ";
                // }
                // cout << "\n \n";
                
                continue;
            }

            if (winsA > a){ // JOS A:LLA ON ENEMMÄN VOITTOJA KUIN SAISI, CONTINUE


                // cout << "winsB: "<< winsB << " b: " << b << " Iteration: " << iterations << " Current: " << vastaus << "\n"; 

                // for (int i2 = 0; i2 < v.size(); i2++)
                // {
                //     cout << i2+1 << " ";
                // }
                // cout << "\n";
                // for (int i2 = 0; i2 < v.size(); i2++)
                // {
                //     cout << v[i2] << " ";
                // }
                // cout << "\n \n";
                
                continue;
            }


            

            p[i] = 1;
            v.push_back(i);
            haku(level + 1, winsA, winsB);



            p[i] = 0;
            v.pop_back();

            if (v.size() > 0 && level < v[level - 1]) // JOS B VOITTI, OTA POIS 
            {
                winsB--;
            }
            if (v.size() > 0 && level > v[level - 1]) // JOS A VOITTI, OTA POIS 
            {
                winsA--;
            }
        }
    }
}

int main() {

    int t;

    cin >> t;

    vector<int> nA;
    vector<int> aA;
    vector<int> bA;

    for (int i = 0; i<t; i++){
        cin >> n >> a >> b;
 
        nA.push_back(n);
        aA.push_back(a);
        bA.push_back(b);

    }


    for (int i = 0; i<t; i++){

        n = nA[i];
        a = aA[i];
        b = bA[i];


        if (a != 0 && b != 0){
            if (a+b > n){
                cout << "0" << "\n";
                continue;
            }
        }
        else if (a+b != 0){
            cout << "0" << "\n";
            continue;
        }





        vastaus = 0;


        // if (a == 0 && b == 0){
        //     vastaus = kertoma(n);
        //     cout << vastaus << "\n";
        // }
        // else if (a == 1 || b == 1){
        //     vastaus = binomialC(n, n-a-b) * kertoma(n);
        //     cout << vastaus << "\n";
        // }


        ll Tasapelivaihtoehdot = binomialC(n, n-a-b);

        // POISTAA TASAPELIT
        n = a+b;

        haku(0, 0, 0);

        n = nA[i];

        vastaus *= Tasapelivaihtoehdot;
        // cout << " " << a << " " << b << " " << n << "\n"; 
        // cout << vastaus*kertoma(n) << "\n ITERATIONS: " << iterations << "\n";

        for(int i = 1; i<=n; i++){ // VASTAUS * KERTOMA (n)
            vastaus = (vastaus * i) % 1000000007;
        }

        cout << vastaus << "\n";



    }


}
 

 

Test 1

Group: 1, 2, 3, 4, 5

Verdict: ACCEPTED

Test 2

Group: 2, 3, 4, 5

Verdict: ACCEPTED

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