洛谷 P1484 种树

题目链接

题目描述

cyrcyr今天在种树，他在一条直线上挖了n个坑。这n个坑都可以种树，但为了保证每一棵树都有充足的养料，cyrcyr不会在相邻的两个坑中种树。而且由于cyrcyr的树种不够，他至多会种k棵树。假设cyrcyr有某种神能力，能预知自己在某个坑种树的获利会是多少（可能为负），请你帮助他计算出他的最大获利。

输入输出格式

输入格式：
第一行，两个正整数n,k。

第二行，n个正整数，第i个数表示在直线上从左往右数第i个坑种树的获利。

输出格式：
输出1个数，表示cyrcyr种树的最大获利。

输入样例#1：
6 3
100 1 -1 100 1 -1

输出样例#1：
200

说明

对于20%的数据，n<=20。

对于50%的数据，n<=6000。

对于100%的数据，n<=500000，k<=n/2,在一个地方种树获利的绝对值在1000000以内。

---

首先应该想到的是O(nk)的dp， 可以获得50分。

正解：贪心+堆（大根堆）

我们可以把选择分为两种， 选择 a[i] 和选择 a[i-1] 、a[i+1]
这样，将坑每3个分成一组，对于每个a[i], 要么就选择a[i],要么就a[i-1],a[i+1]一起选。
然后堆优化贪心。具体贪心看代码。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 500010;
struct node
{
    ll v;//价值
    int id;//编号
    bool operator <(node b) const
    {
        return v < b.v;
    }
};
ll a[N], ans;
int l[N], r[N];//链表
bool bj[N];//是否可以选择
priority_queue<node> q;
int main()
{
    int n, k;
    scanf("%d%d", &n, &k);
    for(int i = 1; i <= n; i++)
    {
        scanf("%lld", &a[i]);
        q.push((node){a[i], i});//默认第一种选择
        l[i] = i-1, r[i] = i+1;
    }
    int len = n;
    for(int i = 1; i <= k; i++)
    {
        while(!q.empty() && bj[q.top().id])
            q.pop();
        if(q.empty() || q.top().v < 0)
            break;//如果堆顶的价值小于0或没有选择的，就没必要继续选择，直接退出
        node t = q.top();q.pop();
        ans += t.v;
        bj[t.id] = bj[l[t.id]] = bj[r[t.id]] = 1;//标记不能选
        a[++len] = a[l[t.id]]+a[r[t.id]]-a[t.id];
        /*新添加一个点， -a[t.id]是做反悔打算，
         即不选择a[t.id]而选择a[l[t.id]]和a[r[t.id]]*/
        int l1 = l[t.id], r1 = r[t.id];
        l[len] = l[l1], r[len] = r[r1];
        r[l[len]] = len, l[r[len]] = len;//把三个点合成一个点
        q.push((node){a[len], len});
    }
    printf("%lld\n", ans);
    return 0;
}