Сложность: medium
Дано целое число n, верните все стробограмматические числа длины n. Ответ можно возвращать в любом порядке.
Стробограмматическое число — это число, которое выглядит одинаково при повороте на 180 градусов (если посмотреть вверх ногами).
Пример:
Input: n = 2
Output: ["11","69","88","96"]
👨💻 Алгоритм:
1️⃣Инициализируйте структуру данных reversiblePairs, которая содержит все пары обратимых цифр. Вызовите и верните результат рекурсивной функции generateStroboNumbers(n, finalLength), где первый аргумент указывает, что текущий вызов создаст все стробограмматические числа длиной n, а второй аргумент указывает длину конечных стробограмматических чисел, которые мы будем генерировать, и будет использоваться для проверки возможности добавления '0' в начало и конец числа.
2️⃣Создайте функцию generateStroboNumbers(n, finalLength), которая вернет все стробограмматические числа длиной n:
Проверьте базовые случаи: если n == 0, верните массив с пустой строкой [""]; если n == 1, верните ["0", "1", "8"].
Вызовите generateStroboNumbers(n - 2, finalLength), чтобы получить все стробограмматические числа длиной (n-2), и сохраните их в subAns.
Инициализируйте пустой массив currStroboNums для хранения стробограмматических чисел длиной n.
3️⃣Для каждого числа в subAns добавьте все reversiblePairs в начало и конец, за исключением случая, когда текущая пара '00' и n == finalLength (потому что нельзя добавить '0' в начало числа), и добавьте это новое число в currStroboNums. В конце функции верните все стробограмматические числа, т.е. currStroboNums.
😎 Решение:
class Solution:
def __init__(self):
self.reversiblePairs = [
('0', '0'), ('1', '1'),
('6', '9'), ('8', '8'), ('9', '6')
]
def generateStroboNumbers(self, n, finalLength):
if n == 0:
return [""]
if n == 1:
return ["0", "1", "8"]
prevStroboNums = self.generateStroboNumbers(n - 2, finalLength)
currStroboNums = []
for prevStroboNum in prevStroboNums:
for a, b in self.reversiblePairs:
if a != '0' or n != finalLength:
currStroboNums.append(a + prevStroboNum + b)
return currStroboNums
def findStrobogrammatic(self, n):
return self.generateStroboNumbers(n, n)
Ставь 👍 и забирай 📚 Базу знаний