题目地址:
https://leetcode.com/problems/strobogrammatic-number-ii/
给定一个 n n n,返回所有的长 n n n的非负整数,其与其旋转 180 ° 180\degree 180°得数相等。以字符串列表的形式返回。
思路是递归。
n
=
0
,
1
n=0,1
n=0,1的时候答案就是空串、0, 1, 8
。对于别的
n
n
n,我们可以先求
n
−
2
n-2
n−2的答案,然后前后加上(1,1) (6,9) (8,8) (9,6)
还有(0,0)
,但(0,0)
只能在内层递归里加,在最外层的递归里不能加(不能有开头
0
0
0)。代码如下:
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class Solution {
public List<String> findStrobogrammatic(int n) {
return dfs(n, n);
}
private List<String> dfs(int k, int n) {
if (k == 0) {
return Arrays.asList("");
}
if (k == 1) {
return Arrays.asList("0", "1", "8");
}
List<String> list = dfs(k - 2, n), res = new ArrayList<>();
for (String s : list) {
if (k < n) {
res.add("0" + s + "0");
}
res.add("1" + s + "1");
res.add("6" + s + "9");
res.add("8" + s + "8");
res.add("9" + s + "6");
}
return res;
}
}
时空复杂度指数级。