LeetCode - 8. String to Integer (atoi)

문제 개요

난이도 - MEDIUM 사용 언어 - C++

문자열에서 정수형으로 parse하는 atoi 함수를 구현하는 문제입니다.

변환 시 적용되는 규칙은 아래와 같습니다.

공백 문자는 모두 무시한다.
‘-’, ‘+’ 문자가 처음 등장하고 이전에 등장한 적이 없다면, 해당 부호가 정수의 부호를 결정합니다. 만약 어떤 부호도 등장하지 않았다면 양수로 가정합니다.
숫자 문자가 아닌 문자가 등장하면 남은 모든 문자는 무시합니다.
숫자 문자는 정수형으로 변환합니다. 만약 숫자가 아니라면 정수 값은 0으로 결정됩니다.
만약 정수형 범위를 벗어난다면 정수형 범위의 최대/최소 값을 반환합니다.

문제 - LeetCode 8. String to Integer (atoi)

풀이

Solution

for (int index = 0; index < s.size() ; index++)
{
    char chr = s[index];
    
    ...
}

먼저 문자열을 순회합니다.

if ((chr < '0' || chr > '9') && !((result == 0) && (chr == '+' || chr == '-')))
{
    if (result == 0 && !(chr == '+' || chr == '-' || chr == ' ')) break;
    if (chr == '.') break;

    if (findSign) break;
    else continue;
}

순회 중 숫자 문자가 아니거나, 이미 부호가 결정된 상태에서 -, + 문자가 탐색되는 경우를 처리합니다.

if (chr == '+') 
{
    if (findSign) break;
    neg = false;
    findSign = true;
}
else if (chr == '-') 
{
    if (findSign) break;
    neg = true;
    findSign = true;
}
else 
{
    int num = (neg ? -(chr - '0') : (chr - '0'));
    if (result > INT_MAX/10 || (result == INT_MAX / 10 && num > 7)) return INT_MAX;
    if (result < INT_MIN/10 || (result == INT_MIN / 10 && num < -8)) return INT_MIN;

    findSign = true;
    result *= 10;
    result = result + num;
}

부호 문자라면 부호를 결정하고, 숫자라면 수를 계산하여 result 변수에 저장합니다.

제출 결과

Solution 1 result

실행 결과 0ms가 나왔습니다.

코드 전문

#include <string>
#include <climits>

class Solution 
{
public:
    int myAtoi(std::string s) 
    {
        if (!s.size()) return 0;

        int result = 0;
        bool neg = false;
        bool findSign = false;

        for (int i = s.size() - 1; i >= 0 ; i--)
        {
            int index = s.size() - i - 1;
            char chr = s[index];

            if ((chr < '0' || chr > '9') && !((result == 0) && (chr == '+' || chr == '-')))
            {
                if (result == 0 && !(chr == '+' || chr == '-' || chr == ' ')) break;
                if (chr == '.') break;

                if (findSign) break;
                else continue;
            }

            if (chr == '+') 
            {
                if (findSign) break;
                neg = false;
                findSign = true;
            }
            else if (chr == '-') 
            {
                if (findSign) break;
                neg = true;
                findSign = true;
            }
            else 
            {
                int num = (neg ? -(chr - '0') : (chr - '0'));
                if (result > INT_MAX/10 || (result == INT_MAX / 10 && num > 7)) return INT_MAX;
                if (result < INT_MIN/10 || (result == INT_MIN / 10 && num < -8)) return INT_MIN;

                findSign = true;
                result *= 10;
                result = result + num;
            }
        }

        return result;
    }
};