[数据结构与算法]字符串转换整数 (atoi)

题目链接

思路

• 将字符串转换为字符数组

• 先过滤掉前面的空格

• 定义一个布尔类型的flag

  • 去除空格后的第一个字符是’-'时:flag为false
  • 去除空格后的第一个字符是’+'时:flag为true;(没有符号时,默认true)

• 定义返回值res;循环数组将数组中的数字累加到res上

  • 结束条件:循环至非数字字符或循环结束

  • 累加前需判断当前是否超过大于1<<31 - 1或者累加后是否大于1<<31 - 1

    • 不大于1<<31 - 1时:开始累加

    • 大于1<<31 - 1时:

      • flag为true返回1<<31 - 1
      • flag为false返回-1 << 31

• 最后根据flag返回res

  • flag为true返回res
  • flag为false返回-res

java

    public int myAtoi(String s) {
        int length = s.length();
        char[] chars = s.toCharArray();

        int index = 0;
        while (index < length && chars[index] == ' ') {
            index++;
        }

        if (index == length) {
        	// 全是空格时直接返回0
            return 0;
        }

        boolean flag = true;
        char first = chars[index];
        if (first == '+') {
            index++;
        } else if (first == '-') {
            index++;
            flag = false;
        }
        int res = 0;
        while (index < length) {
            char c = chars[index];
            if (c > '9' || c < '0') {
                break;
            }
            if (res > Integer.MAX_VALUE / 10 || (res == Integer.MAX_VALUE / 10 && (c - '0') > Integer.MAX_VALUE % 10)) {
                return flag ? Integer.MAX_VALUE : Integer.MIN_VALUE;
            }
            res = res * 10 + (c - '0');
            index++;
        }
        return flag ? res : -res;
    }

go

func myAtoi(s string) int {
	length := len(s)
	arr := []rune(s)
	index := 0
	for index < length && arr[index] == ' ' {
		index++
	}
	if index == length {
		// 全是空格时直接返回0
		return 0
	}

	flag := true
	if arr[index] == '+' {
		index++
	} else if arr[index] == '-' {
		index++
		flag = false
	}

	res := 0
	for index < length {
		c := arr[index]
		if c > '9' || c < '0' {
			break
		}
		r := (int)(c - '0')
		if res > math.MaxInt32/10 || (res == math.MaxInt32/10 && r > math.MaxInt32%10) {
			if flag {
				return math.MaxInt32
			}
			return math.MinInt32
		}
		res = res*10 + r
		index++
	}
	if flag {
		return res
	}
	return -res
}