实现数组加一

题目来源于力扣–https://leetcode-cn.com/

数组中的各个元素由0-9之间的数字组成,组成一个非负整数,包括0,9。除了0之外,不会有数字以0开头。数组的首位是数字的最高位,比如{1,2,3}加一操作后变成{1,2,4},{0}加一变成{1},{9}加一变成{1,0}

从数组末尾开始遍历数组,判断当前元素是不是9,如果当前元素小于9,直接加一,返回数组。
如果当前元素为9,暂且让这个位置元素为0,继续往数组前边走,继续判断元素是不是9,如果不是,加一返回数组。
如果走到数组首位元素,判断元素还是9,说明,数组内的元素全都是9,这时候可以在建立一个临时的数组,将数组首位置为1,其余位置全为0,返回数组。

main函数代码

#include<stdio.h>
#include<malloc.h>
int* plusOne(int* digits, int digitsSize, int* returnSize);
int main(){
	int a = 0;
	// 操作数组,可以修改 
	int digits[1] = {9};
	int* returnSize = &a;
	int* temp = plusOne(digits,1,returnSize);
	// 遍历数组 
	for(int i=0;i<*returnSize;i++){
		printf("%d ",*(temp+i));
	}
	return 0;
} 

加一代码

int* plusOne(int* digits, int digitsSize, int* returnSize){
	// 将数组的大小给 *returnSize
	*returnSize = digitsSize;
	// 判断元素是不是9 
	for(int i=digitsSize-1;i>=0;i--){
		// 不是9加一 
		if(digits[i]!=9){
			digits[i]++;
			return digits;
		}else{
			// 置为0 
			digits[i]=0;
		}
	}
	// 程序走到这说明数组内的元素全是9
	// 临时数组长度,需要进位,数组长度加一 
	*returnSize = digitsSize+1;
	// 申请空间 
	int* temp = (int*)malloc(sizeof(int) * (*returnSize));
	//首位置1 
	temp[0] = 1;
	// 其余位置为0 
	for(int j=1;j<*returnSize;j++){
		temp[j] = 0;
	}
	// 返回临时数组 
	return temp;
}
上一篇:【leetcode】18. 四数之和


下一篇:Drools引擎学习