题目大意:
题解:
我们考虑最后一个位置是从哪里接收到的消息。 发出消息给最后
一个位置的地方必然位于[n−an,n−1]。然后考虑这个位置是什么。
设S=min(k-ak),n-an≤k≤n−1,对应的最小的k是k0。显然,除非
第一个位置可以直接发消息到最后一个位置,那么消息从k0传递过
来一定不会比从其他位置传递过来更劣。
因此,我们得到了结论:对每个位置n,除非它是直接由起点接
受到的消息,否则向它发送消息的前驱是固定的。这个前驱就是满足
n−an≤k≤n−1且使得k−ak最小的k。
我们可以利用ST表,在O(nlogn)内找到每个点的前驱。
然后通过倍长序列,显然可以用倍增在O(nlogn)内找到以每个
点为起点是传递信息花费的时间。
#include<cstdio>
#include<cstdlib>
using namespace std;
typedef long long ll;
inline char nc()
{
static char buf[100000],*p1=buf,*p2=buf;
if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
return *p1++;
}
inline void read(int &x)
{
char c=nc(),b=1;
for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}
const int N=250005;
const int K=21;
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
namespace ST{
int st[N<<1][K],Log[N<<1];
inline void Pre(int n,int *a){
for (int i=2;i<=n;i++) Log[i]=Log[i>>1]+1;
for (int i=1;i<=n;i++) st[i][0]=a[i];
for (int k=1;k<K;k++)
for (int i=1;i<=n;i++)
{
st[i][k]=st[i][k-1];
if (i+(1<<(k-1))<=n)
st[i][k]=Min(st[i][k],st[i+(1<<(k-1))][k-1]);
}
}
inline int Query(int l,int r){
int t=Log[r-l+1];
return Min(st[l][t],st[r-(1<<t)+1][t]);
}
}
int n,a[N<<1];
ll ans=0;
int main()
{
using namespace ST;
int l,r,ret,nxt,flag=0;
freopen("sequence.in","r",stdin);
freopen("sequence.out","w",stdout);
read(n);
for (int i=1;i<=n;i++)
{
read(a[i]),a[n+i]=a[i];
if (a[i]!=1) flag=1;
}
if (!flag) return printf("%lld\n",(ll)n*(n-1)),0;
for (int i=1;i<=n+n;i++) a[i]=Max(i-a[i],1);
Pre(n<<1,a);
for (int i=1;i<=n;i++)
{
l=n+i-1,r=n+i-1,ret=0;
while (l>i)
{
nxt=Query(l,r),r=l-1,l=nxt;
ret++;
}
ans+=ret;
}
printf("%lld\n",ans);
return 0;
}