HDU 6047 Maximum Sequence 数论 贪心

来源:互联网 发布:经济增加值算法 编辑:程序博客网 时间:2024/06/14 17:08

    题意是说有两个数组,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;}