| Task: | Kortit II |
| Sender: | manttila |
| Submission time: | 2024-10-31 13:38:27 +0200 |
| Language: | C++ (C++17) |
| Status: | READY |
| Result: | 100 |
| group | verdict | score |
|---|---|---|
| #1 | ACCEPTED | 3 |
| #2 | ACCEPTED | 5 |
| #3 | ACCEPTED | 26 |
| #4 | ACCEPTED | 28 |
| #5 | ACCEPTED | 38 |
| test | verdict | time | group | |
|---|---|---|---|---|
| #1 | ACCEPTED | 0.06 s | 1, 2, 3, 4, 5 | details |
| #2 | ACCEPTED | 0.06 s | 2, 3, 4, 5 | details |
| #3 | ACCEPTED | 0.06 s | 3, 4, 5 | details |
| #4 | ACCEPTED | 0.06 s | 4, 5 | details |
| #5 | ACCEPTED | 0.06 s | 5 | details |
| #6 | ACCEPTED | 0.06 s | 5 | details |
Code
#include<bits/stdc++.h>
using namespace std;
#define debug(x) cout << #x << " = " << x << "\n"
#define pb push_back
typedef long long ll;
typedef vector<ll> vi;
typedef pair<ll,ll> pi;
const ll M = 1000000007;
const ll INF = LLONG_MAX/2-1;
ll g[2004][2004];
void triangle(){
g[2][1] = 1;
for(int i=2; i<=2001; i++){
for(int r=0; r<=i; r++){
g[i+1][r] = ((r*g[i][r]%M + (i+1-r)*g[i][r-1]%M)%M + i*g[i-1][r-1]%M)%M;
}
}
}
vi fact(2002);
void factorials(){
int sol=1;
for(int i=2; i<2002; i++){
fact[i-1] = sol;
sol = ((sol%M)*i)%M;
}
}
vector<vi> comb;
vector<vi> calcom(int n) {
vector<vi> nCr(n + 1, vi(n + 1, 0));
for (int i = 0; i <= n; i++) {
nCr[i][0] = 1;
nCr[i][i] = 1;
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j < i; j++) {
nCr[i][j] = (nCr[i - 1][j - 1] + nCr[i - 1][j]) % M;
}
}
return nCr;
}
void solve(){
int n,a,b; cin >> n >> a >> b;
if(a+b>n || ((a==0)^(b==0))){
cout << 0 << "\n";
}else if(a==0 && b==0){
cout << fact[n] << "\n";
}else{
cout << ((comb[n][n-a-b]%M * g[a+b][a]%M)%M * fact[n]%M)%M << "\n";
}
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0);
int t=1; cin >> t;
triangle();
factorials();
comb = calcom(2002);
while(t--) solve();
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: ACCEPTED
| input |
|---|
| 841 19 3 12 19 19 13 19 7 13 20 11 15 ... |
| correct output |
|---|
| 40291066 0 0 0 0 ... |
| user output |
|---|
| 40291066 0 0 0 0 ... |
Test 4
Group: 4, 5
Verdict: ACCEPTED
| 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 494558320 0 340694548 ... |
Test 5
Group: 5
Verdict: ACCEPTED
| input |
|---|
| 1000 892 638 599 966 429 655 1353 576 1140 1403 381 910 ... |
| correct output |
|---|
| 0 0 0 249098285 0 ... |
| user output |
|---|
| 0 0 0 249098285 0 ... |
Test 6
Group: 5
Verdict: ACCEPTED
| input |
|---|
| 1000 2000 1107 508 2000 1372 249 2000 588 65 2000 1739 78 ... |
| correct output |
|---|
| 750840601 678722180 744501884 159164549 868115056 ... |
| user output |
|---|
| 750840601 678722180 744501884 159164549 868115056 ... |
