Candies POJ 3159 (堆优化的SPFA,)

开始看半天看不懂题目意思,原来是~~~

分差约束系统:

全都是两个未知数的差小于等于某个常数(大于等于也可以,因为左右乘以-1就可以化成小于等于)。这样的不等式组就称作差分约束系统。

也就是说求1到N的最短路就行了。

但是这道题数据比较大,m = 150000,只能用堆优化的SPFA

#include
#include
#include
#include
#include
#include 
using namespace std;

const int inf = 0x3f3f3f3f;
const int maxn = 150005;
int Left[maxn],Right[maxn],Next[maxn],key[maxn],dict[maxn],n,m,cnt;
bool vis[maxn];

void init() {
    cnt = 0;
    memset(vis,false,sizeof(vis));
    memset(Left,-1,sizeof(Left));
    memset(dict,inf,sizeof(dict));
}

void add(int u,int v,int value) {
    key[cnt] = value;
    Next[cnt] = Left[u];
    Left[u] = cnt;
    Right[cnt] = v;
    cnt++;
}

void SPFA(int start) {
    dict[start] = 0;
    vis[start] = true;
    stack q; //头文件里加stack再定义一个就行;
    while(!q.empty()) q.pop();
    q.push(start);
    while(!q.empty()) {
        int temp = q.top(); //和队列不同的是这里改成q.top;
        q.pop();
        vis[temp] = false;
        for(int i = Left[temp];i != -1;i = Next[i]) {
            if(dict[Right[i]] > dict[temp] + key[i]) {
                dict[Right[i]] = dict[temp] + key[i];
                if(!vis[Right[i]]) {
                    vis[Right[i]] = true;
                    q.push(Right[i]);
                }
            }
        }
    }
}

int main() {
    while(scanf("%d%d",&n,&m) != EOF) {
        init();
        int u,v,value;
        for(int i = 0;i < m;i++) {
            scanf("%d%d%d",&u,&v,&value);
            add(u,v,value);
        }
        SPFA(1);
        printf("%d\n",dict[n]);
    }
}


你可能感兴趣的:(最短路)