LightOJ 1074 – Extended Traffic 【SPFA判负环】

来源：互联网发布：mac地址采集摄像机编辑：程序博客网时间：2024/05/16 15:12

题目链接：http://www.lightoj.com/volume_showproblem.php?problem=1074

给你一个有向图，求原点1到给定询问点的最短距离是多少，由于可能存在负环，不能用dij,如果点在负环中，则无解，dfs判环。

代码：

#include <iostream>#include <stdio.h>#include <string.h>#include <algorithm>#include <math.h>#include <string>#include <set>#include <queue>#include <stack>#include <vector>#include <malloc.h>using namespace std;int t;int n,m,a,b;int p[21000];const int MAXN = 41000;const int INF = 0x3f3f3f3f;struct Edge{    int v;    int cost;    Edge(int _v=0,int _cost=0)    {        v = _v;        cost = _cost;    }};vector<Edge> E[MAXN];void addedge(int u,int v,int cost){    E[u].push_back(Edge(v,cost));}bool vis[MAXN];int cnt[MAXN];int dist[MAXN];///////判断负环int cir[MAXN];void dfs(int u){    cir[u]=true;    for(int i=0;i<E[u].size();i++)    {        if (!cir[E[u][i].v])            dfs(E[u][i].v);    }}////////////////void SPFA(int start,int n){    memset(vis,false,sizeof(vis));    for(int i=1;i<=n;i++) dist[i]= INF;    vis[start] = true;    dist[start] = 0;    queue<int> que;    while (!que.empty()) que.pop();    que.push(start);    memset(cnt,0,sizeof(cnt));    cnt[start] =1;    memset(cir,false,sizeof(cir));    while (!que.empty())    {        int u = que.front();que.pop();        vis[u]=false;        for(int i=0;i<E[u].size();i++)        {            int v = E[u][i].v;            if (cir[v]) continue;            if (dist[v]>dist[u] + E[u][i].cost)            {                dist[v]=dist[u] + E[u][i].cost;                if (!vis[v])                {                    vis[v] = true;                    que.push(v);                    cnt[v]++;                    if (cnt[v] > n)                        dfs(v);                }            }        }    }}int main(){    scanf("%d",&t);    for(int cases=1;cases<=t;cases++)    {        scanf("%d",&n);        for(int i=1;i<=n;i++)            {                scanf("%d",&p[i]);                E[i].clear();            }        scanf("%d",&m);        while (m--)        {            scanf("%d%d",&a,&b);            addedge(a,b,(p[b]-p[a])*(p[b]-p[a])*(p[b]-p[a]));        }        SPFA(1,n);        printf("Case %d:\n",cases);        scanf("%d",&m);        while (m--)        {            scanf("%d",&b);            if(cir[b] || dist[b] < 3 || dist[b] == INF)                printf("?\n");            else                printf("%d\n",dist[b]);        }    }    return 0;}

0 0