POJ 1860 最短路

Bellman Ford可以求是否存在負權迴路,我們把他改成求正權迴路。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <cmath>
#include <set>
#include <iostream>
const double PI=acos(-1.0);
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int MAXN=550;
double dist[MAXN];
struct Edge
{
    int u,v;
    double c,r;
    Edge(int _u=0,int _v=0,double _c=0,double _r=0):u(_u),v(_v),c(_c),r(_r){}
};
vector<Edge>E;
bool bellman_ford(int start,int n,double V)
{
    for(int i=1;i<=n;i++) dist[i]=0;
    dist[start]=V;
    for(int i=1;i<n;i++) {
        bool flag=false;
        for(int j=0;j<E.size();j++) {
            int u=E[j].u;
            int v=E[j].v;
            if(dist[v]<(dist[u]-E[j].c)*E[j].r) {
                dist[v]=(dist[u]-E[j].c)*E[j].r;
                flag=true;
            }
        }
        if(!flag) return false;
    }
    for(int j=0;j<E.size();j++)
        if(dist[E[j].v]<(dist[E[j].u]-E[j].c)*E[j].r)
            return true;
    return false;
}
int main()
{
    int n,m,s;
    int a,b,c;
    double v;
    while(~scanf("%d%d",&n,&m)&&n&&m) {
        scanf("%d%lf",&s,&v);
        E.clear();
        for(int i=0;i<m;i++) {
            scanf("%d%d",&a,&b);
            double r1,c1,r2,c2;
            scanf("%lf%lf%lf%lf",&r1,&c1,&r2,&c2);
            E.push_back(Edge(a,b,c1,r1));
            E.push_back(Edge(b,a,c2,r2));
        }
        if(bellman_ford(s,n,v)) {
            printf("YES\n");
        }
        else printf("NO\n");
    }
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章