【回溯】【leetcode】相邻位数字差为k的n位数

题目:

返回所有长度为 n 且满足其每两个连续位上的数字之间的差的绝对值为 k 的 非负整数 。

请注意,除了 数字 0 本身之外,答案中的每个数字都 不能 有前导零。例如,01 有一个前导零,所以是无效的;但 0 是有效的。

你可以按 任何顺序 返回答案。

示例 1:

输入:n = 3, k = 7
输出:[181,292,707,818,929]
解释:注意,070 不是一个有效的数字,因为它有前导零。

提示:

  • 2 <= n <= 9
  • 0 <= k <= 9

来源:

967. 连续差相同的数字

解题思路:回溯

定义一个数组path存储n位数

  • 结果满足条件:path的大小等于n时
  • 递归调用条件:path中最后一个数字+k或-k两种情况。注意k=0时,两种情况相同会导致结果重复。
class Solution {
public:
    vector<int> result;
    vector<int> path;
    vector<int> numsSameConsecDiff(int n, int k) {
        for (int i = 1; i <= 9; i++) {
            path.push_back(i);
            back(n, k);
            path.pop_back();
        }
        return result;
    }
    void back(int n, int k) {
        if (path.size() == n) {
            int v = path[0];
            for (int i = 1; i < n; i++) {
                v = v * 10 + path[i];
            }
            result.push_back(v);
            return;
        }
        int pre = path[path.size() - 1];
        if (pre + k < 10) {
            path.push_back(pre+k);
            back(n, k);
            path.pop_back();
        }
        if (k!=0 && pre - k >= 0) {
            path.push_back(pre-k);
            back(n, k);
            path.pop_back();
        }
    }
};

猜你喜欢

转载自blog.csdn.net/hbuxiaoshe/article/details/115129657