题目地址:
https://leetcode.com/problems/numbers-with-same-consecutive-differences/
给定两个正整数 n n n和 k k k,要求返回所有长 n n n的正整数,使得每个数的每两个相邻位置的差的绝对值恰好是 k k k。
直接DFS暴搜即可。要注意 k = 0 k=0 k=0的情况,如果 k = 0 k=0 k=0的话,只会向下走一个分支。代码如下:
class Solution {
public:
int n, k;
vector<int> numsSameConsecDiff(int _n, int _k) {
n = _n, k = _k;
vector<int> res;
dfs(0, 0, res);
return res;
}
void dfs(int i, int x, vector<int> &res) {
if (i == n) {
res.push_back(x);
return;
}
if (!i) {
for (int j = 1; j <= 9; j++) dfs(i + 1, j, res);
} else {
int y = x % 10;
if (y + k <= 9) dfs(i + 1, x * 10 + y + k, res);
if (k && y - k >= 0) dfs(i + 1, x * 10 + y - k, res);
}
}
};
时间复杂度 O ( 2 n ) O(2^n) O(2n),空间 O ( n ) O(n) O(n)。