題目來源
題意:給兩個長度爲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;
}