【C语言刷LeetCode】402. 移掉K位数字(M)

给定一个以字符串表示的非负整数 num,移除这个数中的 k 位数字,使得剩下的数字最小。

注意:

num 的长度小于 10002 且 ≥ k。
num 不会包含任何前导零。
示例 1 :

输入: num = "1432219", k = 3
输出: "1219"
解释: 移除掉三个数字 4, 3, 和 2 形成一个新的最小的数字 1219。
示例 2 :

输入: num = "10200", k = 1
输出: "200"
解释: 移掉首位的 1 剩下的数字为 200. 注意输出不能有任何前导零。
示例 3 :

输入: num = "10", k = 2
输出: "0"
解释: 从原数字移除所有的数字,剩余为空就是0。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-k-digits
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

这道题还是有点意思的,贪心和栈很容易想到,但实现起来还有小技巧。

首先轮询num字符串,放入栈中,如果找到更小的就把栈中元素置换出来,类似单调栈,不过这里是直接去掉,注意此处需要一个while循环。

最难的就是0的处理,有两处, if((idx != -1) || (num[i] != '0')),即数组首位不放0,但中间位置可以放0,这一句代码太简洁了,我也是从其他地方抄来的。

另一处便是 if (idx == -1) array[0] = '0'。

char * removeKdigits(char * num, int k){
    static char array[102400]; // 模拟栈
    int idx = -1; //
    int len = strlen(num);
    int i, j;
    int count = 0;
    int flag = 0;

    memset(array, 0, 102400);
    
    for (i = 0; i < len; i++) {
        
        while ((idx >= 0) && (num[i] < array[idx]) && (k > 0)) {
            k--;
            array[idx] = 0;
            idx--;
        }
        
        if((idx != -1) || (num[i] != '0')) {
            idx++;
            array[idx] = num[i];
        }
    }
    
    while ((k > 0) && (idx >= 0)) {
        array[idx] = 0;
        idx--; 
        k--;
    }
    
    if (idx == -1) array[0] = '0';
    
    return array;
}
发布了140 篇原创文章 · 获赞 17 · 访问量 11万+

猜你喜欢

转载自blog.csdn.net/jin615567975/article/details/104415807