967. 连续差相同的数字
2022-11-05 本文已影响0人
程序员小2
题目:
返回所有长度为 n 且满足其每两个连续位上的数字之间的差的绝对值为 k 的 非负整数 。
请注意,除了 数字 0 本身之外,答案中的每个数字都 不能 有前导零。例如,01 有一个前导零,所以是无效的;但 0 是有效的。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 3, k = 7
输出:[181,292,707,818,929]
解释:注意,070 不是一个有效的数字,因为它有前导零。
示例 2:
输入:n = 2, k = 1
输出:[10,12,21,23,32,34,43,45,54,56,65,67,76,78,87,89,98]
示例 3:
输入:n = 2, k = 0
输出:[11,22,33,44,55,66,77,88,99]
示例 4:
输入:n = 2, k = 2
输出:[13,20,24,31,35,42,46,53,57,64,68,75,79,86,97]
提示:
2 <= n <= 9
0 <= k <= 9
java代码:
class Solution {
public int[] numsSameConsecDiff(int N, int K) {
Set<Integer> cur = new HashSet();
for (int i = 1; i <= 9; ++i)
cur.add(i);
for (int steps = 1; steps <= N-1; ++steps) {
Set<Integer> cur2 = new HashSet();
for (int x: cur) {
int d = x % 10;
if (d-K >= 0)
cur2.add(10*x + (d-K));
if (d+K <= 9)
cur2.add(10*x + (d+K));
}
cur = cur2;
}
if (N == 1)
cur.add(0);
int[] ans = new int[cur.size()];
int t = 0;
for (int x: cur)
ans[t++] = x;
return ans;
}
}