LightOJ - 1074（SPFA求负环最短路）

题目传送门

题解：给出n个点的权值，点i–>点j 的边的权值=（j-i）的三次方，给出m条边，以及q次查询；
由题意很容易知道可能存在负边权，所以可能被存在负环；而SPFA跑最短路可以很简单地解决负环问题；

#include <set>
#include <map>
#include <cmath>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include<cstring>
#include <stdio.h>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn = 1e6+5;
int t,n,m,q;
struct Edge{
    int to,next,val;
}edge[maxn];
int head[maxn],tot,dis[maxn],cnt[maxn],a[500];
bool vis[maxn],v[maxn];
void add_edge(int u,int v,int w){
    edge[tot].to=v;
    edge[tot].next=head[u];
    edge[tot].val=w;
    head[u]=tot++;
}
void init()
{
    memset(vis,false,sizeof(vis));
    memset(v,false,sizeof(v));
    memset(dis,INF,sizeof(dis));
    memset(cnt,0,sizeof(cnt)); tot=0;
    memset(head,-1,sizeof(head));
}
void SPFA(int s)
{
    queue<int>q;
    q.push(s); dis[s]=0; vis[s]=true;cnt[s]++;
    while (!q.empty()){
        int x=q.front() ;q.pop();
        vis[x]=false;
        for(int i=head[x];i!=-1;i=edge[i].next){
            int y=edge[i].to;
            if(dis[y]>dis[x]+edge[i].val){
                dis[y]=dis[x]+edge[i].val;
                if(!vis[y]&&!v[y]){
                    q.push(y),vis[y]=true;
                    cnt[y]++;
                // cnt数组标记一个点的入队次数，超过入队上限n即表示存在负环    
                    if(cnt[y]>=n) v[y]=true;
               //  v数组标记负环     
                }
            }
        }
    }
}
int main()
{
    cin>>t;
    int cas=0;
    while (t--){
        init();
        scanf("%d",&n);
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        scanf("%d",&m);
        for(int i=1,u,v;i<=m;i++){
            scanf("%d%d",&u,&v);
            int d=(a[v]-a[u])*(a[v]-a[u])*(a[v]-a[u]);
            add_edge(u,v,d);
        }
        SPFA(1);
        int ob;
        scanf("%d",&q);
        printf("Case %d:\n",++cas);
        while (q--){
            scanf("%d",&ob);
            if(dis[ob]<3||v[ob]||dis[ob]==INF)
                printf("?\n");
            else
                printf("%d\n",dis[ob]);
        }
    }
   return 0;
}