题意:
给出一个序列a,一个序列b,长度都是n,现在要求在a序列的后面的n+1..2*n的位置上,每个位置赋值,要求赋值之和最大,求其和。要求i位置上的赋值要先选取1-n中的一个k,ai满足以下条件:
ai<=max{(aj)-j|bk<=j< i} 每个k只能选取一次
题解:
限制条件意思是每个ai,计算出(ai)-i,然后在给aj赋值的时候,不能超过a序列已经赋值的一段后缀的最大值。那么这个后缀很有趣,an+1赋值完成后,对an+2的赋值必然形成限制,这个限制具体表现为:如果an+1比后缀中的最大值还大,那么对于an+2的赋值是有利的,如果小,没任何影响。那么要想赋值的n个数字和值最大,贪心的思想是:要先保证第一个数最大,然后第二个数最大……同时,注意到bk越小,那么MAX只可能更大,因此待赋值位置可以赋更大的值。所以,先把b排序,n+i位置的赋值选取的bk就是bi。那么限制集中的元素的index满足了单调性,就用一个单调队列维护他就可以了。算法因为有了priority_queue的操作,复杂度是O(nlogn)。
Code:
#include<bits/stdc++.h>
using namespace std;
#define MAX 250050
#define MOD 1000000007
struct ST{
int key;
int index;
bool operator < (ST a)const
{
return key<a.key;
}
};
priority_queue <ST>lim;
int n;
int b[MAX];
long long ans;
int main(){
int num=1;
while (scanf("%d",&n)!=EOF){
while (!lim.empty()){
lim.pop();
}
for (int i=1;i<=n;i++){
int temp;
scanf("%d",&temp);
ST aa;
aa.key = temp-i;aa.index=i;
lim.push(aa);
}
ans =0;
for (int i=0;i<n;i++){
scanf("%d",b+i);
}
sort(b,b+n);
for (int i=0;i<n;i++){
ST tt = lim.top();
while (tt.index<b[i]){
lim.pop();
tt = lim.top();
}
if (tt.index>=b[i]){
ans+=tt.key;
// cout<<tt.key<<endl;
ans%=MOD;
ST temp;
temp.key = tt.key-(n+i+1);
temp.index = n+i;
lim.push(temp);
}
}
cout<<ans<<endl;
}
return 0;
}