P1792 [国家集训队]种树:带悔贪心

传送门

题意

分析

水群的时候发现的题目
原本想了一个很假的二维$dp$，后来发现数据范围之后直接带走
这道题正解是一个带悔贪心，这也是我第一次接触到这种类型的题目，记录一下
一般来说，贪心问题都符合局部正确到整体正确，放在这道题很明显，比如

1 19 20 19 1

如果是局部贪心的话，我们会首先选择$20$，但是从全局来看，这样我们就不能去选择左右两边的两个$19$，这种贪心只能保证局部最优，无法保证整体最优
所以，我们需要引入带悔贪心这一策略
带悔贪心，顾名思义，就是可以后悔的贪心，可以将前面一步的策略删除重新选择，怎么处理呢
比如我选择了第$3$棵树，然后发现如果选择第$2$棵树和第$4$棵树情况更优，怎么处理呢，我们将第$3$棵树的权值设置为$a[2]+a[4]-a[3]$，然后删除节点$2$和$4$，这样我们如果需要选择第$2$棵树和第$4$棵树，只需要选择第$3$棵树即可，这样既选择了第$2$棵树和第$4$棵树，又删除了第$3$棵树

代码

#include <bits/stdc++.h>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define dl(x) printf("%lld\n",x);
#define di(x) printf("%d\n",x);
#define _CRT_SECURE_NO_WARNINGS
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> PII;
typedef vector<int> VI;
const int INF = 0x3f3f3f3f;
const int N = 2e5 + 10;
const ll mod = 1000000007;
const double eps = 1e-9;
const double PI = acos(-1);
template<typename T>inline void read(T &a) {
    char c = getchar(); T x = 0, f = 1; while (!isdigit(c)) {if (c == '-')f = -1; c = getchar();}
    while (isdigit(c)) {x = (x << 1) + (x << 3) + c - '0'; c = getchar();} a = f * x;
}
int gcd(int a, int b) {return (b > 0) ? gcd(b, a % b) : a;}
int pr[N],ne[N],a[N];
bool st[N];
priority_queue<PII> q;
int n,m,ans;

void del(int x){
    ne[pr[x]] = ne[x];
    pr[ne[x]] = pr[x];
    st[x] = 1;
}

void work(){
    while(st[q.top().se]) q.pop();
    PII t = q.top();
    q.pop();
    ans += t.fi;
    a[t.se] = a[pr[t.se]] + a[ne[t.se]] - a[t.se];
    del(pr[t.se]);
    del(ne[t.se]);
    q.push(mp(a[t.se],t.se));
}

int main() {
    read(n),read(m);
    if(m * 2 > n) {
        puts("Error!");
        return 0;
    }
    for(int i = 1;i <= n;i++){
        pr[i] = i - 1;
        ne[i] = i + 1;
        read(a[i]);
        q.push(mp(a[i],i));
    }   
    pr[1] = n;
    ne[n] = 1;
    while(m--) work();
    di(ans);
    return 0;
}