Codeforces Round #645 (Div. 2) D补题

题目大意：
n个月，每个月有d[i]天，这个月每天可以获得的拥抱数从1到d[i]，共有x天，询问在连续的x天内，一共最大可以获得多少拥抱，这个月份可以延伸到第二年
.
思路：
1.既然可以延伸到第二年，而且是x是小于每个月的天数之和的，所以x的范围肯定不超过一年，那只需要多开一倍的空间模拟第二年。
2.通过观察可以发现，要想拥抱数最多，x天的最后一天一定是月末结尾，假设当前在月末，按照一般情况来说，这x天的区间往前移一天和往后移一天，获得的拥抱数都会变小，所以只需要枚举月末为结尾的时候即可。
3.求一个天数的前缀和，是用来判断到当前月是否可以满足了x天，再求一个获得的拥抱数的前缀和，用来求总的整月内获得的拥抱数，最后也要加上剩下天数（不足整月）的拥抱数。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<map>
using namespace std;
typedef long long ll;
const int N=2e5+5;
int d[2*N];
ll h[2*N];//hug数
ll t[2*N];//天数
ll ans;
int main(){
int n;
ll x;
cin>>n>>x;

for(int i=1;i<=n;i++){
    scanf("%d",&d[i]);
    d[i+n]=d[i];
}

for(ll i=1;i<=2*n;i++){
    t[i]=t[i-1]+d[i];
    h[i]=h[i-1]+(ll)(1+d[i])*d[i]/2;
}

for(ll i=1;i<=2*n;i++){
    if(t[i]>=x){
        ll p=lower_bound(t+1,t+2*n+1,t[i]-x)-t;
        ll res=x-(t[i]-t[p]);
        ll last=((d[p]-res+1)+d[p])*res/2;
        ans=max(ans,last+h[i]-h[p]);
    }
}
cout<<ans<<endl;

return 0;}