🖥️ 문자열 계산하기 🙋

코딩테스트 연습 - 문자열 계산하기

문제 설명

---

my_string은 "3 + 5"처럼 문자열로 된 수식입니다. 문자열 my_string이 매개변수로 주어질 때, 수식을 계산한 값을 return 하는 solution 함수를 완성해주세요.

제한사항

---

• 연산자는 +, -만 존재합니다.
• 문자열의 시작과 끝에는 공백이 없습니다.
• 0으로 시작하는 숫자는 주어지지 않습니다.
• 잘못된 수식은 주어지지 않습니다.
• 5 ≤ my_string의 길이 ≤ 100
• my_string을 계산한 결과값은 1 이상 100,000 이하입니다.
  • my_string의 중간 계산 값은 -100,000 이상 100,000 이하입니다.
  • 계산에 사용하는 숫자는 1 이상 20,000 이하인 자연수입니다.
  • my_string에는 연산자가 적어도 하나 포함되어 있습니다.
• return type 은 정수형입니다.
• my_string의 숫자와 연산자는 공백 하나로 구분되어 있습니다.

입출력 예

---

my_string result

"3 + 4"

7

입출력 예 설명

---

입출력 예 #1

• 3 + 4 = 7을 return 합니다.

제출 코드

---

2022년 12월 28일

#include <string>
#include <vector>

using namespace std;

int solution(string my_string) {
    
    int length = my_string.size();
    int temp = 0;
    vector<int> store;

    for (int i = 0; i < length; i++)
    {
        if (isdigit(my_string[i]))
        {
            temp = temp * 10 + my_string[i] - '0';
        }
        else if (my_string[i] == '+')
        {
            store.push_back(0);
        }
        else if (my_string[i] == '-')
        {
            store.push_back(-1);
        }
        else
        {
            if (temp != 0)
            {
		            store.push_back(temp);
            }
            temp = 0;
        }
    }

    store.push_back(temp);

    int sLength = store.size();
    int answer = store[0];

        for (int i = 1; i < sLength; i += 2)
        {
            if (store[i] == 0)
            {
                answer += store[i + 1];
            }
            else
            {
                answer -= store[i + 1];
            }
        }

    return answer;
}

<aside> ☝ 모든 상황을 가정해서 코드를 작성했다.

</aside>

🙋 더 알아보기

#include <sstream>으로 더 간단하게 풀 수 있다.

#include <vector>
#include <string>
#include <sstream>

using namespace std;

int solution(string my_string) {
    int answer = 0;

    stringstream ss;
    ss.str(my_string);

    ss >> answer;

    char ch = ' ';
    int temp = 0;

    while (ss)
    {
        if (ch == '+')
            answer += temp;
        else if (ch == '-')
            answer -= temp;
        ss >> ch >> temp;
    }

    return answer;
}

해당 문제를 푼 다른 사람의 코드를 가져왔다. 그래서 이 sstream 라이브러리가 어떻게 작동하기에 이렇게 간단하게 풀 수 있는지 알아보려고 한다.

#include <string>       
#include <iostream>     
#include <sstream>      
int main () {

  std::stringstream ss;

  ss << 100 << ' ' << 200;

  int foo,bar;
  ss >> foo >> bar;

  std::cout << "foo: " << foo << '\\n';
  std::cout << "bar: " << bar << '\\n';

  return 0;
}

위 코드에서 stringstream을 간략히 이해할 수 있다.

1. stringstream 형식을 갖는 ss를 선언해주었지만 초기화는 하지 않았다.
2. ss에 << 연산자를 이용해 값을 넣어주었다.
3. foo, bar라는 int형 변수를 만들어주었다.
4. ss에서 >> 연산자를 이용해 foo와 bar에 무언가를 넣었다.
5. 각 값을 출력하니 foo : 100 / bar : 200이 나왔다.

→ 여기서 알 수 있는 점은 띄어쓰기를 기준으로 숫자가 나뉘어 각 변수에 들어갔다.

덧붙이자면, stringstream은 형식에 맞게 자료를 꺼낼 수 있다. 나뉘는 분기점은 공백과 ‘\n’이다.

#include <string>    
#include <iostream>     
#include <sstream>     

int main () {
  std::stringstream ss;
  ss.str ("Example string");
  std::string s = ss.str();
  std::cout << s << '\\n';
  return 0;
}

위 코드를 보면 stromgstream의 멤버 함수인 str()을 이해하기 쉽다.

string str() const;void str(const string& s);

매개변수로 string을 받으면 문자열 변수는 이전 내용을 삭제하고 인자값으로 받은 문자열을 할당한다.

std::stringstream

std::stringstream::stringstream

std::stringstream::str