CF传送门
洛谷传送门
解题思路
很显然的结论是,要尽可能多的重复。
但是也不能每次都重复:
假设第 i 天 (m_i=4),那么到第i天至少要保证已经有四条线,也就是说前 i-1 天最多重复 i-5 条。
所以我们用一个单调栈维护一个最长的 (i-m[i]) 上升后缀,这也就求出了每一段区间的重复的条数。
最后统计答案即可。
细节还是很多的。
AC代码
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<iomanip>
#include<ctime>
#include<stack>
using namespace std;
const int maxn=100005;
int n,a[maxn],b[maxn],c[maxn],num,res,cnt;
long long ans;
stack<int> s;
int main()
{
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
for(int i=1;i<=n;i++){
if(s.empty()){
s.push(i);
continue;
}
while((!s.empty())&&(s.top()-a[s.top()]>=i-a[i])) s.pop();
s.push(i);
}
while(!s.empty()){
b[++cnt]=s.top();
c[cnt]=b[cnt]-a[b[cnt]]-1;
s.pop();
}
res=cnt;
for(int i=1;i<=n;i++){
if(res>1&&i>b[res]) res--;
if(i-num>c[res]) num++;
ans=ans+(num-a[i]-1);
}
cout<<ans;
return 0;
}