2491 移掉K位数字
小b有个正整数num(长度为n),她希望移掉num的k位数字,
使得得到的数值最小。
输入
第一行输入一个正整数n,表示num的位数; 第二行输入一个长度为n的字符串(字符为0-9),表示该n位非负整数num (原num中没有前导0); 第三行输入一个正整数k; 其中0<k≤n<10002输出
输出一个字符串,表示去掉k位后最小的数输入样例
7 1432219 3输出样例
1219
思路:
删除前K个arr[i]>arr[i+1]的组合,这个”前K个“是动态的,思路已经是套路了。
但是自己写链表写炸了,题解大佬都是用栈写的。。。
题解代码:
#include<iostream>
#include<string.h>
#include<stdio.h>
#include<math.h>
#include<set>
#include<algorithm>
#define LL long long
#define INF 0x3f3f3f3f
using namespace std;
const int N=2e5+100;
const int M=4e5+100;
int arr[N];
string removeKDigits(string str, int k)
{
int n = str.size(),top = 0,count = 0; //count用来记录删除的次数
if(k==n) return "0"; //如果k等于str的长度则直接返回"0"
for(int i = 0; i<n; i++)
{
while(top>0&&str[top-1]>str[i]&&count<k)
{
//如果前一个数字比当前的数字大,则top又回到前一个数的位置
top--;
count++;
}
str[top++] = str[i];
}
//上面的循环操作结束后会存在两种情况,
//1:通过上面的循环操作已经把k个数都删完了。
//2:通过上面的循环操作还没把k个数全部删完
top = min(top,n-k); //情况1则top = top;
//情况2则top = n-k;(top表示字符串最后一位的下一位)
int i = 0;
for(; i<top; i++) //把前面有0的去除
if(str[i]!='0') break;
if(i==top) return "0"; //剩下的全部为零
string res = "";
for(; i<top; i++)
res+=str[i];
return res;
}
int main()
{
int n,k;
string str;
while(cin>>n>>str>>k)
{
cout<<removeKDigits(str,k)<<endl;
}
return 0;
}
THE END;