洛谷P1106删数问题 C++

题目链接:https://www.luogu.com.cn/problem/P1106
题目描述:
键盘输入一个高精度的正整数N(不超过250位去掉其中任意k个数字后剩下的数字按原左右次序将组成一个新的正整数。编程对给定的N和k,寻找一种方案使得剩下的数字组成的新数最小。
输入格式:
n (高精度的正整数) k(需要删除的数字个数)
输出格式:最后剩下的最小数。
输入样例:
175438 4
输出样例:13
首先这道题和leetcode上的移除K位数字有点像,有兴趣的同学可以自己去看看,几乎是一模一样。这道题用单调栈是比较简单的,思想就是使高位的数字尽可能小。举个栗子,例如425,首先将4推入栈中,从2开始比较,2比4小,要删除4,如果不这么做,那么我们无论如何得不到最小数。
那么我们就知道了,维护一个单调栈,当前数字比栈顶小,则删除栈顶元素,如果当前是单调递增的数字的话,那么我们就触发不了这个机制,那么就按顺序从栈顶删除K个即可. 当然,最后输出也要去前导零,而且也要注意栈为空的情况(有可能删除全部数字),就输出0,否则输出栈。
代码:

#include 
#include 
#include 
#include  
#include 
#include 
using namespace std;
int main(){
     
 string n,res;
 int k,m;
 cin >> n >> k;
 m=n.length()-k;
 for(char c:n){
     
  while(k&&res.size()&&c<res.back()){
     
   res.pop_back();
   k--;
  }
  res.push_back(c);
 }
 res.resize(m);//为什么?就相当于单调递增移除后面的K位,或者K没有为0,则定义m个数字即可。
 while(!res.empty()&&res[0]=='0'){
     
  res.erase(res.begin());
 }
 if(res.empty()) cout << "0";
 else cout << res;
 return 0;
}

你可能感兴趣的:(算法,贪心算法)