2017 Multi-University Training Contest

来源:互联网 发布:c语言编译器安卓 编辑:程序博客网 时间:2024/06/07 03:17

这个题目简单的贪心即可(后面附上官方证明),实现起来要注意ai数组要不断更新,b数组排序后从头往后依次使用即可。

官方证明:预处理:a_i -= i ,易证明从最小的b开始选每次选最大的一定可以使结果最大。 证明思路:如果条件改为a_i<=max{a_j-j|b_k<=j<=n},那么b的顺序与最后的结果无关。条件改回来后,由于每次要计算一个数的最大值时都有a_(n+1)...a_(i-1)在范围中,所以每次只需让a_i - i尽可能大,那么就把大的数尽早用上,每次一定考虑尽量多的数字,这样取得的数字就尽可能的大。 所以说每次就是求区间最值,加在答案上。由于贪心的思路,每次要求的区间的下界是单调不降的,故可以用单调队列优化到O(n)的复杂度。 由于1 ≤ b_i ≤ n,对b排序可以用哈希排序(桶排序)完成。进一步观察,可以发现这样贪心时 a_(n+1)...a_i 其实是单调不增的,所以并不需要每次求区间最值了,选第一个数时就选最大的,后面的选择顺序与最终结果无关了。

        #include<iostream>        #include<cstdio>        #include<algorithm>        using namespace std;                const int maxx=250010;        int a[2*maxx],b[2*maxx];int as[2*maxx];                int main(){          int n;        while(scanf("%d",&n)==1)        {        for(int i=1;i<=n;i++)        {        scanf("%d",&a[i]);        as[i]=a[i]-i;}for(int i=1;i<=n;i++)        {        scanf("%d",&b[i]);}int Max=as[n];for(int i=n;i>=1;i--){if(as[i]>Max){Max=as[i];}else{as[i]=Max;}}sort(b+1,b+n+1);int pos=1;long long ans=0;for(int i=n+1;i<=2*n;i++){a[i]=as[b[pos++]];//cout<<"b[pos="<<b[pos]<<endl;//cout<<"ai="<<a[i]<<endl;ans+=a[i];ans%=1000000007;as[i]=a[i]-i;int j=i;while(j>=1&&as[j]>as[j-1]){as[j-1]=as[j];j--;}}//cout<<ans;printf("%lld\n",ans);}}


原创粉丝点击