POJ 3159 Candies [差分约束系统 SPFA+STACK]

题意：

有一包糖果要分给n个学生，对于输入数据u v w代表学生v得到的糖果数比学生u得到的糖果数 至多可以多出w个。

问满足条件的情况下，学生n比学生1至多可以多出多少个糖果。

思路：

很显然的差分约束方程求解问题。

先根据u v w构造差分方程：d[ v ]-d[ u ]<=w;

再用该方程建图：addedge(u,v,w):建立一条从u出发到v的权值为w的单向边。

然后进行spfa找出最短路即可。

但是这题卡就卡在数据比较怪，SPFA如果用队列则超时，用STACK才能AC。

#include<iostream>
#include<vector>
#include<string>
#include<queue>
#include<algorithm>
#define llong long long
#define Min(a,b) (a<b?a:b)
#define Max(a,b) (a>b?a:b)
#define Abs(a) ((a)>0?(a):-(a))
#define Mod(a,b) (((a)-1+(b))%(b)+1)
using namespace std;
const int N=30005;
const int M=150005;
const int inf=1000000000;
int n,m;
struct
{
	int v,next,w;
}edge[M];
int edgehead[N];
bool vis[N];
int d[N];
int k;
int stack[N];
int sp;
void addedge(int u,int v,int w)
{
	edge[k].v=v;
	edge[k].w=w;
	edge[k].next=edgehead[u];
	edgehead[u]=k++;
}
void spfa()
{
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=n;i++)
		d[i]=inf;
	d[1]=0;
	vis[1]=true;
	sp=0;
	stack[sp++]=1;
	while(sp)
	{
		int now=stack[--sp];
		vis[now]=false;
		for(int i=edgehead[now];i;i=edge[i].next)
		{
			int v=edge[i].v;
			int w=edge[i].w;
			if(d[v]>d[now]+w)
			{
				d[v]=d[now]+w;
				if(!vis[v])
				{
					vis[v]=true;
					stack[sp++]=v;
				}
			}
		}
	}
	printf("%d\n",d[n]);
}
int main()
{
	scanf("%d%d",&n,&m);
	int u,v,w;
	k=1;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&u,&v,&w);
		addedge(u,v,w);
	}
	spfa();
	return 0;
}

以上的spfa+stack 顺利AC了。

以下的dijstra+heap始终WA，不明白为什么，根据discuss说用队列返回的WA实际上很可能是TLE。

查了不少博客，有用priority_queue AC的，但是他们的priority_queue和我的

priority_queue<int,vector<int>,cmp> que;

不一样，或者是我的queue耗时比较多吧。

蛋疼了，先到这。

#include<iostream>
#include<vector>
#include<string>
#include<queue>
#include<algorithm>
#define llong long long
#define Min(a,b) (a<b?a:b)
#define Max(a,b) (a>b?a:b)
#define Abs(a) ((a)>0?(a):-(a))
#define Mod(a,b) (((a)-1+(b))%(b)+1)
using namespace std;
const int N=30005;
const int M=150005;
const int inf=1000000000;
int n,m;
struct
{
	int v,next,w;
}edge[M];
int edgehead[N];
bool vis[N];
int d[N];
int k;
void addedge(int u,int v,int w)
{
	edge[k].v=v;
	edge[k].w=w;
	edge[k].next=edgehead[u];
	edgehead[u]=k++;
}
struct cmp
{
	bool operator()(const int a,const int b)
	{
		return d[a]>d[b];
	}
};
void dijstra()
{
	priority_queue<int,vector<int>,cmp> que;
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=n;i++)
		d[i]=inf;
	d[1]=0;
	que.push(1);
	while(!que.empty())
	{
		int now=que.top();
		que.pop();
		if(vis[now])
			continue;
		vis[now]=true;
		for(int i=edgehead[now];i;i=edge[i].next)
		{
			int v=edge[i].v;
			int w=edge[i].w;
			if(!vis[v]&&d[v]>d[now]+w)
			{
				d[v]=d[now]+w;
				que.push(v);
			}
		}
	}
	printf("%d\n",d[n]);
}
int main()
{
	scanf("%d%d",&n,&m);
	int u,v,w;
	k=1;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&u,&v,&w);
		addedge(u,v,w);
	}
	dijstra();
	return 0;
}