#include <bits/stdc++.h>
using namespace std;
const long long k_Mod = 1e9 + 7;
const int k_N = 100 + 3;
int n;
pair<long long, bool> dp[3 * k_N][3 * k_N][3 * k_N][3];
long long get_answer(int ab, int ac, int bc, int stage){
if(ab + ac + bc == 0)
return 1;
cout << ab << " " << ac << " " << bc << " " << stage << endl;
auto &[answer, solved] = dp[ab][ac][bc][stage];
if(solved)
return answer;
solved = true;
answer = 0;
if(stage == 0){
if(ab + ac == 0)
return answer = get_answer(ab, ac, bc, 1);
if(ab) answer += get_answer(ab - 1, ac, bc, 1) * ab % k_Mod;
if(ac) answer += get_answer(ab, ac - 1, bc + 1, 1) * ac % k_Mod;
return answer %= k_Mod;
}
if(stage == 1){
if(ab + bc == 0)
return answer = get_answer(ab, ac, bc, 2);
if(ab) answer += get_answer(ab - 1, ac + 1, bc, 2) * ab % k_Mod;
if(bc) answer += get_answer(ab, ac, bc - 1, 2) * bc % k_Mod;
return answer %= k_Mod;
}
if(stage == 2){
if(ac + bc == 0)
return answer = get_answer(ab, ac, bc, 0);
if(ac) answer += get_answer(ab, ac - 1, bc, 0) * ac % k_Mod;
if(bc) answer += get_answer(ab + 1, ac, bc - 1, 0) * bc % k_Mod;
return answer %= k_Mod;
}
assert(false);
}
void solve_test(){
static pair<int, int> pos[3 * k_N];
for(int i = 1; i <= 3 * n; ++i)
pos[i] = {-1, -1};
int x;
for(int i = 0; i < 2 * n; ++i){
cin >> x;
if(pos[x].first == -1)
pos[x].first = 0;
else
pos[x].second = 0;
}
for(int i = 0; i < 2 * n; ++i){
cin >> x;
if(pos[x].first == -1)
pos[x].first = 1;
else
pos[x].second = 1;
}
for(int i = 0; i < 2 * n; ++i){
cin >> x;
if(pos[x].first == -1)
pos[x].first = 2;
else
pos[x].second = 2;
}
int ab = 0, ac = 0, bc = 0;
for(int i = 1; i <= 3 * n; ++i){
if(pos[i].first == pos[i].second)
continue;
//if(pos[i].first > pos[i].second)
// swap(pos[i].first, pos[i].second);
if(pos[i].second == 1)
ab++;
else if(pos[i].first == 0)
ac++;
else
bc++;
}
//cout << ab << " " << ac << " " << bc << endl;
cout << get_answer(ab, ac, bc, 0) << "\n";
}
int main(){
ios::sync_with_stdio(false);
cin.tie(NULL);
int t;
cin >> n >> t;
while(t--)
solve_test();
}
Compilation message
fishing.cpp: In function 'long long int get_answer(int, int, int, int)':
fishing.cpp:17:11: warning: decomposition declaration only available with -std=c++1z or -std=gnu++1z
auto &[answer, solved] = dp[ab][ac][bc][stage];
^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
5 ms |
384 KB |
Output isn't correct |
2 |
Incorrect |
5 ms |
512 KB |
Output isn't correct |
3 |
Incorrect |
13 ms |
1536 KB |
Output isn't correct |
4 |
Incorrect |
69 ms |
5112 KB |
Output isn't correct |
5 |
Incorrect |
899 ms |
37764 KB |
Output isn't correct |
6 |
Incorrect |
1489 ms |
58516 KB |
Output isn't correct |
7 |
Execution timed out |
2091 ms |
76152 KB |
Time limit exceeded |
8 |
Execution timed out |
2098 ms |
75020 KB |
Time limit exceeded |
9 |
Execution timed out |
2097 ms |
76904 KB |
Time limit exceeded |
10 |
Execution timed out |
2096 ms |
76652 KB |
Time limit exceeded |