题目连接<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]); }