pku3159 差分约束系统（SPFA + 栈）

Candies

小孩A认为小孩B比自己多出的最多不会超过c个糖果，也就是 B - A <= c，正好符合差分约束方程，就是A到B的边权w(A, B) = c;用 SPFA + 栈 能过。

这里有两种加边方式：

第一种：我以前用的，用这个超时了，因为每次加边都是将边夹在邻接表的最后面，需要一个查找时间，这题数据量大，自然就超时了。

代码

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define INF 0xfffffff
#define MM 150004
#define NN 30004
int edNum, N;
struct node{
int e, v;
}edge[MM];

int next[MM];
int dis[NN];
int root[NN];
int que[NN];
int mark[NN];
int stack[NN];

void add(int a, int b, int c){
int tmp;
edge[edNum].e = b;
edge[edNum].v = c;
next[edNum] = root[a];
root[a] = edNum++;
}

void Spfa()
{
int i, quNum, tmp, nxt, cur;
for (i = 1; i <= N; i++){
dis[i] = INF;
}
dis[1] = 0;
quNum = 0;
for (i = 1; i <= N; i++){
if (root[i] != -1){
que[quNum++] = i;
mark[i] = 1;
}
}

for (i = 0; i != quNum; i = (i + 1) % (N + 1)){
cur = que[i];
tmp = root[cur];
while (tmp != -1){
nxt = edge[tmp].e;
if (dis[nxt] > dis[cur] + edge[tmp].v){
dis[nxt] = dis[cur] + edge[tmp].v;
if (!mark[nxt]){
mark[nxt] = 1;
que[quNum] = nxt;
quNum = (quNum + 1) % (N + 1);
}
}
tmp = next[tmp];
}
mark[cur] = 0;
}
}

void Spfa1()
{
int i, top, tmp, nxt, cur;
for (i = 1; i <= N; i++){
dis[i] = INF;
}
dis[1] = 0;
top = 0;

for (i = 1; i <= N; i++){
if (root[i] != -1){
stack[++top] = i;
mark[i] = 1;
}
}

while (top){
cur = stack[top--];
tmp = root[cur];
mark[cur] = 0;
while (tmp != -1){
nxt = edge[tmp].e;
if (dis[nxt] > dis[cur] + edge[tmp].v){
dis[nxt] = dis[cur] + edge[tmp].v;
if (!mark[nxt]){
mark[nxt] = 1;
stack[++top] = nxt;
}
}
tmp = next[tmp];
}

}
}

int main()
{
int M, a, b, c, i;
scanf("%d%d", &N, &M);
edNum = 0;
for (i = 0; i <= N; i++){
root[i] = -1;
mark[i] = 0;
}
while (M--){
scanf("%d%d%d", &a, &b, &c);
add(a, b, c);
}
Spfa1();
printf("%d\n", dis
);
//system("pause");
return 0;
}