LeetCode筆試題 Plus One （給數組加一）

題目描述

給定一個由整數組成的非空數組所表示的非負整數，在該數的基礎上加一。

最高位數字存放在數組的首位， 數組中每個元素只存儲一個數字。

你可以假設除了整數 0 之外，這個整數不會以零開頭。

示例 1:

輸入: [1,2,3]
輸出: [1,2,4]
解釋: 輸入數組表示數字 123。

示例 2:

輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入數組表示數字 4321。

/*
思路：從數組最後一位開始+1
	  先判斷最後一位是否爲9
*/
void plusOne(vector<int>&nums)
{
	auto it = nums.rbegin();
	bool flag = false;//標誌位 是否需要進位 false不需要
	*it += 1;
	for (; it != nums.rend();++it)
	{		
		if (flag)
		{
			*it+=1;
			flag = false;
		}	
		if (*it >= 10)
		{
			flag = true;
			*it %= 10;
		}
	}
	//注意，此時還需要判斷是否需要進位 比如999+1 = 1000 進位1
	//！！！一定要將flag判斷至沒有進位爲止
	if (flag)
	{
		nums.insert(nums.begin(), 1);
	}
}
int main()
{
	vector<int>nums = {9,9,9};
	plusOne(nums);
	for (int val : nums)
	{
		cout << val;
	}
	cout << endl;
	getchar();
	return 0;
}

 這個問題還有擴展就是 大數字的相加 是如何實現的？

詳細可參考我的另一篇博客 利用STL容器進行大數字的相加相減

https://blog.csdn.net/zhangfei5354/article/details/89283637