[백준] 3613번 Java vs C++

3613번: Java vs C++

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


int main() {
	string s;
	cin >> s;
	if (s[0] < 97) {
		cout << "Error!"; //첫 글자가 대문자면 에러
		return 0;
	}
	vector<int> underscoreIndex;
	vector<int> capitalLetterIndex;
	for (int i = 1; i < s.size(); i++) {
		if (s[i] == '_')
			underscoreIndex.push_back(i);
		else if (char c = s[i] < 97)
			capitalLetterIndex.push_back(i);
	}
	if (underscoreIndex.size() == 0 && capitalLetterIndex.size() == 0) {
		cout << s; //언더바도 안 나오고 대문자도 안 나오면 그대로 출력
		return 0;
	}
	else if (underscoreIndex.size() != 0 && capitalLetterIndex.size() != 0) {
		cout << "Error!"; //언더바도 나오고 대문자도 나오면 에러
		return 0;
	}
	else if (s[s.size() - 1] == '_' || s[0] == '_') {
		cout << "Error!";//첫글자나 마지막 글자가 언더바면 에러
		return 0;
	}
	else if (underscoreIndex.size() != 0) {// 대문자가 안 나오는 C++ 형식의 경우
		int previous = underscoreIndex[0];
		for (int i = 1; i < underscoreIndex.size(); i++) {
			if (underscoreIndex[i] - previous == 1) {//언더바가 연속으로 두개 나오면 에러
				cout << "Error!";
				return 0;
			}
			previous = i;
		}
		int count = 0;
		for (int i : underscoreIndex) {
			s.erase(i - count, 1);
			s.replace(i - count, 1, 1, s[i - count] - 32);
			count++;
		}
	}
	else if (capitalLetterIndex.size() != 0) {//언더바가 안 나오는 Java 형식의 경우
		int count = 0;
		for (int i : capitalLetterIndex) {
			s.replace(i + count, 1, 1, s[i + count] + 32);
			s.insert(i + count, 1, '_');
			count++;
		}
	}
	cout << s;
	return 0;
}