Submission #99261

#	Time	Username	Problem	Language	Result	Execution time	Memory
99261		imeimi2000	Growing Vegetable is Fun 3 (JOI19_ho_t3)	C++17	15 / 100	29 ms	1152 KiB

joi2019_ho_t3

#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

const int inf = 1e7;
int n;
char str[401];
int tr[256];
vector<int> pos[3];
int dp[401][401][3];
int main() {
    ios_base::sync_with_stdio(0); cin.tie(0);
    tr['R'] = 0;
    tr['G'] = 1;
    tr['Y'] = 2;
    cin >> n >> str;
    for (int i = 0; i < n; ++i) pos[tr[str[i]]].push_back(i + 1);
    for (int j = 0; j <= pos[1].size(); ++j) {
        for (int k = 0; k <= pos[2].size(); ++k) {
            if (j + k == 0) continue;
            for (int l = 0; l < 3; ++l) dp[j][k][l] = inf;
        }
    }
    for (int i = 0; i <= pos[0].size(); ++i) {
        for (int j = 0; j <= pos[1].size(); ++j) {
            for (int k = 0; k <= pos[2].size(); ++k) {
                if (i + j + k == 0) continue;
                if (i) dp[j][k][0] = min(dp[j][k][1], dp[j][k][2]) + abs(i + j + k - pos[0][i - 1]);
                else dp[j][k][0] = inf;
                if (j) dp[j][k][1] = min(dp[j - 1][k][0], dp[j - 1][k][2]) + abs(i + j + k - pos[1][j - 1]);
                else dp[j][k][1] = inf;
                if (k) dp[j][k][2] = min(dp[j][k - 1][0], dp[j][k - 1][1]) + abs(i + j + k - pos[2][k - 1]);
                else dp[j][k][2] = inf;
            }
        }
    }
    int ans = inf;
    for (int i = 0; i < 3; ++i) ans = min(ans, dp[pos[1].size()][pos[2].size()][i]);
    printf("%d\n", ans < inf ? ans / 2 : -1);
    return 0;
}

Compilation message (stderr)

joi2019_ho_t3.cpp: In function 'int main()':
joi2019_ho_t3.cpp:19:46: warning: array subscript has type 'char' [-Wchar-subscripts]
     for (int i = 0; i < n; ++i) pos[tr[str[i]]].push_back(i + 1);
                                              ^
joi2019_ho_t3.cpp:20:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int j = 0; j <= pos[1].size(); ++j) {
                     ~~^~~~~~~~~~~~~~~~
joi2019_ho_t3.cpp:21:27: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for (int k = 0; k <= pos[2].size(); ++k) {
                         ~~^~~~~~~~~~~~~~~~
joi2019_ho_t3.cpp:26:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i <= pos[0].size(); ++i) {
                     ~~^~~~~~~~~~~~~~~~
joi2019_ho_t3.cpp:27:27: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for (int j = 0; j <= pos[1].size(); ++j) {
                         ~~^~~~~~~~~~~~~~~~
joi2019_ho_t3.cpp:28:31: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
             for (int k = 0; k <= pos[2].size(); ++k) {
                             ~~^~~~~~~~~~~~~~~~

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