poj3159

链接：点击打开链接

题意：班上有n个同学，现在有一些糖要分给他们，设第i个同学得到的糖为p[i]，分糖必须满足条件：第i个同学要求第j个同学的糖不能超过自己k个，即p[j] - p[i] <= k，k >= 0。要求在满足这些条件的情况下，求出p[n] - p[1]的最大值

代码：

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <queue>
#include <algorithm>
using namespace std;
int dis[150005],vis[150005];
struct node{
    int to,w;
}p[150005];
int next[150005],node[150005];
int n,m;
void spfa(int s){                           //这个题卡STL因此用数组模拟队列，discuss里说好像是spfa+栈
    int i,j,u,t;                            //或者反着建图，但是不到为什么反着反倒TLE，正着建图然后数
    int q[150005];                          //数组模拟队列就可以
    t=0;
    q[t++]=s;
    dis[s]=0;vis[s]=1;
    while(t){
        u=q[--t];
        vis[u]=0;
        for(i=node[u];i!=-1;i=next[i])
        if(dis[p[i].to]>(dis[u]+p[i].w)){
            dis[p[i].to]=dis[u]+p[i].w;
            if(!vis[p[i].to]){
                q[t++]=p[i].to;
                vis[p[i].to]=1;
            }
        }
    }
}                                           //spfa模板    
int main(){                                 //就是求有向图最短路，因为假设2比1最多多2个，3比2最多多3个
    int i,j,a,b,c;                          //1比3最多多4个，假如选择1->2->3则3比1最多多5个不符合1比3最
    while(scanf("%d%d",&n,&m)!=EOF){        //多多4个这个条件，因此选择最短路
        memset(vis,0,sizeof(vis));
        memset(node,-1,sizeof(node));
        for(i=1;i<=m;i++){                  //建图，node数组表示以a为起点的所有边中编号最大的那个
            scanf("%d%d%d",&a,&b,&c);       //next表示与第i条边同起点的，没有则是-1
            p[i].to=b;p[i].w=c;
            next[i]=node[a];
            node[a]=i;
        }
        for(i=1;i<=n;i++)
        dis[i]=99999999;
        spfa(1);
        printf("%d\n",dis[n]);
    }
    return 0;
}