分析:
每个点每个模m意义下时刻的最短天数是独立的,而且最优选择是不变的
考虑在每一个模m意义下时刻的最优选择,设为(f[i])
于是(f[i]=min(f[i],f[(i+1)%m]+1))
然后可以(O(n))暴力做
发现中途转移会形成一个环,把环找出来之后直接(O(1))算
前后一些零碎的转移暴力做就好了,复杂度(O(m))
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<iostream>
#include<map>
#include<string>
#define maxn 1000005
using namespace std;
inline long long getint()
{
long long num=0,flag=1;char c;
while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
while(c>='0'&&c<='9')num=num*10+c-48,c=getchar();
return num*flag;
}
int n,m;
int a[maxn];
long long ans;
int tim[maxn];
long long sum[maxn];
int main()
{
n=getint(),m=getint();
for(int i=0;i<m;i++)a[i]=getint();
for(int i=m-2;i>=0;i--)a[i]=min(a[i],a[i+1]+1);
a[m-1]=min(a[m-1],a[0]+1);
for(int i=m-2;i>=0;i--)a[i]=min(a[i],a[i+1]+1);
int now=0,i;
for(i=1;i<=n;i++)
{
if(tim[now])
{
int tmp=i-tim[now];
ans+=(sum[i-1]-sum[tim[now]-1])*((n-i+1)/tmp);
i+=(n-i+1)/tmp*tmp;break;
}
tim[now]=i,sum[i]=sum[i-1]+a[now];
ans+=a[now],now=(now+a[now])%m;
}
for(;i<=n;i++)ans+=a[now],now=(now+a[now])%m;
printf("%lld
",ans);
}