题目来源
题意:给两个长度为n的序列a, b,现要求max(sum(a[n + 1] + … + a[2n])),其中a[i] = max(a[b[k]] … a[i - 1]),其中b[k]为b[1 … n]中的一个,每个a[i]选择的b[k]不能重复
常规方法是用线段数来做,不过如果有比较巧的方法也是可以的。我找到了一种别人写的比较巧秒的方法来解这道题,大家不妨看看。
借鉴来源
代码:
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn=250000+100;
const int mod=1e9+7;
int a[maxn],b[maxn];
int main()
{
int n;
while(~scanf("%d",&n))
{
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
a[i]=a[i]-(i+1);//在处理的时候就减去自身的序号
}
for(int i=1;i<=n;i++)
scanf("%d",&b[i]);
for(int i=n-2;i>=0;i--)
a[i]=max(a[i],a[i+1]);//对a数组进行预处理,后面就不用一个个找最大值了,巧就巧在这个地方了
sort(b,b+n);//将b数组排序,贪心思想,为了使ans和最大
long long ans=0;//数据类型注意
int most=-1;//保存从n开始的a[i]-i的最大值
for(int i=1;i<=n;i++)
{
int kk=max(most,a[b[i]-1]);//前面对a数组的处理使此步简便,只需比较most与a[b[i]-1],因为在a[b[i]-1]至a[n-1]之间a[b[i]-1]值最大
ans=(ans+kk)%mod;
most=max(most,kk-n-i);
}
printf("%lld\n",ans);
}
return 0;
}