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

Code

#include <bits/stdc++.h>

using namespace std;


int main() {
    long long n, t, c, a, b, kortit;
    cin >> t;

    for (long long index = 0LL; index < t; index++) {
        cin >> kortit >> a >> b;
        long long tasat = kortit - a - b;
        vector<long long> v;

        if (tasat < 0)cout << 0 << endl; 
        else {
            n = kortit - tasat;
            //vastaus *= tasat
            long long chos = 1LL, fac = 1LL, factasat = 1LL, facminus = 1LL;
            c = 0LL; 
            if ((a != 0 and b != 0) or n == 0) {
                for (long long index = 1LL; index <= kortit; index++) {
                    fac *= index;
                    fac %= 1000000007LL;
                }
                for (long long index = 1LL; index <= tasat; index++) {
                    factasat *= index;
                    factasat %= 1000000007LL;
                }
                for (long long index = 1LL; index <= kortit - tasat; index++) {
                    facminus *= index;
                    facminus %= 1000000007LL;
                }
                if (a == 0 and b == 0) c = 1;
                else {
                    //c
                    vector<vector<long long>> previous = {{1LL}, {2LL,3LL,2LL}, {1LL}}, now;

                    for (long long index = 4LL; index < n; index++) {
                        now.push_back({1});
                        now.push_back({2});
                        long long doubles = 4LL;
                        for (long long d = 1LL; d <= index-3LL; d++) {
                            now[1].push_back(doubles);
                            doubles = doubles*2 % 1000000007LL;
                        }
                        now[1].push_back(doubles - 1);
                        now[1].push_back(accumulate(previous[1].begin(), previous[1].end(), 1) % 1000000007LL);

                        for (long long i = 1LL; i < (index-1LL)/2LL; i++) {
                            long long uusi = (previous[i].back() - previous[i][0] + now[i].back()) % 1000000007LL;
                            now.push_back({uusi});
                            for (long long a = 0LL; a < index-1LL; a++) {
                                now[i+1].push_back((-previous[i][a] + previous[i+1][a] + now.back().back()) % 1000000007LL);
                            }
                        }
                        if (index % 2 == 0) {
                            v.clear();
                            for (long long t = index-1LL; t >= 0; t--) {
                                v.push_back(now[now.size()-1][t] % 1000000007LL);
                            }
                            now.push_back(v);
                        }

                        previous = now;
                        now.clear();
                    }

                    c = accumulate(previous[min(a, b)-1].begin(), previous[min(a, b)-1].end(), 0);
                }

                chos = fac/(factasat*facminus) % 1000000007LL;
                cout << c*chos*fac % 1000000007LL << endl; 
            }
            else cout << 0 << endl; 
        }
    }
}

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
0
0
0
0
0
...

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
0
5040
0
0
0
...

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)