문자열 관련 문제를 대비하여 정규표현식 공부함.
2018 코딩테스트 문제2번을 풀면서 잘못했던 점은.. 문자열만 자를 생각부터 먼저하다보니 문제를 자세히 보지 않았다... 자세히 봤으면 자르고 시작하는 것이 아니었다..
#include <iostream>
#include <vector>
#include <string>
using namespace std;
vector<char> options;
vector<int> scores;
void init_options(string &result)
{
for (int i = 0; i < 4; i++) {
options.push_back('-');
}
int idx = 2;
for (int i = result.length() - 1 ; i >= 0; i--) {
if ((result[i] == '*') || (result[i] == '#')) {
options[idx] = result[i];
}
if ((result[i] == 'S') || (result[i] == 'D') || (result[i] == 'T')) {
idx--;
}
}
}
bool is_digit(char c)
{
if ((c >= '0') && (c <= '9'))
return true;
return false;
}
int get_score(string score, char square)
{
int iscore = stoi(score);
if (square == 'S')
return iscore;
else if (square == 'D')
return iscore * iscore;
else
return iscore * iscore * iscore;
}
void init_scores(string& result)
{
int idx = 0;
string score;
for (int i = 0; i < result.length(); i++) {
if (is_digit(result[i])) {
score.push_back(result[i]);
}
if ((result[i] == 'S') || (result[i] == 'D') || (result[i] == 'T')) {
idx++;
scores.push_back(get_score(score, result[i]));
score = "";
}
}
}
int solution(string& result)
{
init_options(result);
init_scores(result);
// calc answer
int sum = 0;
for (int i = 0; i < 3; i++) {
int score = 0;
switch (options[i]) {
case '-':
score = scores[i];
break;
case '*':
score = scores[i] * 2;
break;
case '#':
score = scores[i] * -1;
break;
}
if (options[i + 1] == '*')
score *= 2;
sum += score;
}
return sum;
}
int main()
{
string result = "1D2S3T*";
int answer = solution(result);
return 0;
}