数据结构专题 - 解题报告 - J

原题洛谷P1484种树，又捡到一题做过的，lucky。
所谓用堆来解决问题，从来就是一个词，贪心，堆的使用是否成功取决于你贪心的策略，而贪心一般来说处理起全局最优是有难度的，本题引入了一个小有名气的进阶贪心策略，反悔贪心。
通过读题，我们发现贪心有一个尴尬之处在于，你如果选择了此刻的最大值i，那么在选择两个的条件下，
你只能选择a[j]+a[i] (j !=i-1 && j != i+1) ，
但是这无法保证一定存在 a[i]+a[j] > a[i-1]+a[i+1]，这种时候怎么处理呢？
我们想，选择a[i]和选择a[i-1]+a[i+1]一定是不兼容的，所以我们可以计算一个反悔值a[i-1]+a[i+1]-a[i]，压入队列，如果后面计算时能够调用它自然会替换之前的a[i]，注意压入前要更新反悔节点左右的关系，方便做之后的反悔节点。
这里处理一下左右边界使成环就行 l[1] = n; r[n] = 1;。其他的按着题意来。
AC代码如下：

#include<bits/stdc++.h>
#define maxn 200005
#define maxm 30000005
#define FOR(a, b, c) for(int a=b; a<=c; a++)
#define qhqh 1000000007
#define inf 2147483647
#define llinf 9223372036854775807
#define ll long long
#define pi acos(-1.0)
#define ls p<<1
#define rs p<<1|1

using namespace std;

ll ans;
int n, m, a[maxn];
int l[maxn], r[maxn];
bool vis[maxn];
struct node
{
    int x, id;
    bool operator < (const node &k) const		//排序规则
    {return x < k.x;}
};
priority_queue<node> q;

inline int read()
{
    char c=getchar();long long x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return x*f;
}

int main()
{
    n = read();
    m = read();
    if(m > n/2)		//error条件
    {
        puts("Error!");
        exit(0);
    }
    FOR(i, 1, n)
    {
        a[i] = read();
        q.push((node){a[i], i});
        r[i] = i+1; l[i] = i-1;
    }
    l[1] = n; r[n] = 1;
    FOR(i, 1, m)
    {
        while(vis[q.top().id])
            q.pop();
        node k = q.top(); q.pop();
        int kid = k.id;
        ans += k.x;
        vis[l[kid]] = vis[r[kid]] = 1;
        a[kid] = a[r[kid]] + a[l[kid]] - a[kid];			//反悔元素
        k.x = a[kid];			
        q.push(k);		//把反悔元素push进去
        l[kid] = l[l[kid]]; r[kid] = r[r[kid]];		//这里是更新反悔元素左右间关系
        r[l[kid]] = kid; l[r[kid]] = kid;
    }
    cout<<ans;
    return 0;
}