문제

https://www.acmicpc.net/problem/2941

예전에는 운영체제에서 크로아티아 알파벳을 입력할 수가 없었다. 따라서, 다음과 같이 크로아티아 알파벳을 변경해서 입력했다.

예를 들어, ljes=njak은 크로아티아 알파벳 6개(lj, e, š, nj, a, k)로 이루어져 있다. 단어가 주어졌을 때, 몇 개의 크로아티아 알파벳으로 이루어져 있는지 출력한다.

dž는 무조건 하나의 알파벳으로 쓰이고, d와 ž가 분리된 것으로 보지 않는다. lj와 nj도 마찬가지이다. 위 목록에 없는 알파벳은 한 글자씩 센다.

입력

첫째 줄에 최대 100글자의 단어가 주어진다. 알파벳 소문자와 '-', '='로만 이루어져 있다.

단어는 크로아티아 알파벳으로 이루어져 있다. 문제 설명의 표에 나와있는 알파벳은 변경된 형태로 입력된다.

출력

입력으로 주어진 단어가 몇 개의 크로아티아 알파벳으로 이루어져 있는지 출력한다.

---

풀이

있는 그대로 구현

입력 받은 문자열에서 2~3글자씩 추출하여, 크로아티아 알파벳을 발견할 때마다 cnt를 증가시키는 방법이다. 주의할 점은 추출한 문자열의 길이에 따라, 인덱스 i의 위치를 잘 조정해줘야 한다는 것이다.

#include <iostream>
#include <vector>
#include <string>
#define NUM 7
using namespace std;

string croatia[NUM] = {"c=", "c-", "d-", "lj", "nj", "s=", "z="};

bool findCroatia(string token) {
	for(int j = 0; j < NUM; j++){
		if(croatia[j] == token) 
			return true; 
	}
	return false; 
}

int main()
{
	string s;
	cin >> s;

	int cnt = 0; // 크로아티아 알파벳의 개수 

	for(int i = 0; i < s.size(); i++){
		if(s.substr(i, 3) == "dz="){
			cnt++; 
			i += 2; // 다음 다음 다음 문자로 넘어가기 
			continue; 
		}
		
		string token = s.substr(i, 2);
		if(findCroatia(token)){
			i += 1; // 다음 다음 문자로 넘어가기 
		}
		
		cnt++; // 다음 문자로 넘어가기
	}

	cout << cnt; 
	
	return 0; 
}

단일 문자로 치환하기

첫번째 방법은 크로아티아 알파벳의 길이 (2글자, 3글자)에 따라 풀이 방식이 달라진다는 점에서 일관성이 떨어진다고 볼 수 있다. 다른 방식으로 풀어보자.

#include <iostream>
#include <vector>
#include <string>
using namespace std;

int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	vector<string> croatia {"c=", "c-", "dz=", "d-", "lj", "nj", "s=", "z="}; 
	
	string s;
	cin >> s;

	int index = 0; 
	for(int i = 0; i < croatia.size(); i++){
		string item = croatia[i];
		while(true){ // 크로아티아 알파벳이 발견될 때마다 #으로 치환한다. 
			index = s.find(item); 
			if(index == string::npos) break; // 더 이상 없으면 break 
			s.replace(index, item.length(), "#"); 
		}
	}

	cout << s.size(); 
	
	return 0; 
}

크로아티아 알파벳이 발견될 때마다 모두 단일 문자 #으로 치환하기 때문에, 크로아티아 알파벳의 글자수와 상관없이 일관성 있게 풀 수 있다.

틀린 풀이

#include <iostream>
#include <vector>
#include <string>
using namespace std;

int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	vector<string> croatia {"c=", "c-", "dz=", "d-", "lj", "nj", "s=", "z="}; 
	
	string s;
	cin >> s;

	int index = 0; 
	for(int i = 0; i < croatia.size(); i++){
		string item = croatia[i];
		if((index = s.find(item)) != string::npos){
			s.replace(index, item.length(), "#"); 
		}
	}

	cout << s.size(); 
	
	return 0; 
}

c=c= 처럼 한 문자열 내에서 동일한 크로아티아 알파벳이 여러번 등장할 수도 있기 때문에, 위의 코드처럼 하나의 크로아티아 알파벳에 대해 한번만 find 함수를 실행하면 오답 처리가 된다. find 함수를 while문으로 감싸서, 더 이상 크로아티아 알파벳이 발견되지 않을 때까지 find 함수를 반복해서 실행해줘야 한다.