【vijos】【spfa最短路】想越狱的小杉

描述

小杉看了看自己的纹身，明白了整个管道网是由N个小房间和若干小房间之间的单向的管道组成的。

小房间编号为不超过N的正整数。

对于某个管道，小杉只能在人品不超过一定程度时通过。

小杉一开始在房间1，现在小杉想知道，每个小房间他最多能够以人品多少的状态到达。

注意，小杉的人品在出发以后是不会改变的。

格式

输入格式

每组测试数据的

第一行有一个正整数N(1<=N<=2000)。

接下来若干行描述管道，每行三个正整数A,B,R(1<=A,B<=N)，表示A房间有一条到达B房间的管道，且小杉的人品不超过R时可以通过（注意从B房间不可由此管道到达A房间，即管道是单向的)

整个输入数据以一行0 0 0结束

特别地，对于30%的数据，有N<=100

输出格式

对每组测试数据输出N-1行，分别表示对于2到N号的小房间，小杉最多能够以人品多少的状态到达。

代码

SPFA变式…

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#define maxn 2000+100
#define INF 1000000000-1
using namespace std;

int n,a,b,r;
int edge[maxn][maxn],d[maxn],cnt[maxn];
bool vis[maxn],flag=true;
queue<int> q;
//d[i]代表从出发点到i点的最大RP
//更新d[i]当d[cur]>d[i] 且edge[cur][i]>d[i]
//d[i]初始为0
void spfa(){
q.push(1); d[1]=INF;
while(!q.empty()&&flag){
int cur=q.front(); q.pop(); vis[cur]=false;
for(int i=1;i<=n;i++){
if(edge[cur][i]!=-1){
if(d[cur]>d[i]&&edge[cur][i]>d[i]){
d[i]=min(d[cur],edge[cur][i]);
if(!vis[i]){
q.push(i); cnt[i]++;
if(cnt[i]>n){
flag=false; break;
}
}
}
}
}
}
return;
}

int main(){
//freopen("in.txt","r",stdin);
scanf("%d",&n);
memset(edge,-1,sizeof(edge));
memset(vis,0,sizeof(vis));
memset(cnt,0,sizeof(cnt));
memset(d,0,sizeof(d));
while(scanf("%d%d%d",&a,&b,&r)!=EOF){
if(a==b&&b==r&&a==0) break;
edge[a][b]=r;
}
spfa();
for(int i=2;i<=n;i++) printf("%d\n",d[i]);
return 0;
}