福建中海建设有限公司网站辽宁省建设银行网站

题意

给定n，代表整数的长度，给定k代表两个相邻数字之间的间隔。求所有的值构成的组合

题目链接

https://leetcode.com/problems/numbers-with-same-consecutive-differences/description/

题解

dfs，有k位置要选，第一个位置我可以从1-9中选择，第二个位置是前一个位置的+k或者-k，要满足加减后的值满足>= 0并且 < k。注意k == 0的时候有重复，要去重。

class Solution {
public:vector<int> numsSameConsecDiff(int n, int k) {vector<int> res;for(int i = 1; i <= 9; i++) {dfs(i, 1, res, n, k);}return res;}void dfs(int x, int u, vector<int>& res, int n, int k) {if (u == n) {res.push_back(x);return;}int tmp = x % 10;if(tmp + k < 10) {dfs(x*10 + tmp + k, u+1, res, n, k);}if(tmp - k >= 0 && k != 0) {dfs(x*10 + tmp - k, u+1, res, n, k);}}
};

时间复杂度: $O(2^n)$
空间复杂度：$O(1)$