Submission #238511

#	Time	Username	Problem	Language	Result	Execution time	Memory
238511		Sorting	Fishing Game (RMI19_fishing)	C++14	0 / 100	6 ms	384 KiB

fishing

#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) * 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] = {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(){
    cout << "2\n";
    return 0;

    ios::sync_with_stdio(false);
    cin.tie(NULL);

    int t;
    cin >> n >> t;

    while(t--)
        solve_test();
}

Compilation message (stderr)

fishing.cpp: In function 'int get_answer(int, int, int, int)':
fishing.cpp:18:11: warning: decomposition declaration only available with -std=c++1z or -std=gnu++1z
     auto &[answer, solved] = dp[ab][ac][bc][stage];
           ^

• Subtask 1: 0 / 100