poj 3159 Candies （spfa+栈）

最新推荐文章于 2022-02-21 09:22:11 发布

原创最新推荐文章于 2022-02-21 09:22:11 发布 · 441 阅读

0 ·

CC 4.0 BY-SA版权

图论专栏收录该内容

87 篇文章

订阅专栏

本文介绍了一种解决差分约束问题的方法，并通过对比队列和栈实现的SPFA算法，展示了在存在负权边的情况下，使用栈实现的SPFA算法能够更有效地解决问题。

题意：简单的说就是给出m给 si 与sj的关系，其中sj的糖数不能比si的多c个，即sj-si <= c 最后求 flymouse（n）最多能比soonpy（1）多多少糖，

思路：话说这是个差分约束问题其实就是求最短路问题只要求出dis[n]就行了原先用队列实现spfa 但超时了，因为是每个点入队列超过30000 次后改为栈实现 ps：有负环时，用栈比队列快，（原因是啥，我也不知道）

栈实现：
#include <stdio.h>
#include <string.h>
#define VM 30005
#define EM 150005
#define inf 0x3f
int head[VM],ep;
struct edge
{
    int v,w,next;
}e[EM];

void addedge(int cu,int cv,int cw)
{
    ep ++;
    e[ep].v = cv;
    e[ep].w = cw;
    e[ep].next = head[cu];
    head[cu] = ep;
}

void spfa (int n)
{
    int vis[VM],stack[VM],dis[VM];
    memset (vis,0,sizeof(vis));
    memset (dis,0x3f,sizeof(dis));
    dis[1] = 0;
    vis[1] = 1;
    int top = 1;
    stack[0] = 1;
    while (top)
    {
        int u = stack[--top];
        vis[u] = 0;
        for (int i = head[u];i != -1;i = e[i].next)
        {
            int v = e[i].v;
            if (dis[v] > dis[u] + e[i].w)
            {
                dis[v] = dis[u] + e[i].w;
                if (!vis[v])
                {
                    vis[v] = 1;
                    stack[top++] = v;
                }
            }
        }
    }

    printf ("%d\n",dis[n]);
}
int main ()
{
    int n,m,v1,v2,cost;
    ep = 0;
    memset (head,-1,sizeof(head));
    scanf ("%d%d",&n,&m);
    while (m--)
    {
        scanf ("%d%d%d",&v1,&v2,&cost);
        addedge (v1,v2,cost);
    }
    spfa (n);
    return 0;
}

队列实现：
#include <stdio.h>
#include <string.h>
#define VM 300005
#define EM 1500005
#define inf 0x3f
int head[VM],ep;
struct edge
{
    int v,w,next;
}e[EM];

void addedge(int cu,int cv,int cw)
{
    ep ++;
    e[ep].v = cv;
    e[ep].w = cw;
    e[ep].next = head[cu];
    head[cu] = ep;
}

void spfa (int n)
{
    int vis[VM],queue[VM],dis[VM];
    memset (vis,0,sizeof(vis));
    memset (dis,0x3f,sizeof(dis));
    dis[1] = 0;
    int front = -1,rear = 0;
    queue[rear++] = 1;
    vis[1] = 1;
    while (front != rear)
    {
        front = (front+1)%(n+1);
        int u = queue[front];
        vis[u] = 0;
        for (int i = head[u];i != -1;i = e[i].next)
        {
            if (dis[e[i].v] > dis[u]+e[i].w)
               {
                   dis[e[i].v] = dis[u]+e[i].w;
                   if (!vis[e[i].v])
                   {
                       vis[e[i].v] = 1;
                       queue[rear] = e[i].v;
                       rear = (rear+1)%(n+1);
                   }
               }
        }
    }
    printf ("%d\n",dis[n]);
}
int main ()
{
    int n,m,v1,v2,cost;
    ep = 0;
    memset (head,-1,sizeof(head));
    scanf ("%d%d",&n,&m);
    while (m--)
    {
        scanf ("%d%d%d",&v1,&v2,&cost);
        addedge (v1,v2,cost);
    }
    spfa (n);
    return 0;
}