No.66 Plus One

No.66 Plus One

Given a non-negative number represented as an array of digits, plus one to the number.

The digits are stored such that the most significant digit is at the head of the list.

题意：给定一个数组表示的非负数，实现该数加1的功能

注意：从后向前加，注意边界和进位即可。

细节！

#include "stdafx.h"
#include <map>
#include <vector>
#include <iostream>
using namespace std;

class Solution
{
public:
    vector<int> plusOne(vector<int> &digits)
    {//给定一个数组表示的非负数，实现该数加1
        int carry = 0;//保存进位
        int size = digits.size();
        vector<int> res;
        if(size == 0)
        { 
            res.push_back(1);
            return res;
        }
        res = digits;
        for(int i=size-1; i>=0 ; i--)//从后向前加
        {
            if(i == size-1)
            {
                res[i] = (digits[i]+1)%10;
                carry = (digits[i]+1)/10;
            }
            if(carry == 0)
                break;
            else
            {
                res[i] = (digits[i]+carry)%10;
                carry = (digits[i]+carry)/10;
            }
        }
        if(carry!=0)
            res.insert(res.begin(),carry);
        
        return res;
    }
};

int main()
{
    Solution sol;
    int data[] = {9,9,8,4,2};
    vector<int> test(data,data+sizeof(data)/sizeof(int));
    vector<int> res;
    for(const auto &i : test)
        cout << i << " ";
    cout << endl;
    res = sol.plusOne(test);
    for(const auto &i : res)
        cout << i << " ";
    cout << endl;    
}