This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include<bits/stdc++.h>
#define MOD 1000000007
#define all(x) x.begin(),x.end()
#define ff first
#define ss second
#define pb push_back
#define fast ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
using namespace std;
vector<int>a,b,c;
vector<vector<vector<vector<int>>>>dp;
inline int f(int x,int y,int z,int t){
if(x==a.size() && y==b.size() && z==c.size())return 0;
if(dp[x][y][z][t]!=-1)return dp[x][y][z][t];
int mn=1e9;
if(x!=a.size() && t!=1){
int ind=upper_bound(all(b),a[x])-b.begin();
ind=min(ind,y);
int ind2=upper_bound(all(c),a[x])-c.begin();
ind2=min(ind2,z);
mn=min(mn,f(x+1,y,z,1)+y-ind+z-ind2);
}
if(y!=b.size() && t!=2){
int ind=upper_bound(all(a),b[y])-a.begin();
ind=min(ind,x);
int ind2=upper_bound(all(c),b[y])-c.begin();
ind2=min(ind2,z);
mn=min(mn,f(x,y+1,z,2)+x-ind+z-ind2);
}
if(z!=c.size() && t!=3){
int ind=upper_bound(all(a),c[z])-a.begin();
ind=min(ind,x);
int ind2=upper_bound(all(b),c[z])-b.begin();
ind2=min(ind2,y);
mn=min(mn,f(x,y,z+1,3)+x-ind+y-ind2);
}
return dp[x][y][z][t]=mn;
}
int32_t main(){
fast;
int n;
cin>>n;
string s;
cin>>s;
for(int i=0;i<n;i++){
if(s[i]=='R'){
a.pb(i);
}
if(s[i]=='G'){
b.pb(i);
}
if(s[i]=='Y'){
c.pb(i);
}
}
dp.assign((int)a.size()+5,vector<vector<vector<int>>>((int)b.size()+5,vector<vector<int>>((int)c.size()+5,vector<int>(4,-1))));
int ans=f(0,0,0,0);
if(ans==1e9)cout<<-1<<endl;
else cout<<ans<<endl;
}
Compilation message (stderr)
joi2019_ho_t3.cpp: In function 'int f(int, int, int, int)':
joi2019_ho_t3.cpp:15:6: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
15 | if(x==a.size() && y==b.size() && z==c.size())return 0;
| ~^~~~~~~~~~
joi2019_ho_t3.cpp:15:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
15 | if(x==a.size() && y==b.size() && z==c.size())return 0;
| ~^~~~~~~~~~
joi2019_ho_t3.cpp:15:36: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
15 | if(x==a.size() && y==b.size() && z==c.size())return 0;
| ~^~~~~~~~~~
joi2019_ho_t3.cpp:18:6: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
18 | if(x!=a.size() && t!=1){
| ~^~~~~~~~~~
joi2019_ho_t3.cpp:25:6: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
25 | if(y!=b.size() && t!=2){
| ~^~~~~~~~~~
joi2019_ho_t3.cpp:32:6: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
32 | if(z!=c.size() && t!=3){
| ~^~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |