[boj : 12739] 돌림판(small)

www.acmicpc.net/problem/12739

12739번: 돌림판 (Small)

문제 : 조건에 맞게 하나씩 시뮬레이션 돌리면 된다.

 

해설 : 주의할점은 '동시에' 돌림판들이 바뀌어야 된다는 점. 이를 위해 새로운 문자열을 선언한 후 이를 붙여넣고 다시 비교해주는 식으로 구현해 주자. 

실수만 없다면 다들 잘 풀 수 있는 문제!!

 

#include<bits/stdc++.h>

using namespace std;

int n, k;
int arr[1000];
string s;

int draw(int x);

int main()
{
	string ans = "";
	cin >> n >> k >> s;
	
	while (k--) {
		ans = "";
		for (int i = 0; i < s.length(); i++) {
			if (s[i] == 'R')arr[i]=0;
			else if (s[i] == 'G')arr[i]=1;
			else arr[i]=2;
		}
		for (int i = 0; i < s.length(); i++) {
			int ret = draw(i);
			if (ret == 2)
				ans += 'B';
			else if (ret == 1)
				ans += 'R';
			else
				ans += 'G';
		}
		s = ans;
	}
	arr[0] = arr[1] = arr[2] = 0;
	for (int i = 0; i < s.length(); i++) {
		if (s[i] == 'R')arr[0]++;
		else if (s[i] == 'G')arr[1]++;
		else arr[2]++;
	}
	for (int i = 0; i < 3; i++)
		cout << arr[i] << " ";
	return 0;
}
int draw(int x)
{
	int len = s.length();
	int cnt[3] = { 0,0,0 };

	if (x == 0)
		cnt[arr[len - 1]]++;
	else
		cnt[arr[x - 1]]++;	
	cnt[arr[x]]++;
	cnt[arr[(x + 1) % len]]++;

	if ((cnt[0]==1 && cnt[1]==1 && cnt[2]==1) || (cnt[0] == 3 || cnt[1] == 3 || cnt[2] == 3))return 2;
	if ((cnt[0] == 2 && cnt[1] == 1) || (cnt[1] == 2 && cnt[2] == 1) || (cnt[2] == 2 && cnt[0] == 1))return 1;
	else return 0;
}