移掉 K 位数字的思路探讨与源码
移掉 K 位数字的题目如下图,该题属于栈类和字符串类型的题目,主要考察对于栈方法的使用和字符串搜索方法的理解。本文的题目作者想到2种方法,分别是单调栈方法和数学方法,其中单调栈方法使用Java进行编写,而数学方法使用Python进行编写,当然这可能不是最优的解法,还希望各位大佬给出更快的算法。
本人认为该题目可以使用单调栈方法的思路进行解决,首先新建一个双端队列,并且计算数组的长度,然后开始遍历循环,把数组的元素取出后进行判断,然后将队列的最后的元素取出并删除,再将k值减去1,再把当前元素加入到队列中。再将所有下标小于k的队列元素抛出,并判断队列是否为空,如果队列非空就将队列的第一个元素取出并进行判断操作,直到最终循环结束并将字符串判断后返回结果。那么按照这个思路我们的Java代码如下:
#喷火龙与水箭龟
class Solution {
public String removeKdigits(String num, int k) {
Deque<Character> queueList = new LinkedList<Character>();
int numLen = num.length();
for (int ir = 0; ir < numLen; ir++) {
char indNum = num.charAt(ir);
while (! queueList.isEmpty() && k > 0 && queueList.peekLast() > indNum) {
queueList.pollLast();
k = k-1;
}
queueList.offerLast(indNum);
}
for (int jr = 0; jr < k; jr++) {
queueList.pollLast();
}
StringBuilder stringBuilder = new StringBuilder();
boolean zeroNum = true;
while (! queueList.isEmpty()) {
char dt = queueList.pollFirst();
if (zeroNum && dt == '0') {
continue;
}
zeroNum = false;
stringBuilder.append(dt);
}
if(stringBuilder.length()==0){
return "0";
}else{
return stringBuilder.toString();
}
}
}
显然,我们的单调栈方法的效果不错,还可以使用数学方法解决。首先初始化列表,计算数组的长度和k值的差值,然后开始遍历循环数组,对于每个元素都判断列表是否非空并且最后一个元素是否大于当前元素,如果是则将列表的元素弹出,并将k值减去1,直到小循环结束并把当前元素插入到列表里面,直到最循环遍历结束并返回结果。所以按照这个思路就可以解决,下面是Python代码:
#喷火龙与水箭龟
class Solution(object):
def removeKdigits(self, num, k):
stackList = []
resNum = len(num) - k
for ind in num:
while k and stackList and stackList[-1] > ind:
stackList.pop()
k = k - 1
stackList.append(ind)
return ''.join(stackList[:resNum]).lstrip('0') or '0'
从结果来说Java版本的单调栈方法的效率不错,而Python版本的数学方法的速度也很好,但应该是有更多的方法可以进一步提速的,希望朋友们能够多多指教,非常感谢。