Kortit II

input/code.cpp: In function 'void haku(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)
      |         ~~~~~~~~~^~~~

#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)
{
    if (v.size() == n)
    {

        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++;
        }

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

            iterations++;

            if (p[i])
                continue;

            // EHDOT :
            // if (level == i) {
            //     cout << "Level: " << level << ", i: " << i << "\n";

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

            p[i] = 1;
            v.push_back(i);
            haku(level + 1);
            p[i] = 0;
            v.pop_back();
        }
    }
}

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(1);

        n = nA[i];

        vastaus *= Tasapelivaihtoehdot;
        // cout << " " << a << " " << b << " " << n << "\n"; 
        cout << vastaus*kertoma(n) << "\n" ;
        // << iterations << "\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