LeetCode 刷题 [C++] 第258题.各位相加 从递归、迭代、O(1)解法到一行代码

题目描述

给定一个非负整数 num，反复将各个位上的数字相加，直到结果为一位数。

示例:

输入: 38
输出: 2
解释: 各位相加的过程为：3 + 8 = 11, 1 + 1 = 2。 由于 2 是一位数，所以返回 2。

进阶:

你可以不使用循环或者递归，且在 O(1) 时间复杂度内解决这个问题吗？

递归法

题意核心是反复将各个位上的数字相加，直到结果为一位数。
因此，最近简单子问题：将当前数各个位上的数相加；
结束条件为：当前数为一个一位数。
递归代码模板：
递归结束条件、处理当前层逻辑、下到下一层、清理当前层。
具体代码实现：

class Solution {
public:
    int addDigits(int num) {
        if(num < 10) return num;   //递归结束条件
        int ret = num%10;	   //处理当前层逻辑
        ret += addDigits(num/10);  //进入下一层
        return addDigits(ret);	   //对结果进行递归处理
    }
};

迭代法

由于题意是反复将各个位上的数字相加，直到结果为一位数。
因此迭代法的需要两层循环：
1.内层为将当前数各位进行相加；
2.外层判断内层结果是否为一位数，若不为一位数，则进入内层继续处理；
3.直至内层处理结果为一位数位置。
具体代码实现：

class Solution {
public:
    int addDigits(int num) {
        while( num/10 != 0 ) {  //外层循环
            int sum = 0;
            while(num) { 	//内层循环
                sum += num%10;
                num /= 10;
            }
            num = sum;
        }
        return num;
    }   
};

O(1)时间复杂度方法

对题意进行详细分析可知：
当num为0时，返回结果为0；
当num为1～8时，返回结果num%9,即为1～8，当num为9时，（即num%9 等于0时）返回结果为9；
当num为10～17时，返回结果num%9,即为1～8，当num为18时，（即num%9 等于 0时）返回结果为9；
当num为19～26时，返回结果num%9,即为1～8，当num为27时，（即num%9 等于 0时）返回结果为9；
以此类推，在num > 0的前提情况下，当num%9 != 0时，返回结果为num%9，当num%9 == 0时，返回结果为9;
具体代码实现：

class Solution {
public:
    int addDigits(int num) {
        if(0==num) return 0;
        int ret = num%9;
        return ret ? ret : 9;
    }
};

AC结果

一行代码实现

对上述解析也可以进一步优化为一句代码，现将num减去一，然后除9取余数，再加一，结果是一样的。这样直接省掉了判断num为零和num%9==0的步骤。
具体代码实现：

class Solution {
public:
    int addDigits(int num) {
        return 1 + (num - 1) % 9;
    }
};

AC结果