洛谷P2048 [NOI2010] 超级钢琴（RMQ+优先队列）

洛谷P2048 [NOI2010] 超级钢琴（RMQ+优先队列）

题意：

有 n 个音符，编号为 1 至 n 。第 i 个音符的美妙度为 $A_i$

我们要找到 k 段超级和弦组成的乐曲，每段连续的音符的个数 x 满足 $L\le x\le R$ ，求乐曲美妙度的最大值。

思路：

三元组(x,l,r)表示左端点为x，右端点(l,r)。在设最大值的下标为t，则问题转化成求$sum[t]-sum[i-1]$，因为sum[i-1]是定值，所以求sum在(l,r)区间内的最大值

还要注意，把最大的(x,l,r)用掉后，要把(x,l,t-1)和(x,t+1,r)放进堆，因为次大值可能存在其中

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=5e5+5;
ll sum[maxn],a[maxn],ans;
int n,L,R,k,table[maxn][30];
struct node{
	int s,l,r,t;
	ll val;
	bool operator < (const node &x)const {
		return val<x.val;
	}
};
priority_queue<node>p;

void init() {
    for (int i = 1; i <= n; i++) table[i][0] = i;
    for (int j = 1; (1 << j) <= n; j++)
        for (int i = 1; i + (1 << j) - 1 <= n; i++) {
            int x = table[i][j - 1], y = table[i + (1 << (j - 1))][j - 1];
            table[i][j] = sum[x] > sum[y] ? x : y;
        }
}
int get(int l, int r) {
    int d = log2(r - l + 1);
    int x = table[l][d], y = table[r - (1 << d) + 1][d];
    return sum[x] > sum[y] ? x : y;
}

int main(){
    scanf("%d%d%d%d",&n,&k,&L,&R);
    for(int i=1;i<=n;i++){
    	scanf("%lld",&a[i]);
    	sum[i]=sum[i-1]+a[i];
	}
    init();
    for(int i=1;i+L-1<=n;i++){
    	int y=min(i+R-1,n);
    	int d=get(i+L-1,y);
    	p.push({i,i+L-1,y,d,sum[d]-sum[i-1]});
    }
    while(k--){
    	node x=p.top();
    	p.pop();
    	ans+=x.val;
    	if(x.t-1>=x.l){
    		int d1=get(x.l,x.t-1);
    	    p.push({x.s,x.l,x.t-1,d1,sum[d1]-sum[x.s-1]});
		}
    	if(x.t+1<=x.r){
    		int d2=get(x.t+1,x.r);
    	    p.push({x.s,x.t+1,x.r,d2,sum[d2]-sum[x.s-1]});
		}
	}
    printf("%lld",ans);
}