402. 移掉K位数字
时间:2020年11月15日
知识点:贪心、模拟
题目链接:https://leetcode-cn.com/problems/remove-k-digits/
题目
给定一个以字符串表示的非负整数 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。
解法
- 我们先来看如何删除一个数字
- 当num=537 我们会删除5 当num=261 我们删除6 可以看到删除的是第一个左边比右边大的
- 当num=234 我们删除最后一个
- 这里我们采用往里放的思路 放入的数字小于数组中的末尾时 数组末尾的数字弹出 循环执行 注意不要删除太多了
- 注意判断一开始是0的情况 和全0的情况
代码
#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
string removeKdigits(string num, int k) {
int count_k = 0;
vector<int> ans;
string str="";
for(char x:num){
while(ans.size() && ans.back() > x && count_k < k){
ans.pop_back();
count_k++;
}
ans.push_back(x);
}
while(k-count_k){
ans.pop_back();
count_k++;
}
int flag = 1;
for(int x:ans){
if (flag && x == '0')
continue;
flag = false;
str += x;
}
if(str=="")
str = '0';
return str;
}
};
int main()
{
string num{"10"};
int k = 2;
Solution s;
cout<<s.removeKdigits(num, k)<<endl;
return 0;
}
今天也是爱zz的一天哦!