(mut digits: Vec<i32>)
| 6 | |
| 7 | impl Solution { |
| 8 | pub fn plus_one(mut digits: Vec<i32>) -> Vec<i32> { |
| 9 | let mut i = digits.len() - 1; |
| 10 | loop { |
| 11 | if digits[i] != 9 { digits[i] += 1; break digits } |
| 12 | /* digits[i] == 9 */ |
| 13 | digits[i] = 0; |
| 14 | if i == 0 { digits.insert(0, 1); break digits } |
| 15 | i -= 1; |
| 16 | } |
| 17 | } |
| 18 | } |
| 19 | |
| 20 | #[cfg(test)] |