传送门
题目描述
小Z是一个小有名气的钢琴家,最近C博士送给了小Z一架超级钢琴,小Z希望能够用这架钢琴创作出世界上最美妙的音乐。
这架超级钢琴可以弹奏出n个音符,编号为1至n。第i个音符的美妙度为Ai,其中Ai可正可负。
一个“超级和弦”由若干个编号连续的音符组成,包含的音符个数不少于L且不多于R。我们定义超级和弦的美妙度为其包含的所有音符的美妙度之和。两个超级和弦被认为是相同的,当且仅当这两个超级和弦所包含的音符集合是相同的。
小Z决定创作一首由k个超级和弦组成的乐曲,为了使得乐曲更加动听,小Z要求该乐曲由k个不同的超级和弦组成。我们定义一首乐曲的美妙度为其所包含的所有超级和弦的美妙度之和。小Z想知道他能够创作出来的乐曲美妙度最大值是多少。
输入格式
输入第一行包含四个正整数n, k, L, R。其中n为音符的个数,k为乐曲所包含的超级和弦个数,L和R分别是超级和弦所包含音符个数的下限和上限。
接下来n行,每行包含一个整数Ai,表示按编号从小到大每个音符的美妙度。
输出格式
输出只有一个整数,表示乐曲美妙度的最大值。
输入输出样例
输入 #1复制
4 3 2 3
3
2
-6
8
输出 #1复制
11
说明/提示
共有5种不同的超级和弦:
- 音符1 ~ 2,美妙度为3 + 2 = 5
- 音符2 ~ 3,美妙度为2 + (-6) = -4
- 音符3 ~ 4,美妙度为(-6) + 8 = 2
- 音符1 ~ 3,美妙度为3 + 2 + (-6) = -1
- 音符2 ~ 4,美妙度为2 + (-6) + 8 = 4
最优方案为:乐曲由和弦1,和弦3,和弦5组成,美妙度为5 + 2 + 4 = 11。
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <vector>
#include <math.h>
#include <map>
#include <queue>
#define maxn 1000005
#define M(x,y) make_pair(x,y)
using namespace std;
typedef long long ll;
const int N=5e5+7;
const int inf=0x3f3f3f3f;
int dp[N][20],c[N],lg[N],num[N][20];
int n,k,l,r;
struct sa
{
int pos;
int l,r;
int num;
int val;
bool operator <(sa i)const {
return val<i.val;
}
};
priority_queue<sa>q;
void RMQ()
{
for(int j=1;j<=20;j++)
for(int i=1;i<=n;i++)
if(i+(1<<j)-1<=n){
if(dp[i][j-1]>dp[i+(1<<(j-1))][j-1]){
dp[i][j]=dp[i][j-1];
num[i][j]=num[i][j-1];
}
else
{
dp[i][j]=dp[i+(1<<(j-1))][j-1];
num[i][j]=num[i+(1<<(j-1))][j-1];
}
}
}
int query(int l,int r)
{
int x=lg[r-l+1];
if(dp[l][x]>dp[r-(1<<x)+1][x])
return num[l][x];
else
return num[r-(1<<x)+1][x];
}
void push(int i,int l,int r)
{
int po=query(l,r);
sa f;
f.pos=po;
f.num=i;
f.l=l;
f.r=r;
f.val=c[po]-c[i-1];
q.push(f);
}
int main()
{
//freopen("in.txt","r",stdin);
scanf("%d%d%d%d",&n,&k,&l,&r);
int x;
lg[0]=-1;
for(int i=1;i<=n;i++)
{
scanf("%d",&x);
c[i]=c[i-1]+x;
lg[i]=lg[i/2]+1;
dp[i][0]=c[i];
num[i][0]=i;
}
RMQ();
for(int i=1;i+l-1<=n;i++){
int ri=min(i+r-1,n);
push(i,i+l-1,ri);
}
ll ans=0;
while(k--){
int d=q.top().pos,left=q.top().l,right=q.top().r;
int e=q.top().val,id=q.top().num;
ans+=e;
q.pop();
if(d!=left)
push(id,left,d-1);
if(d!=right)
push(id,d+1,right);
}
printf("%lld\n", ans);
return 0;
}