poj3159 Candies SPFA

时间:2023-02-07 14:28:06

题目链接:http://poj.org/problem?id=3159

题目很容易理解

就是简单的SPFA算法应用

刚开始用STL里的队列超时了,自己写了个栈,果断过,看来有时候栈还是快啊。。。。

代码:

 #include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#define INF 0x3f3f3f3f
#define maxn 30010
#define max_e 150010
using namespace std;
class node
{
public:
int to;
int w;
int next;
};
node edge[max_e];
int head[maxn];
int inq[maxn];
int dis[maxn];
int tol;
int n,m;
int Q[maxn];
void add(int u,int v,int w)
{
edge[tol].to=v;
edge[tol].w=w;
edge[tol].next=head[u];
head[u]=tol++;
};
void SPFA()
{
int top=;
for(int i=;i<=n;i++)
{
dis[i]=INF;
inq[i]=;
}
dis[]=;
inq[]++;
Q[++top]=;
while(top!=)
{
int u=Q[top];
top--;
inq[u]--;
int tmp=head[u];
while(tmp!=-)
{
int v=edge[tmp].to;
int w=edge[tmp].w;
if(dis[v]>dis[u]+w)
{
dis[v]=dis[u]+w;
if(inq[v]==)
{
Q[++top]=v;
inq[v]++;
}
}
tmp=edge[tmp].next;
} }
}
int main()
{
int a,b,c;
while(scanf("%d%d",&n,&m)!=EOF)
{
tol=;
memset(head,-,sizeof(head));
for(int i=;i<=m;i++)
{
scanf("%d%d%d",&a,&b,&c);
add(a,b,c);
}
SPFA();
cout<<dis[n]<<endl;
}
return ;
}