題目描述
就so.in/.out
【背景描述】
一排 N 個數, 第 i 個數是 Ai , 你要找出 K 個不相鄰的數, 使得他們的和最大。
請求出這個最大和。
【輸入格式】
第一行兩個整數 N 和 K。
接下來一行 N 個整數, 第 i 個整數表示 Ai 。
【輸出格式】
一行一個整數表示最大和, 請注意答案可能會超過 int 範圍
【樣例輸入】
3 2
4 5 3
【樣例輸出】
7
【數據範圍】
對於 20% 的數據, N, K ≤ 20 。
對於 40% 的數據, N, K ≤ 1000 。
對於 60% 的數據, N, K ≤ 10000 。
對於 100% 的數據, N, K ≤ 100000 , 1 ≤ Ai ≤ 1000000000。
貪心,用一個堆來維護最大權值,之後每次從堆裏取出堆頂元素,將它和相鄰的兩個元素合併並修改權值,這可以用鏈表來實現。
代碼:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<set>
#define mem(a,b) memset(a,b,sizeof(a))
#define maxn 100005
using namespace std;
typedef long long ll;
inline int read()
{ char c=getchar();int x=0,y=1;
while(c<'0'||c>'9'){if(c=='-') y=-1;c=getchar();}
while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
return x*y;
}
struct node
{ int id;ll w;
friend bool operator <(const node& a,const node& b){ return a.w==b.w?a.id<b.id:a.w>b.w;}
};
set<node>q;
int n,K,pre[maxn],nex[maxn];
ll A[maxn],ans;
void solve()
{ while(K--)
{ node tmp=*q.begin();q.erase((node){tmp.id,tmp.w});
ans+=A[tmp.id];
A[tmp.id]=A[pre[tmp.id]]+A[nex[tmp.id]]-A[tmp.id];
q.erase((node){pre[tmp.id],A[pre[tmp.id]]});
q.erase((node){nex[tmp.id],A[nex[tmp.id]]});
q.insert((node){tmp.id,A[tmp.id]});
if(pre[pre[tmp.id]]) nex[pre[pre[tmp.id]]]=tmp.id;
if(nex[nex[tmp.id]]) pre[nex[nex[tmp.id]]]=tmp.id;
pre[tmp.id]=pre[pre[tmp.id]];
nex[tmp.id]=nex[nex[tmp.id]];
}
}
int main()
{ //freopen("so13.in","r",stdin);
//freopen("so.out","w",stdout);
n=read();K=read();
A[0]=-1e16;
for(int i=1;i<=n;i++)
A[i]=read(),pre[i]=i-1,nex[i]=i+1,q.insert((node){i,A[i]});
nex[n]=0;
solve();
printf("%lld",ans);
return 0;
}