洛谷 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以内。

因为这两天学弟们在讲关于贪心的题目，我也来总结几个贪心的题目吧，主要是可以后悔的贪心题目

这个题显然可以通过dp来写，但是复杂度要n^2，f[i][j]表示种到第i棵树且种了j棵的最大获利，则f[i][j]=max(f[i-1][j],f[i-2][j-1]+a[i])

 

所以我们考虑贪心，他要求如果取了一个位置，那么剩下来的两个位置就不能取了，考虑先取前k个，用个堆来维护，每次拿出来下标以后，就标记左右两边，我们考虑

1 5 20 19 -5

开始最大选择20，选择第三个后，2和4位置就不能选了，但是5 + 19 > 20，明显是更优的，那我们可以选择放弃以前选的20

就可以a[2] + a[4] - a[3]作为当前的价值，再次放进堆当中维护，如果堆中的值是<0的，那么显然对我们的答案也就没有贡献了，然后还需要用个l和r数组来维护左右边界，注意更新。

不懂可以再看看代码吧：

#include <bits/stdc++.h>
#include <time.h>
#define fi first
#define se second

using namespace std;

typedef long long ll;
typedef double db;
int xx[4] = {1,-1,0,0};
int yy[4] = {0,0,1,-1};
const double eps = 1e-9;
typedef pair<int,int>  P;
const int maxn = 2e6 + 5000;
const ll mod = 1e9 + 7;
inline int sign(db a) { return a < -eps ? -1 : a > eps;}
inline int cmp(db a,db b){ return sign(a - b);}
ll mul(ll a,ll b,ll c) { ll res = 1; while(b) {  if(b & 1) res *= a,res %= c;  a *= a,a %= c,b >>= 1;  }  return res;}
ll phi(ll x) {  ll res = x;  for(ll i = 2; i * i <= x; i++) { if(x % i == 0) res = res / i * (i - 1);   while(x % i == 0) x /= i;   }  if(x > 1) res = res / x  * (x - 1);    return res;}
int fa[maxn];
int Find(int x) {   if(x != fa[x]) return fa[x] = Find(fa[x]);  return fa[x];}
ll c,n,k;
ll a[maxn];
struct node{
    ll val;
    int id;
    bool operator < (const node & b) const{
        return val < b.val;
    }
}t[maxn];
priority_queue<node>q;
int l[maxn],r[maxn];
bool vis[maxn];
int main() {
    ios::sync_with_stdio(false);
    while(cin >> n >> k){
        for(int i = 1;i <= n;i++) cin >> t[i].val,t[i].id = i,a[i] = t[i].val;
        for(int i = 1;i <= n;i++){
            q.push(t[i]);
            l[i] = i - 1;
            r[i] = i + 1;
        }
        ll ans = 0;
        int num = n;
        for(int i = 1;i <= k;i++){
            node x = q.top();
            q.pop();
            while(!q.empty() && vis[x.id]) x = q.top(),q.pop();
            if(x.val < 0) break;
            ans += x.val,num++;
            vis[x.id] = 1,vis[l[x.id]] = 1,vis[r[x.id]] = 1;
            int ll = l[x.id],rr = r[x.id];
            l[num] = l[ll],r[num] = r[rr];
            r[l[num]] = num,l[r[num]] = num;
            a[num] = a[ll] + a[rr] - a[x.id];
            q.push(node{a[num],num});
        }
        cout << ans << endl;
    }
    cerr << "time: " << (long long)clock() * 1000 / CLOCKS_PER_SEC << " ms" << endl;
    return 0;
}