P1484 种树

最新推荐文章于 2022-09-27 22:13:14 发布

Mr.Gzj

最新推荐文章于 2022-09-27 22:13:14 发布

阅读量191

点赞数

分类专栏：优先队列、堆

本文链接：https://blog.csdn.net/Endeavor_G/article/details/89322409

版权

优先队列、堆专栏收录该内容

3 篇文章 0 订阅

订阅专栏

思路：dp很容易想到，dp[i][j]表示种到第i棵树且种了j棵的最大获利，dp[i][j]=max(dp[i-1][j],dp[i-2][j-1]+a[i])

有这样一种思路：对于n个数中最大值a[i]>0，假设它两边都有数，我们可以想到，在k=1的时候，结果为a[i]；而在k=2时，要么a[i-1]和a[i+1]同时被选（即a[i-1]+a[i+1]>a[i]），要么同时放弃（即a[i-1]+a[i+1]<a[i]）才能满足贪心，因为如果只选其一的话，它们中的任意一个都比a[i]小，所以最优情况就是选它们两个或者都不选（它们的和可能小于a[i]）。

所以综上我们可以考虑到：可以将现在最大数的两边合并，如果它们的和大于最大数，就舍弃最大数，其结果增加value[i-1]+value[i+1]-value[i]。而当整个数列都被过完或最大值已为负数（结果只会负增长）的时候，就结束了。

#include<bits/stdc++.h>
using namespace std;
const int N = 5e5 + 1;
typedef long long ll;
int l[N], r[N], n, k, vis[N];
struct node {
    int id, v;
    bool operator <(const node& d)const {
        return v < d.v;
    }
} no[N];
ll ans;
priority_queue<node>q;
int main() {
    cin >> n >> k;
    for(int i = 1; i <= n; i++) {
        cin >> no[i].v;
        no[i].id = i;
        l[i] = i - 1, r[i] = i + 1;
        q.push(no[i]);

    }
    r[0] = 1, l[n + 1] = n;
    while(k--) {
        while(vis[q.top().id])
            q.pop();
        if(q.top().v < 0)
            break;
        int v = q.top().v;
        ans += v;
        int id = q.top().id;
        q.pop();
        vis[l[id]] = vis[r[id]] = 1;
        no[id].v = no[l[id]].v + no[r[id]].v - v;
        l[id] = l[l[id]], r[l[id]] = id;
        r[id] = r[r[id]], l[r[id]] = id;
        q.push({id, no[id].v});
    }
    cout << ans;
    return 0;
}