1. 程式人生 > >【BZOJ 2151】種樹(連結串列+貪心)

【BZOJ 2151】種樹(連結串列+貪心)

有一種顯然錯誤的做法:每次從堆裡取最大的,更新連結串列

比如 19,20,19 這樣的話會先選20。但是很顯然,選兩個19帶來的貢獻可能是更好的

這時我們就要想一種能做到“反悔”的方法

其實只需要做一點點修改,每次再push一個“19+19-20”進去,一樣是對的

那這個“19+19-20”的放在哪兒呢(編號)?只需要放在20的位置就行了,這樣是正確的

其實這樣做的本質 是重寫了這個問題 將其變成“N-2個物品中選M-1”個的形式,如此一直化歸,直到最後變成“N-2(M-1)個物品中選1個”

#include<bits/stdc++.h>
#define N 200005
using namespace std;
template<class T>
inline void read(T &x)
{
    x=0; int f=1;
    static char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')    ch=getchar();
    if(ch=='-') f=-1,ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    x*=f;
}
priority_queue<pair<int,int> >q;
int n,m,a[N],l[N],r[N],ans;
bool vis[N];
int main()
{
    read(n),read(m);
    if(m>(n/2)){
        puts("Error!");
        return 0;
    }
    for(int i=1;i<=n;i++)
    {
        read(a[i]);
        q.push(make_pair(a[i],i));
        l[i]=i-1,r[i]=i+1;
    }
    l[1]=n,r[n]=1;
    while(m--)
    {
        while(vis[q.top().second])  q.pop();
        int val=q.top().first;
        int now=q.top().second;
        q.pop();
        ans+=val, a[now]=a[l[now]]+a[r[now]]-val;
        vis[l[now]]=vis[r[now]]=true;
        l[now]=l[l[now]],r[l[now]]=now;
        r[now]=r[r[now]],l[r[now]]=now;
        q.push(make_pair(a[now],now));
    }
    cout<<ans;
    return 0;
}