Leetcode 1007 Solution

This article provides solution to leetcode question 1007 (numbers-with-same-consecutive-differences)
Categories:
Tags:
Access this page by simply typing in "lcs 1007" in your browser address bar if you have bunnylol configured.

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

Solution

class Solution:
    def numsSameConsecDiff(self, n: int, k: int) -> List[int]:
        ans = []

        def dfs(i, v):
            nonlocal k

            if i == n:
                ans.append(v)
                return

            if i == 0:
                for j in range(1, 10):
                    dfs(i + 1, j)
            else:
                for j in range(0, 10):
                    last_digit = v % 10
                    if abs(last_digit - j) == k:
                        dfs(i + 1, v * 10 + j)

        dfs(0, 0)

        return ans