[LeetCode] 8. String to Integer (atoi)

本题是来自LeetCode上面一道难度为Medium的题，目的是将字符串转换为数字(atoi)。这个函数在标准库中是存在的，只不过本题要求我们自己设计算法。选择本题的动机是看到这个的AC率比较低，只有14%，而这个atoi函数我们又经常使用，很想看看如何设计这个转换算法。

##一、问题描述

##二、问题分析
根据题目描述，输入的字符串只有当’ ‘、’+’、’-‘出现在数字前才是有效的。并且，’+’’-‘最多允许在数字前出现一次。因此本题的第一步是要分析所给的字符串，将关键的数字串提取出来。因此我们首先要去除数字前面的’ ‘，并且记录次数字的正负。第二步就是将数字串计算成为数字。这个看似好做，其实只要建立一个sum变量储存之前已经计算好的结果，读下一位的时候（sum*10+newdigit）即可。比如字符串42，之前已经读取了一位，那么sum=4，现在读取下一位’2’，只需要把4扩展成为十位(*10)，再在个位上+2即可。
然而，问题的关键是，如果数字串超过int类型的最大值（小于int最小值），我们需要返回INT_MAX（INT_MIN）。这是本题的难点，如何判定sum值与INT_MAX(INT_MIN)的大小关系呢？

---

显然，第一个想法是，我们完全还原原来的字符串，最后再与最大最小值比较，大于就返回max,小于就返回min。

        if (sign == 1) {
            return (sum > INT_MAX-1) ? INT_MAX : sum;
        }
        else {
            return (sum > INT_MAX) ? INT_MIN : sign*sum;
        }

然而，这样做结果却是错的，

左思右想，原来是我把sum设为了int类型，然而在存储计算字符串的过程中，sum的真实值已经超过其允许的最小值了，因此overflow了。

---

因此，我就有了第二个想法，用long存储sum的值，然后每次读取字符串的一位之后都与INT_MAX和INT_MIN比较，如果越界那么就停止读取并返回。

class Solution {
public:
    int myAtoi(string str) {
    	int i = str.find_first_not_of(' ');
    	long sum = 0;
    	bool flag = true;
    	if ( str[i] == '+') {
    		i++;
    	}
    	else if ( str[i] == '-') {
    		flag = false;
    		i++;
    	}    	
        while (isdigit(str[i])) {
        	sum = sum * 10 + (str[i] - '0');
        	i++;
        	if (sum > 2147483648) {
        		break;
        	}
        }
        if (flag == true) {
            return (sum > 2147483647) ? 2147483647 : sum;
        }
        else {
            return (sum > 2147483648) ? (-2147483648) : (0-sum);
        }
    }
};

令人高兴的是，这样做完全可行，而且效率也不错。

---

当然，做完之后，我又参考了一下discussion里面的代码，发现了第三种做法。不同于我有点暴力的做法，这种做法使用了最大最小int的特性，虽然效率也一般，但是我个人感觉，还是蛮巧妙的。

class Solution {
public:
    int myAtoi(string str) {
    	int i = str.find_first_not_of(' ');
    	int sum = 0;
    	int sign = 1;
    	if ( str[i] == '+') {
    		i++;
    	}
    	else if ( str[i] == '-') {
    		sign = -1;
    		i++;
    	}    	
        while (isdigit(str[i])) {
        	if (sum > INT_MAX/10 || (sum == INT_MAX/10 && str[i] - '0' > 7)) {
        		return (sign == 1) ? INT_MAX : INT_MIN;
        	}
        	sum = sum * 10 + (str[i] - '0');
        	i++;
        }
        return sign * sum;
    }
};

主要就是看之前sum是否已经是或者大于max/10，这样如果没有大于等于max/10，新读取一位一定不会越界。如果恰好等于max/10只需要看新读取的一位与7的大小关系（INT_MAX=2147483647）。这样的做法可以说真正利用了Int最大最小值的特性了。