[Level 1] 문자열을 정수로 바꾸기 (C++/Python)

C++)

#include <string>
#include <cctype>
using namespace std;

int solution(string s) {
    if (isdigit(s[0]))
        return stoi(s);
    else if (s[0] == '+')
        return stoi(s.substr(1));
    else
        return stoi(s.substr(1)) * -1;
}

 

 

 

Python)

def solution(s):
    if s[0] == '-':
        return -1 * int(s[1:])
    elif s[0] == '+':
        return int(s[1:])
    return int(s)