oj.uz

Submission #238508

# Submission time Handle Problem Language Result Execution time Memory
238508 2020-06-11T15:18:14 Z Sorting Fishing Game (RMI19_fishing) C++17
0 / 100
 152 ms 92792 KB 
#include <bits/stdc++.h>

using namespace std;

const int k_Mod = 1e9 + 7;
const int k_N = 100 + 3;

int n;
pair<int, bool> dp[2 * k_N][2 * k_N][2 * k_N][3];


int 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) * ac % 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] = {0, 0};

    int x;
    for(int i = 0; i < 2 * n; ++i){
        cin >> x;
        if(!pos[x].first)
            pos[x].first = 0;
        else
            pos[x].second = 0;
    }
    for(int i = 0; i < 2 * n; ++i){
        cin >> x;
        if(!pos[x].first)
            pos[x].first = 1;
        else
            pos[x].second = 1;
    }
    for(int i = 0; i < 2 * n; ++i){
        cin >> x;
        if(!pos[x].first)
            pos[x].first = 2;
        else
            pos[x].first = 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();
}

Subtask #1
0 / 100.0

# Verdict Execution time Memory Grader output
1 Incorrect 5 ms 384 KB Output isn't correct
2 Incorrect 5 ms 512 KB Output isn't correct
3 Incorrect 7 ms 1408 KB Output isn't correct
4 Incorrect 7 ms 3968 KB Output isn't correct
5 Incorrect 28 ms 23544 KB Output isn't correct
6 Incorrect 41 ms 33912 KB Output isn't correct
7 Incorrect 59 ms 46200 KB Output isn't correct
8 Incorrect 80 ms 60408 KB Output isn't correct
9 Incorrect 124 ms 76536 KB Output isn't correct
10 Incorrect 152 ms 92792 KB Output isn't correct