【最新2020.07.07】字节跳动 - 2021提前批 - 编程能力测试题

本文链接：https://blog.csdn.net/qq_42247231/article/details/107213307

点击上方蓝字关注我，我们一起学编程
欢迎小伙伴们分享、转载、私信、赞赏

微信搜索：编程笔记本
微信搜索：编程笔记本
微信搜索：编程笔记本

今天分享一道2021字节跳动提前批面试编程能力测试题。（2020.07.07下午的面试，是我的小伙伴参加的面试）

题目：

给定正整数 n ，找到若干个完全平方数（比如 1, 4, 9, 16, …）使得它们的和等于 n 。你需要让组成和的完全平方数的个数最少。

示例：

输入: n = 12
输出: 3
解释: 12 = 4 + 4 + 4.
输入: n = 13
输出: 2
解释: 13 = 4 + 9.

1. 暴力枚举法

题目的要求是找到若干个完全平方数，使其和为 n 。一个最自然的想法就是枚举所有可能的组合，并找到完全平方数个数最少的一组。

这么说可能还不是很清晰，我们换一种说法。

先考虑这样的完全平方数：1, 4, 9, 16, ... ，n 可以看成是下列组合：

n = 1 + (n - 1)
n = 4 + (n - 4)
n = 9 + (n - 9)
...

所以，我们要求和为 n 的最少完全平方数个数，等价于求：和为 n - 1 的最少完全平方数个数加一、和为 n - 4 的最少完全平方数个数加一、和为 n - 9 的最少完全平方数个数加一、…中的最小值。

上面的分析用一个公式表达，即为：

howMuch(n) = min{ howMuch(n - s) + 1 }, s = 1, 4, 9, ...

有了这个公式，我们就可以很容易地将其转化为如下代码：

int howMuch(int n) {
    int n_sqrt = static_cast<int>(sqrt(n));
    
    if (n == n_sqrt * n_sqrt) {
        return 1;
    }
    
    int num = INT_MAX;
    
    for (int i = 1; i <= n_sqrt; ++i) {
        num = min(num, 1 + howMuch(n - i * i))
    }
    
    return num;
}

上面的代码是完全正确的，但是会重复求解很多相同的子问题，并且当 n 稍大时，递归的层数会特别大，导致计算超时或堆栈溢出。

微信搜索：编程笔记本
微信搜索：编程笔记本
微信搜索：编程笔记本

2. 动态规划

从上面提到的公式

howMuch(n) = min{ howMuch(n - s) + 1 }, s = 1, 4, 9, ...

我们不难发现，要求 howMuch(n) 就必须要求 howMuch(n - 1) 、 howMuch(n - 4)、 howMuch(n - 9)…

那么，运用动态规划的思想，我们先将每个小的子问题都求解并存储起来，到后面求解大的问题时直接从内存中取小问题的解即可，无需重复计算与递归。

我们定义一维数组 dp[n] 表示 howMuch(n) 的值，并令初始值 dp[0] = 0 ，这将会有助求解 n 为完全平方数时的解。

下面是一种可供参考的 C++ 代码：

int howMuch(int n) {
    vector<int> dp(n + 1, INT_MAX);
    dp[0] = 0;
    
    int n_sqrt = static_cast<int>(sqrt(n));

    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= n_sqrt; ++j) {
            dp[i] = min(dp[i], 1 + dp[i - j * j]);
        }
    }

    return dp[n];
}