POJ - 3159 Candies（差分约束SPFA）

POJ - 3159 Candies

差分约束题，若 b-a<=c，那么从 a 到 b 有一条长为 c 的边，

所有的 c 都是正数，因此不会存在负环，求 1 号到其他所有点单源最短路即可得到答案，用queue会超时，只能stack。

#include<cstdio>
#include<cstring>
#include<stack>
using namespace std;
struct node
{
    int v,w,next;
} edge[150010];
int head[30001],dis[30001],vis[30001],cnt,n,m;
void init()
{
	memset(head,-1,sizeof head);
    cnt=0;
} 
void add(int u,int v,int w)
{
    edge[cnt].v=v;
    edge[cnt].w=w;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}
void spfa(int s)
{
	memset(dis,0x3f,sizeof dis);
	memset(vis,0,sizeof vis);
    dis[s]=0;
    vis[s]=1;
    stack<int>Q;
    Q.push(s);
    while(!Q.empty())
    {
        int u=Q.top();
        Q.pop();
        vis[u]=0;
        for(int i=head[u];i!=-1;i=edge[i].next)
        {
            int v=edge[i].v;
            if(dis[v]>dis[u]+edge[i].w)
            {
                dis[v]=dis[u]+edge[i].w;
                if(!vis[v])
                {
                    vis[v]=1;
                    Q.push(v);
                }
            }
        }
    }
}
int main()
{
    int i,j,u,v,w;
    while(~scanf("%d%d",&n,&m))
    {
        init();
        for(i=0;i<m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
        }
        spfa(1);
        printf("%d\n",dis[n]);
    }
    return 0;
}