题意是说有两个数组,a和b,然后让你扩展a数组后面的n位,扩展的方法是先在b数组中随意挑出一个数(这个数只能挑一次),之后 ai ≤max{ aj -j│ bk ≤j<i},然后要求出最后n位a数组的和最大的情况。
为了让a数组后面和最大,这里肯定不能小于max,选的一定是这个区间内最大的一个数,先把 aj -j都处理出来,然后再去赋值。在挑选b的时候,因为越往后越是减出来的数字,所以为了保证得到最大的,b数组必须从最小的一个数开始拿起,这样在处理的时候讲b数组sort排序处理一下,后面就不用管b了。
然后昨天想到这后面再怎么找a的方法没想出来,海强想出了用线段树的方法来查询区间最大值,结果优化还是不够,TLE了。最后师弟给提供出了一种思路,才把这个题A了。
在对于数组a处理的时候,先从后往前去读,如果前面的数字比后面的数字要大就不用动,如果相比要小,就把前面的数字改成大的那个。这样一遍处理下来,就能得到每个区间最大的那个值。这样在最后找区间最大值的时候,直接找a数组开始的那一位就可以了。
(PS:这题一开始我写的代码,数组开小了,忘了应该是两倍,只开了250000,所以一直返回TLE,但是没返回RE所以一直没考虑数组的问题,最后实在改不出来了,改了一下数组大小结果A了.....)
下面AC代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int a[500005];
int b[500005];
const int mod=1000000007;
int main()
{
int n;
int i,j,k;
int ans;
while(scanf("%d",&n)!=EOF)
{
ans=0;
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
a[i]=a[i]-i;
}
for(i=n-1;i>=1;i--)
{
if(a[i]<a[i+1])
a[i]=a[i+1];
}
for(i=1;i<=n;i++)
{
scanf("%d",&b[i]);
}
sort(b+1,b+n+1);
k=n+1;
for(i=1;i<=n;i++)
{
ans+=a[b[i]];
a[k]=a[b[i]]-k;
for(j=k-1;j>=1;j--)
{
if(a[k]>a[j])
a[j]=a[k];
else
break;
}
/*for(j=1;j<=k;j++)
{
cout<<"a["<<j<<"] = "<<a[j]<<endl;
}
cout<<"------------"<<endl;*/
k++;
ans%=mod;
}
cout<<ans<<endl;
}
return 0;
}