Submission #199064

#	Time	Username	Problem	Language	Result	Execution time	Memory
199064		mahmoudbadawy	Growing Vegetable is Fun 3 (JOI19_ho_t3)	C++17	15 / 100	699 ms	774708 KiB

joi2019_ho_t3

#include <bits/stdc++.h>

using namespace std;

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

int go(int x,int y,int z,int las)
{
	if(x+y+z==n) return 0;
	int &ret=mem[x][y][z][las];
	if(ret!=-1) return ret;
	ret=(1<<30);
	int cpos=x+y+z;
	if(x<pos[0].size()&&(las!=0||cpos==0))
		ret=min(ret,go(x+1,y,z,0)+abs(cpos-pos[0][x]));
	if(y<pos[1].size()&&(las!=1||cpos==0))
		ret=min(ret,go(x,y+1,z,1)+abs(cpos-pos[1][y]));
	if(z<pos[2].size()&&(las!=2||cpos==0))
		ret=min(ret,go(x,y,z+1,2)+abs(cpos-pos[2][z]));
	return ret;
}

int main()
{
	cin >> n >> s;
	memset(mem,-1,sizeof mem);
	for(int i=0;i<n;i++)
	{
		if(s[i]=='R') pos[0].push_back(i);
		else if(s[i]=='G') pos[1].push_back(i);
		else pos[2].push_back(i);
	}
	int ans=go(0,0,0,0);
	if(ans>=(1<<30)) ans=-1;
	else ans/=2;
	cout << ans << endl;
}

Compilation message (stderr)

joi2019_ho_t3.cpp: In function 'int go(int, int, int, int)':
joi2019_ho_t3.cpp:17:6: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  if(x<pos[0].size()&&(las!=0||cpos==0))
     ~^~~~~~~~~~~~~~
joi2019_ho_t3.cpp:19:6: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  if(y<pos[1].size()&&(las!=1||cpos==0))
     ~^~~~~~~~~~~~~~
joi2019_ho_t3.cpp:21:6: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  if(z<pos[2].size()&&(las!=2||cpos==0))
     ~^~~~~~~~~~~~~~

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