【Leetcode】967. Numbers With Same Consecutive Differences

题目地址：

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

给定两个正整数$n$和$k$，要求返回所有长$n$的正整数，使得每个数的每两个相邻位置的差的绝对值恰好是$k$。

直接DFS暴搜即可。要注意$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(n)$。