【BZOJ】【P2259】【Oibh】【新型计算机】【题解】【最短路】

传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=2259

水最短路,好像成功卡掉了spfa = =、

建图见代码

Code:

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
int n;
int a[maxn];
int d[maxn];
int vis[maxn];
struct edge{int u,v,w;};
vector<edge>G[maxn];
void add(int u,int v,int w){G[u].push_back((edge){u,v,w});}
typedef pair<int,int> pi;
priority_queue<pi,vector<pi>,greater<pi> >q;
int spfa(){
	fill(d+1,d+2+n,1e9+10);
	d[1]=0;q.push(pi(0,1));
	while(!q.empty()){
		int u=q.top().second;vis[u]=0;q.pop();
		for(int i=0;i<G[u].size();i++){
			edge e=G[u][i];
			if(d[u]+e.w<d[e.v]){
				d[e.v]=d[u]+e.w;
				q.push(pi(d[e.v],e.v));
			}
		}
	}return d[n+1];
}
int pre[maxn],nxt[maxn];
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		int x;scanf("%d",&x);
		for(int j=i+1;j<=min(i+x+1,n)&&!pre[j];j++)pre[j]=1,add(j,j-1,1);
		for(int j=i+x+1;j<=n&&!nxt[j];j++)nxt[j]=1,add(j,j+1,1);		
		if(i+x+1<=n+1)
			add(i,i+x+1,0);
		else add(i,n+1,i+x-n);
	}cout<<spfa()<<endl;
	return 0;
}