联考20200521 T2 river


分析:
每个点每个模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);
}

原文地址:https://www.cnblogs.com/Darknesses/p/12931342.html