Submission #1089296

#	Time	Username	Problem	Language	Result	Execution time	Memory
1089296		vjudge1	Growing Vegetable is Fun 3 (JOI19_ho_t3)	C++17	15 / 100	1 ms	348 KiB

joi2019_ho_t3

//don't copy pls)
/*TAAK ZDES NADO RECURSIU PISAT*/

//I'm not in the danger i am the DANGER
#include "bits/stdc++.h"
#include <ext/pb_ds/assoc_container.hpp>
#define ll long long
#define pb push_back
#define int long long
#define F first
#define S second
#define all(x) (x).begin(), (x).end()
#define pii pair<int,int>
#define sigma signed
using namespace std;
using namespace __gnu_pbds;
const int N = 3e5 + 5;
int mod = 1e9 + 7;
const int INF = 1e18;
int n;
string s,t;
vector <string> q;
void rec(char last , int sz){
	if(sz == n){
		q.pb(t);
		return;
	}
	int ans = 0;
	if(last != 'R'){
		t.pb('R');
		rec('R' , sz + 1);
		t.pop_back();
	}
	if(last != 'G'){
		t.pb('G');
		rec('G' , sz + 1);
		t.pop_back();
	}
	if(last != 'Y'){
		t.pb('Y');
		rec('Y' , sz + 1);
		t.pop_back();
	}
}
void Gold(){
	cin >> n >> s;
	s = '+' + s;
	string t = s;
	int ans = 0 , ans1 = INF;
	int cnt1 = 0 , cnt2 = 0;
	for(auto it : s){
		if(it == 'R') cnt1++;
		else if(it == 'G') cnt2++;
	}
	if(cnt1 + cnt2 == n){
		if(abs(cnt1 - cnt2) > 1){
			cout << "-1\n";
			return;
		}
		// cout << cnt1 << ' ' << cnt2 << '\n';
		if(cnt1 >= cnt2){
			for(int i = 1 ; i <= n ; i++){
				if(i % 2){
					for(int j = i ; j <= n ; j++){
						if(s[j] == 'R'){
							ans += (j - i);
							swap(s[i] , s[j]);
							// cout << i << ' ' << j << '\n';
							break;
						}
					}
				}
				else{
					for(int j = i ; j <= n ; j++){
						if(s[j] == 'G'){
							ans += (j - i);
							swap(s[i] , s[j]);
							break;
						}
					}
				}
			}
			ans1 = ans;
		}
		if(cnt2 >= cnt1){
			ans = 0;
			s = t;
			for(int i = 1 ; i <= n ; i++){
				if(i % 2 == 0){
					for(int j = i ; j <= n ; j++){
						if(s[j] == 'R'){
							ans += (j - i);
							swap(s[i] , s[j]);
							break;
						}
					}
				}
				else{
					for(int j = i ; j <= n ; j++){
						if(s[j] == 'G'){
							ans += (j - i);
							swap(s[i] , s[j]);
							break;
						}
					}
				}
			} 	
		}
		cout << min(ans1 , ans);
		return;
	}
	else{
		rec('A' , 0);
		int ans = -1;
		for(auto it : q){
			int ans1 = 0;
			t = '=' + it;
			for(int i = 1 ; i <= n ; i++){
				bool ok = 0;
				for(int j = i ; j <= n ; j++){
					if(t[i] == s[j]){
						ans1 += (j - i);
						swap(s[i] , s[j]);
						ok = 1;
						break;
					}
				}
				if(!ok){
					ans1 = -1;
					break;
				}
			}
			if(ans1 == -1) continue;
			ans = max(ans , ans1);
		}
		cout << ans;
	}
}
sigma main(){
	//freopen("txt.in","r",stdin);
	//freopen("txt.out","w",stdout);
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	srand(time(0));
	int TT = 1;
	// cin >> TT;
	for(int i = 1 ; i <= TT ; i++){
		//cout << "Case " << i << ": ";
		Gold();
	}
}

Compilation message (stderr)

joi2019_ho_t3.cpp: In function 'void rec(char, long long int)':
joi2019_ho_t3.cpp:28:6: warning: unused variable 'ans' [-Wunused-variable]
   28 |  int ans = 0;
      |      ^~~

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