POJ 3159 Candies 差分约束

题目连接<http://poj.org/problem?id=3159>

题意：

给n个小朋友发糖，如果小朋友a发现小朋友b比他多c个糖，a就不开心。要使所有小朋友都开心，问第一个小朋友和最后一个小朋友获得的糖相差最大为多少。

题解：

一、SPFA卡队列，用栈能过。

      不需要判断负环的情况下，用栈更快，也会更省空间。

#include <iostream>
#include <cstdio>
#include <map>
#include <queue>
#include <stack>
#include <string.h>
#include <string>
using namespace std;
typedef long long LL;
int n,m;
struct Edge{
    int to,val,next;
    Edge(int to=0,int val=0,int next=0):to(to),val(val),next(next){}
}e[150005];
int head[150005],now=-1;
void add(int a,int b,int c){
    e[++now]=Edge(b,c,head[a]),head[a]=now;
}
int vis[30005],dis[30005],num[30005];
void spfa(){
    memset(vis,0,sizeof(vis));
    memset(dis,125,sizeof(dis));
    memset(num,0,sizeof(num));
    stack<int>q;
    q.push(1);vis[1]=num[1]=1;dis[1]=0;
    while(!q.empty()){
        int u=q.top();
        q.pop();vis[u]=0;
        for(int i=head[u];~i;i=e[i].next){
            int v=e[i].to;
            if(dis[v]-dis[u]>e[i].val){
                dis[v]=dis[u]+e[i].val;
                if(!vis[v]){
                    q.push(v);
                    vis[v]=1;
                    num[v]++;
                }
            }
        }
    }
}
int main(){
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&m);
    int a,b,c;
    while(m--){
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    spfa();
    printf("%d\n",dis[n]);
}

二、由于题目没有负权边，也可以用Dijstra求最短路。

#include <iostream>
#include <cstdio>
#include <map>
#include <queue>
#include <string.h>
#include <string>
using namespace std;
typedef long long LL;
int n,m;
struct Edge{
    int to,val,next;
    Edge(int to=0,int val=0,int next=0):to(to),val(val),next(next){}
}e[150005];
int head[150005],now=-1;
void add(int a,int b,int c){
    e[++now]=Edge(b,c,head[a]),head[a]=now;
}
struct Node{
    int id,val;
    Node(int id,int val):id(id),val(val){}
    bool operator<(const Node a)const{
        return val>a.val;
    }
};
int vis[30005],dis[30005];
void dij(){
    memset(dis,125,sizeof(dis));dis[1]=0;
    priority_queue<Node>q;
    q.push(Node(1,0));
    while(!q.empty()){
        Node t=q.top();q.pop();
        int u=t.id;
        vis[u]=1;
        for(int i=head[u];~i;i=e[i].next){
            int v=e[i].to;
            if(!vis[v]&&dis[v]-dis[u]>e[i].val){
                dis[v]=dis[u]+e[i].val;
                q.push(Node(v,dis[v]));
            }
        }
    }
}
int main(){
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&m);
    int a,b,c;
    while(m--){
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    dij();
    printf("%d\n",dis[n]);
}