(noip 模擬 flower 種花)

Problem

有n個帶權點圍成一圈,取出m個,使取出點的權值和最大,且任意兩個相鄰點不會都被取到


Solution

如果選第i個點,ans+=val[i]
如果選第i個點左右兩個點,ans+=val[l](前驅)+val[r](後繼)
但是這兩種方案哪個更優是無法確定的
把某個點的權值取出後,將它的權值賦爲val[l]+val[r]-val[i]
這樣以後如果在選這個點,+val-val相互抵消,實際上ans+=val[l]+val[r],相當於選第i個點的前驅和後繼
這裏的前驅後繼並不是i-1和i+1,因爲從隊列裏取出i後,就會讓i出的權值改爲val[l]+val[r]-val[i],之後必須把val[l]和val[r]都賦爲0,纔不會重複
因此需要改前驅和後繼,具體看代碼


Code

// by spli
#include<cstring>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;

const int N=3010;
int n,m;
int a[N];
int L[N],R[N];
bool vis[N];
struct node{
    int k;
};
priority_queue<node> q;
int ans=0;

bool operator <(const node &x,const node &y){
    return a[x.k]<a[y.k];
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i) scanf("%d",&a[i]);
    for(int i=1;i<=n;++i) q.push((node){i});
    L[1]=n;R[1]=2;
    L[n]=n-1;R[n]=1;
    for(int i=2;i<=n-1;++i)
        L[i]=i-1,R[i]=i+1;
    while(m--){
        node f=q.top();
        q.pop();
        while(vis[f.k]){
            f=q.top();
            q.pop();
        }
        ans+=a[f.k];
        a[f.k]=a[L[f.k]]+a[R[f.k]]-a[f.k];
        vis[L[f.k]]=1;
        vis[R[f.k]]=1;
        L[f.k]=L[L[f.k]];
        R[f.k]=R[R[f.k]];
        R[L[f.k]]=f.k;
        L[R[f.k]]=f.k;
        q.push((node){f.k});
    }
    cout<<ans;
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章