其他分享
首页 > 其他分享> > AtCoder Beginner Contest 248 F - Ignore Operations // 贪心 + 大根堆

AtCoder Beginner Contest 248 F - Ignore Operations // 贪心 + 大根堆

作者:互联网

传送门:F - Keep Connect (atcoder.jp)

 

题意:

给定长度为N的操作(ti,yi)。

给定初值为0的x,对其进行操作:当t为1时,将x替换为y;当t为2时,将x加上y。

最多可以跳过k步,求最终x的最大值。

 

思路:

注意到,当t为1时,进行替换操作,那么该位置前面的操作是不会对后面产生任何影响的,也就不会消耗k。

那么我们可以枚举最后一次不跳过的1操作,对于该位置的前面无需考虑,对于该位置的后面:所有的1操作都应跳过(记数量为cnt),且对于2操作选择数值前k - cnt小的跳过。

于是可以从后往前推,同时维护一个大小为k - cnt的大根堆(存跳过的2操作的数值),和一个增加值add(存不跳过的2操作的数值之和),则当每次遇到1操作时,此时的x值为(y + add),动态更新求最大值即可。

 

代码参考:

#include <bits/stdc++.h>
#define LL long long
#define PII pair<int,int>
#define fi first
#define se second
using namespace std;

const int N = 200010;

int n, k;
PII a[N];
priority_queue<int> heap;

// 维护大根堆
void func(LL& add)
{   
    while(heap.size() > k) {
        add += heap.top();
        heap.pop();
    }
}

int main()
{
    cin >> n >> k;
    for(int i = n; i; i--) cin >> a[i].fi >> a[i].se;

    LL ans = -1LL << 60, add = 0;
    for(int i = 1; i <= n; i++) {
        int t = a[i].fi, y = a[i].se;
        if(t == 1) {
            ans = max(ans, y + add);
            -- k;
            func(add);
        }
        if(t == 2) {
            if(y >= 0) add += y;
            else heap.push(y);
            func(add);
        }
        if(k < 0) break;
    }
    if(k >= 0) ans = max(ans, add);

    cout << ans << endl;

    return 0;
}

 

 

标签:Operations,AtCoder,Beginner,int,add,heap,ans,操作,define
来源: https://www.cnblogs.com/jakon/p/16184094.html