4fd800b3a8 2011-02-23        kinaba: #include <iostream>
4fd800b3a8 2011-02-23        kinaba: #include <sstream>
4fd800b3a8 2011-02-23        kinaba: #include <iomanip>
4fd800b3a8 2011-02-23        kinaba: #include <vector>
4fd800b3a8 2011-02-23        kinaba: #include <string>
4fd800b3a8 2011-02-23        kinaba: #include <map>
4fd800b3a8 2011-02-23        kinaba: #include <set>
4fd800b3a8 2011-02-23        kinaba: #include <algorithm>
4fd800b3a8 2011-02-23        kinaba: #include <numeric>
4fd800b3a8 2011-02-23        kinaba: #include <iterator>
4fd800b3a8 2011-02-23        kinaba: #include <functional>
4fd800b3a8 2011-02-23        kinaba: #include <complex>
4fd800b3a8 2011-02-23        kinaba: #include <queue>
4fd800b3a8 2011-02-23        kinaba: #include <stack>
4fd800b3a8 2011-02-23        kinaba: #include <cmath>
4fd800b3a8 2011-02-23        kinaba: #include <cassert>
4fd800b3a8 2011-02-23        kinaba: #include <cstring>
4fd800b3a8 2011-02-23        kinaba: using namespace std;
4fd800b3a8 2011-02-23        kinaba: typedef long long LL;
4fd800b3a8 2011-02-23        kinaba: typedef complex<double> CMP;
4fd800b3a8 2011-02-23        kinaba: 
4fd800b3a8 2011-02-23        kinaba: class ColorfulCards { public:
4fd800b3a8 2011-02-23        kinaba: 	vector <int> theCards(int N, string colors)
4fd800b3a8 2011-02-23        kinaba: 	{
4fd800b3a8 2011-02-23        kinaba: 		vector<bool> isPrime(N+1, true);
4fd800b3a8 2011-02-23        kinaba: 		isPrime[1] = false;
4fd800b3a8 2011-02-23        kinaba: 		for(int p=2; p<=N; ++p)
4fd800b3a8 2011-02-23        kinaba: 			if( isPrime[p] )
4fd800b3a8 2011-02-23        kinaba: 				for(int q=p+p; q<=N; q+=p)
4fd800b3a8 2011-02-23        kinaba: 					isPrime[q] = false;
4fd800b3a8 2011-02-23        kinaba: 
4fd800b3a8 2011-02-23        kinaba: 		vector<int> result(colors.size(), 0);
4fd800b3a8 2011-02-23        kinaba: 		for(int i=0; i<colors.size(); ++i)
4fd800b3a8 2011-02-23        kinaba: 			for(int n=1; n<=N; ++n)
4fd800b3a8 2011-02-23        kinaba: 				if( canItBe(i, n, colors, isPrime) )
4fd800b3a8 2011-02-23        kinaba: 					if( result[i] > 0 )
4fd800b3a8 2011-02-23        kinaba: 						{result[i] = -1; break;}
4fd800b3a8 2011-02-23        kinaba: 					else
4fd800b3a8 2011-02-23        kinaba: 						result[i] = n;
4fd800b3a8 2011-02-23        kinaba: 		return result;
4fd800b3a8 2011-02-23        kinaba: 	}
4fd800b3a8 2011-02-23        kinaba: 
4fd800b3a8 2011-02-23        kinaba: 	bool canItBe(int i, int n, const string& colors, vector<bool>& isPrime)
4fd800b3a8 2011-02-23        kinaba: 	{
4fd800b3a8 2011-02-23        kinaba: 		if( colors[i]=='R' && !isPrime[n] )
4fd800b3a8 2011-02-23        kinaba: 			return false;
4fd800b3a8 2011-02-23        kinaba: 		if( colors[i]=='B' &&  isPrime[n] )
4fd800b3a8 2011-02-23        kinaba: 			return false;
4fd800b3a8 2011-02-23        kinaba: 		return greedyDown(i-1, n-1, colors, isPrime) && greedyUp(i+1, n+1, colors, isPrime);
4fd800b3a8 2011-02-23        kinaba: 	}
4fd800b3a8 2011-02-23        kinaba: 
4fd800b3a8 2011-02-23        kinaba: 	bool greedyUp(int i, int n, const string& colors, vector<bool>& isPrime)
4fd800b3a8 2011-02-23        kinaba: 	{
4fd800b3a8 2011-02-23        kinaba: 		while( i<colors.size() )
4fd800b3a8 2011-02-23        kinaba: 		{
4fd800b3a8 2011-02-23        kinaba: 			if( n >= isPrime.size() )
4fd800b3a8 2011-02-23        kinaba: 				return false;
4fd800b3a8 2011-02-23        kinaba: 			if( colors[i]=='R' && isPrime[n] )
4fd800b3a8 2011-02-23        kinaba: 				{++i; ++n;}
4fd800b3a8 2011-02-23        kinaba: 			else
4fd800b3a8 2011-02-23        kinaba: 			if( colors[i]=='B' && !isPrime[n] )
4fd800b3a8 2011-02-23        kinaba: 				{++i; ++n;}
4fd800b3a8 2011-02-23        kinaba: 			else
4fd800b3a8 2011-02-23        kinaba: 				++n;
4fd800b3a8 2011-02-23        kinaba: 		}
4fd800b3a8 2011-02-23        kinaba: 		return true;
4fd800b3a8 2011-02-23        kinaba: 	}
4fd800b3a8 2011-02-23        kinaba: 
4fd800b3a8 2011-02-23        kinaba: 	bool greedyDown(int i, int n, const string& colors, vector<bool>& isPrime)
4fd800b3a8 2011-02-23        kinaba: 	{
4fd800b3a8 2011-02-23        kinaba: 		while( i>=0 )
4fd800b3a8 2011-02-23        kinaba: 		{
4fd800b3a8 2011-02-23        kinaba: 			if( n < 1 )
4fd800b3a8 2011-02-23        kinaba: 				return false;
4fd800b3a8 2011-02-23        kinaba: 			if( colors[i]=='R' && isPrime[n] )
4fd800b3a8 2011-02-23        kinaba: 				{--i; --n;}
4fd800b3a8 2011-02-23        kinaba: 			else
4fd800b3a8 2011-02-23        kinaba: 			if( colors[i]=='B' && !isPrime[n] )
4fd800b3a8 2011-02-23        kinaba: 				{--i; --n;}
4fd800b3a8 2011-02-23        kinaba: 			else
4fd800b3a8 2011-02-23        kinaba: 				--n;
4fd800b3a8 2011-02-23        kinaba: 		}
4fd800b3a8 2011-02-23        kinaba: 		return true;
4fd800b3a8 2011-02-23        kinaba: 	}
4fd800b3a8 2011-02-23        kinaba: };
4fd800b3a8 2011-02-23        kinaba: 
4fd800b3a8 2011-02-23        kinaba: // BEGIN CUT HERE
4fd800b3a8 2011-02-23        kinaba: #include <ctime>
4fd800b3a8 2011-02-23        kinaba: double start_time; string timer()
4fd800b3a8 2011-02-23        kinaba:  { ostringstream os; os << " (" << int((clock()-start_time)/CLOCKS_PER_SEC*1000) << " msec)"; return os.str(); }
4fd800b3a8 2011-02-23        kinaba: template<typename T> ostream& operator<<(ostream& os, const vector<T>& v)
4fd800b3a8 2011-02-23        kinaba:  { os << "{ ";
4fd800b3a8 2011-02-23        kinaba:    for(typename vector<T>::const_iterator it=v.begin(); it!=v.end(); ++it)
4fd800b3a8 2011-02-23        kinaba:    os << '\"' << *it << '\"' << (it+1==v.end() ? "" : ", "); os << " }"; return os; }
4fd800b3a8 2011-02-23        kinaba: void verify_case(const vector <int>& Expected, const vector <int>& Received) {
4fd800b3a8 2011-02-23        kinaba:  bool ok = (Expected == Received);
4fd800b3a8 2011-02-23        kinaba:  if(ok) cerr << "PASSED" << timer() << endl;  else { cerr << "FAILED" << timer() << endl;
4fd800b3a8 2011-02-23        kinaba:  cerr << "\to: " << Expected << endl << "\tx: " << Received << endl; } }
4fd800b3a8 2011-02-23        kinaba: #define CASE(N) {cerr << "Test Case #" << N << "..." << flush; start_time=clock();
4fd800b3a8 2011-02-23        kinaba: #define END	 verify_case(_, ColorfulCards().theCards(N, colors));}
4fd800b3a8 2011-02-23        kinaba: int main(){
4fd800b3a8 2011-02-23        kinaba: 
4fd800b3a8 2011-02-23        kinaba: CASE(0)
4fd800b3a8 2011-02-23        kinaba: 	int N = 5;
4fd800b3a8 2011-02-23        kinaba: 	string colors = "RRR";
4fd800b3a8 2011-02-23        kinaba: 	int __[] = {2, 3, 5 };
4fd800b3a8 2011-02-23        kinaba: 	  vector <int> _(__, __+sizeof(__)/sizeof(*__));
4fd800b3a8 2011-02-23        kinaba: END
4fd800b3a8 2011-02-23        kinaba: CASE(1)
4fd800b3a8 2011-02-23        kinaba: 	int N = 7;
4fd800b3a8 2011-02-23        kinaba: 	string colors = "BBB";
4fd800b3a8 2011-02-23        kinaba: 	int __[] = {1, 4, 6 };
4fd800b3a8 2011-02-23        kinaba: 	  vector <int> _(__, __+sizeof(__)/sizeof(*__));
4fd800b3a8 2011-02-23        kinaba: END
4fd800b3a8 2011-02-23        kinaba: CASE(2)
4fd800b3a8 2011-02-23        kinaba: 	int N = 6;
4fd800b3a8 2011-02-23        kinaba: 	string colors = "RBR";
4fd800b3a8 2011-02-23        kinaba: 	int __[] = {-1, 4, 5 };
4fd800b3a8 2011-02-23        kinaba: 	  vector <int> _(__, __+sizeof(__)/sizeof(*__));
4fd800b3a8 2011-02-23        kinaba: END
4fd800b3a8 2011-02-23        kinaba: CASE(3)
4fd800b3a8 2011-02-23        kinaba: 	int N = 58;
4fd800b3a8 2011-02-23        kinaba: 	string colors = "RBRRBRBBRBRRBBRRBBBRRBBBRR";
4fd800b3a8 2011-02-23        kinaba: 	int __[] = {-1, -1, -1, -1, -1, -1, -1, -1, 17, 18, 19, 23, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 47, 53 };
4fd800b3a8 2011-02-23        kinaba: 	  vector <int> _(__, __+sizeof(__)/sizeof(*__));
4fd800b3a8 2011-02-23        kinaba: END
4fd800b3a8 2011-02-23        kinaba: CASE(4)
4fd800b3a8 2011-02-23        kinaba: 	int N = 495;
4fd800b3a8 2011-02-23        kinaba: 	string colors = "RBRRBRBBRBRRBBRRBBBRRBBBRR";
4fd800b3a8 2011-02-23        kinaba: 	int __[] = {-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1 };
4fd800b3a8 2011-02-23        kinaba: 	  vector <int> _(__, __+sizeof(__)/sizeof(*__));
4fd800b3a8 2011-02-23        kinaba: END
4fd800b3a8 2011-02-23        kinaba: CASE(5)
4fd800b3a8 2011-02-23        kinaba: 	int N = 1000;
4fd800b3a8 2011-02-23        kinaba: 	string colors = "BBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBB";
4fd800b3a8 2011-02-23        kinaba: 	int __[] = {-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1};
4fd800b3a8 2011-02-23        kinaba: 	  vector <int> _(__, __+sizeof(__)/sizeof(*__));
4fd800b3a8 2011-02-23        kinaba: END
4fd800b3a8 2011-02-23        kinaba: CASE(6)
4fd800b3a8 2011-02-23        kinaba: 	int N = 1;
4fd800b3a8 2011-02-23        kinaba: 	string colors = "B";
4fd800b3a8 2011-02-23        kinaba: 	int __[] = {1};
4fd800b3a8 2011-02-23        kinaba: 	  vector <int> _(__, __+sizeof(__)/sizeof(*__));
4fd800b3a8 2011-02-23        kinaba: END
4fd800b3a8 2011-02-23        kinaba: CASE(6)
4fd800b3a8 2011-02-23        kinaba: 	int N = 2;
4fd800b3a8 2011-02-23        kinaba: 	string colors = "R";
4fd800b3a8 2011-02-23        kinaba: 	int __[] = {2};
4fd800b3a8 2011-02-23        kinaba: 	  vector <int> _(__, __+sizeof(__)/sizeof(*__));
4fd800b3a8 2011-02-23        kinaba: END
4fd800b3a8 2011-02-23        kinaba: CASE(6)
4fd800b3a8 2011-02-23        kinaba: 	int N = 2;
4fd800b3a8 2011-02-23        kinaba: 	string colors = "B";
4fd800b3a8 2011-02-23        kinaba: 	int __[] = {1};
4fd800b3a8 2011-02-23        kinaba: 	  vector <int> _(__, __+sizeof(__)/sizeof(*__));
4fd800b3a8 2011-02-23        kinaba: END
4fd800b3a8 2011-02-23        kinaba: }
4fd800b3a8 2011-02-23        kinaba: // END CUT HERE