《Extended Traffic》最短路spfa

这个问题核心就是负环上元素的处理。
这里给出两种思路。
1.首先在找到一个负环上的元素时，即它的进队次数大于n时，不退出spfa函数，继续更新，直到所有在环上的元素的进队次数都大于n，这是肯定的，因为是负环最后环上的元素都肯定会大于n。注意为了不死循环，当一个元素确定为在负环上时，我们就不再让它入队。
2.dfs法：当我们找到一个负环上的元素时，我们就dfs它能到的所有点，用一个数组标记，最后这些带有标记的点就是负环上的元素了。

#include<iostream>
#include<stdio.h>
#include<queue>
#include<algorithm>
#include<math.h>
#include<stack>
#include<map>
#include<limits.h>
#include<vector>
#include<string.h>
#include<string>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> pii;
const int N = 105*105;
const int M = 2*1e5+5;
#define pi acos(-1)
#define INF 1e9
#define INM INT_MIN
#define MAX 205
#define pb(a)  push_back(a)
#define mk(a,b) make_pair(a,b)
#define dbg(x) cout << "now this num is " << x << endl;
#define sd(ax) scanf("%d",&ax)
#define sld(ax) scanf("%lld",&ax)
#define sdd(ax,bx) scanf("%d %d",&ax,&bx)
#define sddd(a,b,c) scanf("%d %d %d",&a,&b,&c)
#define pr(a) printf("%d\n",a)
#define plr(a) printf("%lld\n",a)
#define pr_(a) printf("%d ",a)
#define _pr(a) printf(" %d",a)
int n,a[205],m,vis[205],in[205],dis[205],use[205];
struct Node
{
    int to,cost;
};
vector<Node> G[205];
void spfa1()//所有点记录
{
    for(int i=1;i<=n;++i) dis[i] = INF;
    memset(in,0,sizeof(in));
    memset(vis,0,sizeof(vis));
    dis[1] = 0;
    queue<int> Q;
    Q.push(1);
    while(!Q.empty())
    {
        int u = Q.front();
        Q.pop();
        vis[u] = 0;
        in[u]++;
        for(int i=0;i<G[u].size();++i)
        {
            Node e = G[u][i];
            if(dis[e.to] > dis[u]+e.cost)
            {
                dis[e.to] = dis[u]+e.cost;
                if(!vis[e.to] && in[e.to] <= n)
                {
                    vis[e.to] = 1;
                    Q.push(e.to);
                }
            }
        }
    }
}
void dfs(int x)
{
    use[x] = 1;
    for(int i=0;i<G[x].size();++i)
    {
        int y = G[x][i].to;
        if(!use[y])
        {
            dfs(y);
        }
    }
}
void spfa2()//dfs所有负环上的点
{
    for(int i=1;i<=n;++i) dis[i] = INF;
    memset(in,0,sizeof(in));
    memset(vis,0,sizeof(vis));
    memset(use,0,sizeof(use));//dfs用的记录数组
    dis[1] = 0;
    queue<int> Q;
    Q.push(1);
    while(!Q.empty())
    {
        int u = Q.front();
        Q.pop();
        vis[u] = 0;
        in[u]++;
        for(int i=0;i<G[u].size();++i)
        {
            Node e = G[u][i];
            if(dis[e.to] > dis[u]+e.cost)
            {
                dis[e.to] = dis[u]+e.cost;
                if(!vis[e.to])
                {
                    vis[e.to] = 1;
                    Q.push(e.to);
                }
                if(in[u] > n)
                {
                    dfs(u);
                    return ;
                }
            }
        }
    }
}
int main()
{
    int t,q,cnt = 0;
    sd(t);
    while(t--)
    {
        cnt++;
        sd(n);
        for(int i=1;i<=n;++i) sd(a[i]);
        for(int i=1;i<=n;++i) G[i].clear();
        sd(m);
        while(m--)
        {
            int x,y;
            sdd(x,y);
            Node p;
            p.to = y;p.cost = (a[y]-a[x])*(a[y]-a[x])*(a[y]-a[x]);
            G[x].pb(p);
        }
        spfa();
        sd(q);
        printf("Case %d:\n",cnt);
        while(q--)
        {
            int x;
            sd(x);
            if(dis[x] == INF || in[x] > n || dis[x] < 3) printf("?\n");
            else printf("%d\n",dis[x]);
        }
    }
}