2375. Construct Smallest Number From DI String
Description
You are given a 0-indexed string pattern
of length n
consisting of the characters 'I'
meaning increasing and 'D'
meaning decreasing.
A 0-indexed string num
of length n + 1
is created using the following conditions:
num
consists of the digits'1'
to'9'
, where each digit is used at most once.- If
pattern[i] == 'I'
, thennum[i] < num[i + 1]
. - If
pattern[i] == 'D'
, thennum[i] > num[i + 1]
.
Return the lexicographically smallest possible string num
that meets the conditions.
Example 1:
Input: pattern = "IIIDIDDD" Output: "123549876" Explanation: At indices 0, 1, 2, and 4 we must have that num[i] < num[i+1]. At indices 3, 5, 6, and 7 we must have that num[i] > num[i+1]. Some possible values of num are "245639871", "135749862", and "123849765". It can be proven that "123549876" is the smallest possible num that meets the conditions. Note that "123414321" is not possible because the digit '1' is used more than once.
Example 2:
Input: pattern = "DDD" Output: "4321" Explanation: Some possible values of num are "9876", "7321", and "8742". It can be proven that "4321" is the smallest possible num that meets the conditions.
Constraints:
1 <= pattern.length <= 8
pattern
consists of only the letters'I'
and'D'
.
Solution
construct-smallest-number-from-di-string.py
class Solution:
def smallestNumber(self, pattern: str) -> str:
n = len(pattern)
used = [False] * 10
ans = "987654321"
def dfs(index, s, prev):
nonlocal ans
if index == n:
ss = "".join(s)
if ss < ans:
ans = ss
return
if pattern[index] == "I":
for i in range(prev + 1, 10):
if used[i]: continue
used[i] = True
s.append(str(i))
dfs(index + 1, s, i)
used[i] = False
s.pop()
else:
for i in range(prev - 1, 0, -1):
if used[i]: continue
used[i] = True
s.append(str(i))
dfs(index + 1, s, i)
used[i] = False
s.pop()
for i in range(1, 10):
used[i] = True
dfs(0, [str(i)], i)
used[i] = False
return ans