Submission #121655

#	Time	Username	Problem	Language	Result	Execution time	Memory
121655		zoooma13	Growing Vegetable is Fun 3 (JOI19_ho_t3)	C++14	0 / 100	6 ms	3584 KiB

joi2019_ho_t3

#include <bits/stdc++.h>
using namespace std;

#define MAX_N 65
#define INF 0x3f3f3f3f

int n;
string s;
vector <int> pos[3];

int dp[3][MAX_N][MAX_N][MAX_N] ,root = -1;
int to_rch[3][MAX_N][MAX_N];
int solve(int t ,int p0=0 ,int p1=0 ,int p2=0){
    int i = p0+p1+p2;
    if(i == n)
        return 0;
    int&ret = (~t ? dp[t][p0][p1][p2] : root);
    if(~ret)
        return ret;
    ret = INF;

    int st[] = {p0 ,p1 ,p2};
    for(int nt=0; nt<3; nt++)
        if(t != nt && st[nt] < pos[nt].size()){
            st[nt]++;
            ret = min(ret ,to_rch[nt][i][st[nt]-1]+solve(nt ,st[0] ,st[1] ,st[2]));
            st[nt]--;
        }
    return ret;
}

int main()
{
    cin >> n;
    cin >> s; n = s.size();

    for(int i=0; i<n; i++){
        if(s[i] == 'R') pos[0].push_back(i);
        if(s[i] == 'G') pos[1].push_back(i);
        if(s[i] == 'Y') pos[2].push_back(i);
    }
    memset(to_rch ,INF ,sizeof to_rch);
    for(int t=0; t<3; t++){
        for(int i=0; i<pos[t].size(); i++)
        for(int o=i; o<n-(pos[t].size()-i-1); o++)
            to_rch[t][o][i] = max(pos[t][i]-o ,0);
    }

    memset(dp ,-1 ,sizeof dp);
    int ans = solve(-1);
    cout << (ans >= 1e9 ? -1 : ans) << endl;
}

Compilation message (stderr)

joi2019_ho_t3.cpp: In function 'int solve(int, int, int, int)':
joi2019_ho_t3.cpp:24:30: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         if(t != nt && st[nt] < pos[nt].size()){
                       ~~~~~~~^~~~~~~~~~~~~~~~
joi2019_ho_t3.cpp: In function 'int main()':
joi2019_ho_t3.cpp:44:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(int i=0; i<pos[t].size(); i++)
                      ~^~~~~~~~~~~~~~
joi2019_ho_t3.cpp:45:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(int o=i; o<n-(pos[t].size()-i-1); o++)
                      ~^~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 5
• Subtask 2: 0 / 55
• Subtask 3: 0 / 15
• Subtask 4: 0 / 25