题目:
返回所有长度为 n 且满足其每两个连续位上的数字之间的差的绝对值为 k 的 非负整数 。
请注意,除了 数字 0 本身之外,答案中的每个数字都 不能 有前导零。例如,01 有一个前导零,所以是无效的;但 0 是有效的。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 3, k = 7
输出:[181,292,707,818,929]
解释:注意,070 不是一个有效的数字,因为它有前导零。
提示:
2 <= n <= 9
0 <= k <= 9
来源:
解题思路:回溯
定义一个数组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();
}
}
};