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;
}