码迷,mamicode.com
首页 > 其他好文 > 详细

【leetcode】Plus One (easy)

时间:2014-11-23 14:21:27      阅读:140      评论:0      收藏:0      [点我收藏+]

标签:style   blog   io   ar   color   os   sp   on   div   

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.

 

分析:思路是很清晰的,就是从数字最低位开始向上循环,如果是9就变成0,如果不是就直接当前位加1,返回。如果全都是9,就在最高位加一个1.

 

但是我写C++的代码比较少,写的时候对于哪里是最高位弄晕了。绕了好久才AC。

高位在前,就是高位是先压入vector的,那么最高位在digits.begin()。先压入的靠近下标0

如果需要新加一个最高位,那digits里面肯定都是0,压入1后,最高位变成最后压入的了,所以还需要翻转一下。

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

class Solution {
public:
    vector<int> plusOne(vector<int> &digits) {
        int i = digits.size() - 1;
        while(i >= 0)
        {
            if(digits[i] == 9)
            {
                digits[i] = 0;
                i--;
            }
            else
            {
                break;
            }
        }
        if(i < 0)
        {
            digits.push_back(1);
            reverse(digits.begin(), digits.end());
        }
        else
        {
            digits[i] += 1;
        }
        
        return digits;
    }
};

int main()
{
    Solution s;
    vector<int> in, out;
    in.push_back(1);
    in.push_back(9);
    in.push_back(9);
    out = s.plusOne(in);

    return 0;
}

 

【leetcode】Plus One (easy)

标签:style   blog   io   ar   color   os   sp   on   div   

原文地址:http://www.cnblogs.com/dplearning/p/4116384.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!