Topcoder-SRM596DIV2-500

#include
#include
using namespace std;
class ColorfulRoad{
#define INF 5000
public:
int getMin(string road){
int count=road.length();
int D[count];
for(int i=1;i D[0]=0;
for(int i=1;i for(int j=0;j if(road[j]=='R' && road[i]=='G') D[i]=min(D[j]+(i-j)*(i-j),D[i]);
if(road[j]=='G' && road[i]=='B') D[i]=min(D[j]+(i-j)*(i-j),D[i]);
if(road[j]=='B' && road[i]=='R') D[i]=min(D[j]+(i-j)*(i-j),D[i]);
}
}
if(D[count-1]==INF)
return -1;
else
return D[count-1];
}
};