D. The Best Vacation
题意:一年有 n 个月,每个月分别有 d[i] 天,每天的贡献分别是当天的日期(例:X月的第Y天的贡献是Y),求长度为 x 的连续段的最大贡献值
思路:可以看出来一年中每一天的贡献值是呈以下这种曲线的:
所以如果我们想要得到的贡献最大的话,就要从峰顶开始从后往前走,所以我们可以分别以每一个峰顶为起点从后往前走找最大值,这样我们预处理一个递增数列就可以暴力做题了
代码附上:
#pragma GCC optimize("Ofast","inline","-ffast-math")
#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<bits/stdc++.h>
using namespace std;
using ll = long long ;
const int N = 1e6+10;
ll pre[N],d[N];
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
for(int i=1; i<N; ++i)
pre[i]=pre[i-1]+i;
ll n,m,ans=0,now=0,sum=0;
cin>>n>>m;
for(int i=0; i<n; ++i)
{
cin>>d[i];
d[i+n]=d[i];
}
for(int i=2*n-1,r=2*n-1; i>=0&&r>=n; --i)
{
if(now+d[i]<=m)
{
now+=d[i];
sum+=pre[d[i]];
}
else
{
ans=max(ans,sum+pre[d[i]]-pre[d[i]+now-m]);
now=max(0ll,now-d[r]);
sum=max(0ll,sum-pre[d[r]]);
r--;
if(r>=i)
i++;
}
}
cout<<ans;
return 0;
}