班上有n个同学,现在有一些糖要分给他们,设第i个同学得到的糖为p[i],分糖必须满足条件:第i个同学要求第j个同学的糖不能超过自己k个,即p[j] - p[i] <= k,k >= 0。要求在满足这些条件的情况下,求出p[n] - p[1]的最大值。
由p[j] - p[i] <= k可得p[j] <= p[i] + k
在单源最短路径的算法中有一步是“若mindis[j] > mindis[i] + dis[i][j],则mindis[j] = mindis[i] + dis[i][j],这样就满足mindis[j] <= mindis[i] + dis[i][j]”。因此本题可以使用单源最短路径的算法来解决,对于“第i个同学要求第j个同学的糖不能超过自己k个,即p[j] - p[i] <= k,k >= 0”这个条件,建立一条边(i->j)=k,由于不含负权路径,因此建立完所有边之后以第1个同学为起点,采用Dijkstra+Heap求最短路径即可。除了Dijkstra也可以利用Spfa+Stack算法求解,但由于数据原因必须用Stack,如果用Queue则会超时。
#include<iostream>
#include<stack>
#include<algorithm>
using namespace std;
const int MAXN = 30005;
const int INF = 9999999;
const int MAXE = 150005;
typedef struct Edge
{
int v, w;
int next;
}Edge;
Edge edge[MAXE];
int dist[MAXN], adj[MAXN];
bool in_s[MAXN];
int size, n, m;
void add_edge(int u, int v, int w)
{
edge[size].v = v;
edge[size].w = w;
edge[size].next = adj[u];
adj[u] = size++;
}
void spfa()
{
int i, u, v, w;
int sta[MAXN];
int top=0;
memset(in_s, false, sizeof(in_s));
memset(dist, INF, sizeof(dist));
sta[++top] = 1;
dist[1] = 0;
in_s[1] = true;
while (top)
{
u = sta[top--];
in_s[u] = false;
for (i=adj[u]; i!=-1; i=edge[i].next)
{
v = edge[i].v;
w = edge[i].w;
if (dist[v]>dist[u]+w)
{
dist[v] = dist[u] + w;
if (!in_s[v])
{
in_s[v] = true;
sta[++top] = v;
}
}
}
}
}
int main()
{
int i, a, b, c;
scanf ("%d %d", &n, &m);
for (i=0; i<=n; i++) adj[i] = -1;
size = 0;
for (i=0; i<m; i++)
{
scanf("%d %d %d", &a, &b, &c);
add_edge(a, b, c);
}
spfa();
printf ("%d\n", dist[n]);
return 0;
}