这道题肯定第一眼都想到了n 2 2 的暴力,就是dp[i]=max(dp[j-1]+abs(i-j-a[j])),那么我们可不可以再想的深一点呢?
注意到我们的转移每次都是从前面的状态经过最小的abs(i-j-a[j])转移到了最终状态,也就是说我们的目标就是从dp[0]经过最小的代价‘走’到了dp[n]。那这看上去不是最短路么?
emmmmmm…看上去好像是那么回事,不过如果按照 dp 的方式建边,边数显然太多,所以我们需要规划一种建边的方法,使得边数控制在 nlogn 级别。
那我们就会有一个想法:简化建边过程,比如,我们可以把从一个点连出的很多边简化成 i~i+x+1 的一条边和i+x+1向左向右连出的很多代价为 1 的边(因为显然每多覆盖或者少覆盖一个都要多花 1 的代价),如下图,我们可以把灰色的边替换成红色的边:
因为边数太多,我们肯定不能用普通方法来保存每个节点的边,所以我们用一个 las[i] 表示以第 i 条边的起点扩展的上一条边的编号。因为一定不会存在先扩展了长边再扩展了短边的情况(长边指的是坐标跨度大,不是代价大),所以我们每次枚举一定是从长到短来枚举的。
有了这一步后,我们就可以愉快的 dijkstra 了:
#include<queue>
#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
#define PAIR pair<int,int>
int cnt,n,d[1000005],id[1000005],las[2000005],to[2000005],v[2000005],vis[1000005],vis2[1000005];
priority_queue<PAIR,vector<PAIR>,greater<PAIR> > q;
void addedge(int s,int t,int w){to[++cnt]=t,v[cnt]=w,las[cnt]=id[s],id[s]=cnt;}
int main()
{
int x;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&x);
for(int j=i+1;j<=i+x+1 && j<=n && !vis[j];j++)
vis[j]=1,addedge(j,j-1,1);
for(int j=i+x+1;j<=n && !vis2[j];j++)
vis2[j]=1,addedge(j,j+1,1);
if(i+x+1<=n+1) addedge(i,i+x+1,0);
else addedge(i,n+1,i+x-n);
}
memset(d,0x3f,sizeof(d));
d[1]=0,q.push(make_pair(0,1));
while(!q.empty())
{
PAIR NOW=q.top();q.pop();
int dis=NOW.first,now=NOW.second;
for(int i=id[now];i>0;i=las[i])
{
int nex=to[i];
if(d[now]+v[i]<d[nex])
d[nex]=d[now]+v[i],q.push(make_pair(d[nex],nex));
}
}
printf("%d",d[n+1]);
return 0;
}