删数游戏(单调栈)

P1106 删数问题

题意:删去k个数字,使剩下的数字最小。

题解:我们考虑两个数字,肯定高位低的数字更小,随后再尽量保证低位小。

利用单调栈维护一个n-k长的上升序列,如果新入栈的数比栈顶小就一直出栈直到不小。

因为序列要保证长度,每个数只会入栈一次,出栈就删一个数。不过很有可能没删完就结束了。

所以循环结束,需要把n-k以上的删完。特判下0。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<stack>
#define FOR(i,a,b) for(int i=a;i<=b;i++)
#define ll long long
using namespace std;

#define M 500050
#define Color 500050

int K;
stack<int>t;
int x;
int A[M];
string st;

int main(){
    cin>>st>>K;
    int k=K;
    FOR(i,1,st.size()){
        x=st[i-1]-'0';
        while(!t.empty()&&k&&x<t.top())t.pop(),k--;
        t.push(x);
    }
    while(k&&!t.empty())k--,t.pop();
    int cnt=st.size()-K;
    while(!t.empty()){
        A[cnt--]=t.top();
        t.pop();
    }
    int s=1;
    while(A[s]==0&&s<st.size()-K)s++;
    FOR(i,s,st.size()-K)printf("%d",A[i]);
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
单调栈是一种常见的数据结构,在解决一些特定问题时非常有用。它可以用来求解一些范围内的最大值、最小值、最长连续递增子序列等问题。下面是一个简单的 Python 实现: ``` class MonotonicStack: def __init__(self): self.stack = [] def push(self, val): while self.stack and self.stack[-1] < val: self.stack.pop() self.stack.append(val) def pop(self): self.stack.pop() def top(self): return self.stack[-1] def is_empty(self): return len(self.stack) == 0 ``` 这个单调栈实现了四个基本操作: - push(val):将元素 val 压入栈中。在压入之前,如果栈顶元素比 val 小,则弹出栈顶元素,直到栈顶元素比 val 大或栈为空,然后将 val 压入栈中。 - pop():弹出栈顶元素。 - top():返回栈顶元素。 - is_empty():判断栈是否为空。 使用单调栈的时候,需要根据具体问题来实现 push 操作。下面是一个例子,求解一个数组中每个元素右边第一个比它大的数: ``` def next_larger(nums): n = len(nums) res = [-1] * n stack = MonotonicStack() for i in range(n): while not stack.is_empty() and nums[stack.top()] < nums[i]: res[stack.top()] = nums[i] stack.pop() stack.push(i) return res ``` 在这个例子中,我们维护了一个单调递减的栈,每次遇到一个比栈顶元素大的数,就将栈顶元素弹出,并将栈顶元素的答案设为当前数。这样,最后栈中剩下的元素都没有右边比它们大的数,它们的答案就是 -1。时间复杂度为 O(n)。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值