POJ - 3159 Candies(差分约束SPFA)(未完成)

POJ - 3159 Candies

#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <stack>
#define inf 0x3f3f3f3f
using namespace std;
struct node
{
    int v,w,next;
} edge[150010];
int head[30001],dis[30001],vis[30001],cnt,n,m,s;
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()
{
    for(int i=1;i<=n;i++)
    {
        dis[i]=inf;
        vis[i]=0;
    }
    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))
    {
        s=1;
        for(i=1;i<=n;i++)
        head[i]=-1;
        cnt=0;
        for(i=0; i<m; i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
        }
        spfa();
        printf("%d\n",dis[n]);
    }
    return 0;
}
上一篇:【AtCoder】C - Colorful Candies map+双指针


下一篇:21天活动Zyf打卡第六天Promise9913,分糖果