JZOJ 4726. 【NOIP2016提高A组模拟8.22】种花

Description

经过三十多个小时的长途跋涉，小Z和小D终于到了NOI现场——南山南中学。一进校园，小D就被花所吸引了（不要问我为什么），遍和一旁的种花园丁交（J）流（L）了起来。
他发现花的摆放竟有如此奥秘：圆形广场共有 N 个种花的位置，顺时针编号1到N。并且每个位置都有一个美观度ai ，如果在这里种花就可以得到这ai 的美观度。但由于地处南山土壤肥力欠佳，两株花不能种在相邻的位置（1号和N号也算相邻位置）。校方一共给了 M 株花，经过园丁的精妙摆放，才能如此吸引小D。所以现在小D也想知道应该如何摆这 N 株花。

Input

输入第一行包含两个整数N,M 。
接下来一行包含N个正整数，依次描述美观度a1,a2,…,an 。

Output

输出一个整数，表示最佳植树方案可以得到的美观度。如果无解输出“Error!”，不包含引号。

Sample Input

7 3
1 2 3 4 5 6 7

Sample Output

15

Data Constraint

对于50%的数据满足N<=3000 。
对于100%的数据满足M<=N<=200000 ，-1000<=ai<=1000 。

Solution

-这题的解法是——贪心+堆。

• 以 $A[i]$ 为关键字建大根堆，用一个链表存放当前物品。

• 最初链表中元素是 $1~N$，$i$ 的后继是 $i+1$，前驱是 $i-1$ 。

• （当然，特别的，$1$ 的前驱是 $N$，$N$ 的后继是 $1$）。

• 执行 M 次操作，每一次操作都将堆顶元素 $k$ 取出，$ans+=A[k]$ 。

• 然后在链表中删除 $k$ 的前驱 $pre$ 和后继 $nxt$，令 $A[k]=A[pre]+A[nxt]-A[k]$ ，并更新堆。

• 时间复杂度 $O(MlogN)$

Code

#include<cstdio>
#include<queue>
using namespace std;
const int N=200001;
int ans;
int a[N],l[N],r[N];
bool bz[N];
struct data
{
    int x;
    friend bool operator <(const data x,const data y)
    {
        return a[x.x]<a[y.x];
    }
}t;
priority_queue<data>q;
inline int read()
{
    int X=0,w=1; char ch=0;
    while(ch<'0' || ch>'9') {if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0' && ch<='9') X=(X<<3)+(X<<1)+ch-'0',ch=getchar();
    return X*w;
}
int main()
{
    int n=read(),m=read();
    if(m>n>>1)
    {
        printf("Error!");
        return 0;
    }
    for(int i=1;i<=n;i++)
    {
        a[t.x=i]=read();
        l[i]=i==1?n:i-1;
        r[i]=i==n?1:i+1;
        q.push(t);
    }
    while(m--)
    {
        for(t=q.top(),q.pop();bz[t.x];t=q.top(),q.pop());
        ans+=a[t.x];
        a[t.x]=a[l[t.x]]+a[r[t.x]]-a[t.x];
        bz[l[t.x]]=bz[r[t.x]]=true;
        l[t.x]=l[l[t.x]];
        r[t.x]=r[r[t.x]];
        l[r[t.x]]=r[l[t.x]]=t.x;
        q.push(t);
    }
    printf("%d",ans);
    return 0;
}