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

题目

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

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

示例1:

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

示例2:

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

示例3:

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

提示:

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

解法

解法一:

Java

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
public String removeDigit(String number, char digit) {
TreeSet<String> result = new TreeSet<String>(new Comparator<String>() {
public int compare(String o1, String o2) {
return o2.compareTo(o1);
}
});
StringBuilder sb = new StringBuilder(number);
for (int i = 0;i < sb.length();i++) {
if (digit == sb.charAt(i)) {
if (i == 0) {
result.add(sb.substring(i + 1));
} else if (i == sb.length() - 1) {
result.add(sb.substring(0, sb.length() - 1));
} else {
result.add(sb.substring(0, i) + sb.substring(i + 1, sb.length()));
}
}
}
return result.first();
}
0%