2259. 移除指定数字得到的最大结果

题目描述

给你一个表示某个正整数的字符串 number 和一个字符 digit 。

从 number 中恰好移除一个等于 digit 的字符后，找出并返回按 十进制 表示最大的结果字符串。生成的测试用例满足 digit 在 number 中出现至少一次。

示例 1：

输入：number = "123", digit = "3"
输出："12"
解释："123" 中只有一个 '3' ，在移除 '3' 之后，结果为 "12" 。

示例 2：

输入：number = "1231", digit = "1"
输出："231"
解释：可以移除第一个 '1' 得到 "231" 或者移除第二个 '1' 得到 "123" 。
由于 231 > 123 ，返回 "231" 。

示例 3：

输入：number = "551", digit = "5"
输出："51"
解释：可以从 "551" 中移除第一个或者第二个 '5' 。
两种方案的结果都是 "51" 。

提示：

2 <= number.length <= 100
number 由数字 '1' 到 '9' 组成
digit 是 '1' 到 '9' 中的一个数字
digit 在 number 中出现至少一次

解法

方法一：暴力枚举

我们可以枚举字符串 \(\textit{number}\) 的所有位置 \(\textit{i}\)，如果 \(\textit{number}[i] = \textit{digit}\)，那么我们取 \(\textit{number}\) 的前缀 \(\textit{number}[0:i]\) 和后缀 \(\textit{number}[i+1:]\) 拼接起来，即为移除 \(\textit{number}[i]\) 后的结果。我们取所有可能的结果中最大的即可。

时间复杂度 \(O(n^2)\)，空间复杂度 \(O(n)\)。其中 \(n\) 为字符串 \(\textit{number}\) 的长度。

Python3JavaC++GoTypeScriptPHP

class Solution:
    def removeDigit(self, number: str, digit: str) -> str:
        return max(
            number[:i] + number[i + 1 :] for i, d in enumerate(number) if d == digit
        )

class Solution {
    public String removeDigit(String number, char digit) {
        String ans = "0";
        for (int i = 0, n = number.length(); i < n; ++i) {
            char d = number.charAt(i);
            if (d == digit) {
                String t = number.substring(0, i) + number.substring(i + 1);
                if (ans.compareTo(t) < 0) {
                    ans = t;
                }
            }
        }
        return ans;
    }
}

class Solution {
public:
    string removeDigit(string number, char digit) {
        string ans = "0";
        for (int i = 0, n = number.size(); i < n; ++i) {
            char d = number[i];
            if (d == digit) {
                string t = number.substr(0, i) + number.substr(i + 1, n - i);
                if (ans < t) {
                    ans = t;
                }
            }
        }
        return ans;
    }
};

func removeDigit(number string, digit byte) string {
    ans := "0"
    for i, d := range number {
        if d == rune(digit) {
            t := number[:i] + number[i+1:]
            if strings.Compare(ans, t) < 0 {
                ans = t
            }
        }
    }
    return ans
}

function removeDigit(number: string, digit: string): string {
    const n = number.length;
    let last = -1;
    for (let i = 0; i < n; ++i) {
        if (number[i] === digit) {
            last = i;
            if (i + 1 < n && number[i] < number[i + 1]) {
                break;
            }
        }
    }
    return number.substring(0, last) + number.substring(last + 1);
}

class Solution {
    /**
     * @param String $number
     * @param String $digit
     * @return String
     */
    function removeDigit($number, $digit) {
        $max = 0;
        for ($i = 0; $i < strlen($number); $i++) {
            if ($number[$i] == $digit) {
                $tmp = substr($number, 0, $i) . substr($number, $i + 1);
                if ($tmp > $max) {
                    $max = $tmp;
                }
            }
        }
        return $max;
    }
}

方法二：贪心

我们可以枚举字符串 \(\textit{number}\) 的所有位置 \(\textit{i}\)，如果 \(\textit{number}[i] = \textit{digit}\)，记录 \(\textit{digit}\) 最后一次出现的位置 \(\textit{last}\)，并且如果 \(\textit{i} + 1 < \textit{n}\) 且 \(\textit{number}[i] < \textit{number}[i + 1]\)，那么我们可以直接返回 \(\textit{number}[0:i] + \textit{number}[i+1:]\)，即为移除 \(\textit{number}[i]\) 后的结果。这是因为如果 \(\textit{number}[i] < \textit{number}[i + 1]\)，那么移除 \(\textit{number}[i]\) 后，结果一定会更大。

遍历结束，我们返回 \(\textit{number}[0:\textit{last}] + \textit{number}[\textit{last}+1:]\) 即可。

时间复杂度 \(O(n)\)，空间复杂度 \(O(n)\)。其中 \(n\) 为字符串 \(\textit{number}\) 的长度。

Python3JavaC++Go

class Solution:
    def removeDigit(self, number: str, digit: str) -> str:
        last = -1
        n = len(number)
        for i, d in enumerate(number):
            if d == digit:
                last = i
                if i + 1 < n and d < number[i + 1]:
                    break
        return number[:last] + number[last + 1 :]

class Solution {
    public String removeDigit(String number, char digit) {
        int last = -1;
        int n = number.length();
        for (int i = 0; i < n; ++i) {
            char d = number.charAt(i);
            if (d == digit) {
                last = i;
                if (i + 1 < n && d < number.charAt(i + 1)) {
                    break;
                }
            }
        }
        return number.substring(0, last) + number.substring(last + 1);
    }
}

class Solution {
public:
    string removeDigit(string number, char digit) {
        int n = number.size();
        int last = -1;
        for (int i = 0; i < n; ++i) {
            char d = number[i];
            if (d == digit) {
                last = i;
                if (i + 1 < n && number[i] < number[i + 1]) {
                    break;
                }
            }
        }
        return number.substr(0, last) + number.substr(last + 1);
    }
};

func removeDigit(number string, digit byte) string {
    last := -1
    n := len(number)
    for i := range number {
        if number[i] == digit {
            last = i
            if i+1 < n && number[i] < number[i+1] {
                break
            }
        }
    }
    return number[:last] + number[last+1:]
}

2259. 移除指定数字得到的最大结果

题目描述

解法

方法一：暴力枚举

方法二：贪心

评论