一起学Rust-实战leetcode（六）

fn my_atoi(s: String) -> i32 {
    // 获取字节数组引用
    let bytes = s.as_bytes();

    // 字符串截取的起始和结束
    let (mut start, mut idx) = (0, 0);
    // 是否为负数
    let mut negative = false;

    // 是否遇到数字，是否遇到符号
    let (mut num, mut symbol) = (false, false);
    //定义各字符的ascii码遍历
    let (zero, plus, sub, nine, space) = (b'0', b'+', b'-', b'9', b' ');

    while idx < bytes.len() {

        let ascii = bytes[idx];
        //当遇到符号或者数字后，存在其他字符，则停止。
        if (num || symbol) && (ascii == space || ascii == plus || ascii == sub) {
            break;
        }

        if ascii == space || ascii == plus || ascii == sub {
            start += 1;
            if ascii == plus || ascii == sub {
                symbol = true;
            }
            if ascii == sub {
                negative = true;
            }
        }

        if ascii >= zero && ascii <= nine {
            num = true;
        }

        if ascii != space && ascii != plus && ascii != sub && (ascii < zero || ascii > nine) {
            break;
        }

        idx += 1;
    }
    
    // 截取开始和结束相等的情况 一定是0
    if start == idx {
        return 0;
    }

    use std::str::FromStr;
    //截取字符串，转换失败时会返回ParseIntError
    if let Ok(v) = i32::from_str(&s[start..idx]) {
        if negative {
            return v.checked_mul(-1).unwrap_or(std::i32::MIN);
        }
        return v;
    }

    if negative {
        return std::i32::MIN;
    }
    return std::i32::MAX;
}