Submission #1151356

#	Time	Username	Problem	Language	Result	Execution time	Memory
1151356		koukirocks	Growing Vegetable is Fun 3 (JOI19_ho_t3)	C++20	100 / 100	56 ms	109168 KiB

joi2019_ho_t3

#include <bits/stdc++.h>
#define speed ios_base::sync_with_stdio(0); cin.tie(0)
#define all(x) (x).begin(),(x).end()
#define F first
#define S second
//#pragma GCC optimize("O3,unroll-loops")
//#pragma GCC target("avx,avx2")
//#pragma GCC target("popcnt")
 
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef double db;
typedef long double ldb;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
 
const ll MAX=2e5+10,P=1e9+7;
const ll INF=0x3f3f3f3f,oo=0x3f3f3f3f3f3f3f3f;
const ldb eps=1e-6;
const ldb PI=acos(-1.0);
const int dir[4][2]={{0,1},{0,-1},{1,0},{-1,0}};
template<typename T>
using vvector = vector<vector<T>>;

int dp[210][210][210][3];
int p[3][410];
int pre[3][410];

int main() {
    speed;
    int n;
    cin>>n;
    string s;
    cin>>s;
    s='-'+s;
    int cnt[3]={0,0,0};
    pre[0][0]=pre[1][0]=pre[2][0]=0;
    for (int i=1;i<=n;i++) {
        for (int k=0;k<3;k++) pre[k][i]=pre[k][i-1];
        if (s[i]=='R') {
            cnt[0]++;
            p[0][cnt[0]]=i;
            pre[0][i]++;
        }
        else if (s[i]=='G') {
            cnt[1]++;
            p[1][cnt[1]]=i;
            pre[1][i]++;
        }
        else if (s[i]=='Y') {
            cnt[2]++;
            p[2][cnt[2]]=i;
            pre[2][i]++;
        }
    }
    if (cnt[0]>(n+1)/2 or cnt[1]>(n+1)/2 or cnt[2]>(n+1)/2) {
        cout<<"-1\n";
        return 0;
    }
    memset(dp,0x3f,sizeof(dp));
    dp[0][0][0][0]=0;
    dp[0][0][0][1]=0;
    dp[0][0][0][2]=0;
    for (int i=0;i<=cnt[0];i++) {
        for (int j=0;j<=cnt[1];j++) {
            for (int k=0;k<=cnt[2];k++) {
                if (i!=0) dp[i][j][k][0]=min(dp[i-1][j][k][1],dp[i-1][j][k][2])+abs(pre[1][p[0][i]]-j)+abs(pre[2][p[0][i]]-k);
                if (j!=0) dp[i][j][k][1]=min(dp[i][j-1][k][0],dp[i][j-1][k][2])+abs(pre[0][p[1][j]]-i)+abs(pre[2][p[1][j]]-k);
                if (k!=0) dp[i][j][k][2]=min(dp[i][j][k-1][1],dp[i][j][k-1][0])+abs(pre[1][p[2][k]]-j)+abs(pre[0][p[2][k]]-i);
                // cout<<i<<" "<<j<<" "<<k<<" ijk\n";
                // cout<<p[0][i]<<" "<<p[1][j]<<" "<<p[2][k]<<"\n";
                // cout<<dp[i][j][k][0]<<"\n"<<dp[i][j][k][1]<<"\n"<<dp[i][j][k][2]<<"\n";
            }
        }
    }
    cout<<min({dp[cnt[0]][cnt[1]][cnt[2]][0],dp[cnt[0]][cnt[1]][cnt[2]][1],dp[cnt[0]][cnt[1]][cnt[2]][2]})/2<<"\n";
    return 0;
}

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