首页 > 代码库 > POJ 3159 Candies(差分约束系统)

POJ 3159 Candies(差分约束系统)

题目地址:POJ 3159

第一发差分约束的题。。就当作最短路来做了。。。直接建图+spfa。。不过我用的spfa+slf优化都超时。。看了讨论区里的。。把spfa换成栈就过了。。。

代码如下:

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include <set>
#include <algorithm>

using namespace std;
const int INF=0x3f3f3f3f;
int d[40000], vis[40000], source, sink, head[40000], cnt, q[400000];
struct node
{
    int u, v, w, next;
} edge[200000];
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()
{
    memset(vis,0,sizeof(vis));
    memset(d,INF,sizeof(d));
    d[source]=0;
    int top=0;
    q[++top]=source;
    while(top)
    {
        int u=q[top--];
        vis[u]=0;
        for(int i=head[u]; i!=-1; i=edge[i].next)
        {
            int v=edge[i].v;
            if(d[v]>d[u]+edge[i].w)
            {
                d[v]=d[u]+edge[i].w;
                if(!vis[v])
                {
                    vis[v]=1;
                    q[++top]=v;
                }
            }
        }
    }
    printf("%d\n",d[sink]);
}
int main()
{
    int n, m, i, u, v, w;
    scanf("%d%d",&n,&m);
    source=1;
    sink=n;
    memset(head,-1,sizeof(head));
    cnt=0;
    while(m--)
    {
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
    }
    spfa();
    return 0;
}


POJ 3159 Candies(差分约束系统)