2375. Construct Smallest Number From DI String

Medium

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', then num[i] < num[i + 1].

• If pattern[i] == 'D', then num[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'.

解題

可以看這篇的解釋

func smallestNumber(pattern string) string {
    arr := []byte{'1', '2', '3', '4', '5', '6', '7', '8', '9'}
    ans := make([]byte, 0)

    for i:=0; i<len(pattern); i++ {
        fmt.Println(i)
        if pattern[i] == 'I' {
            ans = append(ans, arr[0])
            arr = arr[1:]
        } else { // D
            count := 1
            for i+1<len(pattern) && pattern[i+1] == 'D' {
                count++
                i++
            }

            if count >= len(arr) {
                ans = append(ans, reverse(arr)...)
                return string(ans) 
            }

            ans = append(ans, reverse(arr[:count+1])...)
            arr = arr[count+1:]
            i++
        }
    }

    if len(ans) == len(pattern) {
        ans = append(ans, arr[0])
    }

    return string(ans) 
}

func reverse(s []byte) []byte {
    for i, j := 0, len(s)-1; i < j; i, j = i+1, j-1 {
        s[i], s[j] = s[j], s[i]
    }

    return s
}