CSES - Datatähti 2025 alku - Results
Submission details
Task:Kortit II
Sender:paju
Submission time:2024-11-08 00:19:38 +0200
Language:C++ (C++17)
Status:READY
Result:8
Feedback
groupverdictscore
#1ACCEPTED3
#2ACCEPTED5
#30
#40
#50
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1, 2, 3, 4, 5details
#2ACCEPTED0.02 s2, 3, 4, 5details
#3--3, 4, 5details
#4--4, 5details
#5--5details
#6--5details

Code

#include <iostream>
#include <string>
#include <list>
#include <vector>
#include <algorithm>
#include <numeric>
using namespace std;


int factorial(int n) 
{
    int out = 1;
    for (int i = 1; i < n+1; i++)
    {
        out *= i;
    }
    return out;
}

int ncr(int a, int b)
{
    return factorial(a)/(factorial(a-b)*factorial(b));
}

int main()
{
    int modulo = 1000000007;
    int gameamout;
    cin >> gameamout;
    list<vector<int>> allgamedata;
    int kortit = 0;
    int pisteetA = 0;
    int pisteetB = 0;
    vector<int> tokenit;
    for (int i = 0; i < gameamout; i++)
    {
        int a, b, c;
        cin >> a >> b >> c;
        vector<int> input;
        input.push_back(a);
        input.push_back(b);
        input.push_back(c);
        allgamedata.push_back(input);
    }
    for (int i = 0; i < gameamout; i++)
    {
        tokenit = allgamedata.front();
        pisteetB = tokenit[2];
        pisteetA = tokenit[1];
        kortit = tokenit[0];
        int totalp = pisteetA + pisteetB;
        int kortitfactorial = factorial(kortit);

        if (totalp > kortit)
        {
            cout << 0 << endl;
        }
        else if ((pisteetA == 0 || pisteetB == 0) && (pisteetA > 0 || pisteetB > 0))
        {
            cout << 0 << endl;
        }
        else if (pisteetA == 0 && pisteetB == 0)
        {
            cout << factorial(kortit) << endl;
        }
        else
        {
            long long alapakanpermutaatiotINT = 0;
            long long tasapelitilanteet = ncr(kortit, kortit-totalp);
            if(pisteetA == 1 || pisteetB == 1)
            {
                //permutaatioita vain 1
                long long result = kortitfactorial * (alapakanpermutaatiotINT+1)*tasapelitilanteet;
                cout << result % modulo << endl;
            }
            else{
                //lasketaan alapakan permutaatiot
                vector<int> alapakkapermutaatiot(kortit);
                iota(alapakkapermutaatiot.begin(), alapakkapermutaatiot.end(),1);
                do{
                    int a = 0;
                    int b = 0;
                    for (int i = 0; i < kortit; i++)
                    {
                        if(alapakkapermutaatiot[i] > i+1)
                        {
                            b++;
                        }
                        else if(alapakkapermutaatiot[i] < i+1){
                            a++;
                        }
                    }
                    if(a == pisteetA && b == pisteetB)
                    {
                        //cout << alapakkapermutaatiot[0] << " " << alapakkapermutaatiot[1] << " "  << alapakkapermutaatiot[2] << " "  << alapakkapermutaatiot[3] << " " << endl; 
                        alapakanpermutaatiotINT++;
                    }
                    
                }while(next_permutation(alapakkapermutaatiot.begin(),alapakkapermutaatiot.end()));
                long long result = kortitfactorial * alapakanpermutaatiotINT;
                //cout << result << " "<<modulo << endl;
                cout << result % modulo << endl;
            }
        }
        allgamedata.pop_front();
    }


    return 0;
}

Test details

Test 1

Group: 1, 2, 3, 4, 5

Verdict: ACCEPTED

input
54
4 4 0
3 1 3
3 2 2
4 0 4
...

correct output
0
0
0
0
0
...

user output
0
0
0
0
0
...

Test 2

Group: 2, 3, 4, 5

Verdict: ACCEPTED

input
284
6 1 0
5 0 2
7 1 5
7 7 5
...

correct output
0
0
35280
0
36720
...

user output
0
0
35280
0
36720
...

Test 3

Group: 3, 4, 5

Verdict:

input
841
19 3 12
19 19 13
19 7 13
20 11 15
...

correct output
40291066
0
0
0
0
...

user output
(empty)

Test 4

Group: 4, 5

Verdict:

input
1000
15 12 6
7 1 6
44 4 26
6 6 5
...

correct output
0
5040
494558320
0
340694548
...

user output
(empty)

Test 5

Group: 5

Verdict:

input
1000
892 638 599
966 429 655
1353 576 1140
1403 381 910
...

correct output
0
0
0
249098285
0
...

user output
(empty)

Test 6

Group: 5

Verdict:

input
1000
2000 1107 508
2000 1372 249
2000 588 65
2000 1739 78
...

correct output
750840601
678722180
744501884
159164549
868115056
...

user output
(empty)