程序设计思维与实践 Week7 作业 C - TT 的美梦

这一晚，TT 做了个美梦！

在梦中，TT 的愿望成真了，他成为了喵星的统领！喵星上有 N 个商业城市，编号 1 ～ N，其中 1 号城市是 TT 所在的城市，即首都。

喵星上共有 M 条有向道路供商业城市相互往来。但是随着喵星商业的日渐繁荣，有些道路变得非常拥挤。正在 TT 为之苦恼之时，他的魔法小猫咪提出了一个解决方案！TT 欣然接受并针对该方案颁布了一项新的政策。

具体政策如下：对每一个商业城市标记一个正整数，表示其繁荣程度，当每一只喵沿道路从一个商业城市走到另一个商业城市时，TT 都会收取它们（目的地繁荣程度 - 出发地繁荣程度）^ 3 的税。

TT 打算测试一下这项政策是否合理，因此他想知道从首都出发，走到其他城市至少要交多少的税，如果总金额小于 3 或者无法到达请悄咪咪地打出 ‘?’。
Input
第一行输入 T，表明共有 T 组数据。（1 <= T <= 50）

对于每一组数据，第一行输入 N，表示点的个数。（1 <= N <= 200）

第二行输入 N 个整数，表示 1 ～ N 点的权值 a[i]。（0 <= a[i] <= 20）

第三行输入 M，表示有向道路的条数。（0 <= M <= 100000）

接下来 M 行，每行有两个整数 A B，表示存在一条 A 到 B 的有向道路。

接下来给出一个整数 Q，表示询问个数。（0 <= Q <= 100000）

每一次询问给出一个 P，表示求 1 号点到 P 号点的最少税费。
Output
每个询问输出一行，如果不可达或税费小于 3 则输出 ‘?’。
Sample Input
2
5
6 7 8 9 10
6
1 2
2 3
3 4
1 5
5 4
4 5
2
4
5
10
1 2 4 4 5 6 7 8 9 10
10
1 2
2 3
3 1
1 4
4 5
5 6
6 7
7 8
8 9
9 10
2
3 10
Sample Output
Case 1:
3
4
Case 2:
?
?
解题思路：
首先输入n个点的权值，然后每当添加一条边的时候，就算出边的权值来的建立图。

 E[u].push_back(Edge(v,(a[v]-a[u])*(a[v]-a[u])*(a[v]-a[u])));

这题由于存在负环，所以采用spfa算法。
建立一个队列
队列中存储被成功松弛的点
每次从队首取点并松弛其邻接点
如果邻接点成功松弛则将其放入队列
如果存在负环，那么最短路经过的边数会大于等于n。并且用cir数组来标识负环上的点。

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);
}

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
using namespace std;
const int MAXN = 205;
const int INF = 0x3f3f3f3f;
struct Edge
{
    int v,W;
    Edge(int _v = 0, int _W = 0)
    {
        v = _v;
        W = _W;
    }
};
vector<Edge> E[MAXN];
bool vis[MAXN];
int cnt[MAXN];
int dist[MAXN];

bool 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> q;
   
    q.push(start);
    memset(cnt,0,sizeof(cnt));
    cnt[start] = 1;
    memset(cir,false,sizeof(cir));
    while(!q.empty())
    {
        int u = q.front();
        q.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].W)
            {
                dist[v] = dist[u] + E[u][i].W;
                if(!vis[v])
                {
                    vis[v] = true;
                    q.push(v);
                    cnt[v]++;
                    if(cnt[v] >=n)
                        dfs(v);
                }
            }
        }
    }
}
int a[MAXN];
int main()
{
    int t;
    int iCase = 0;
    scanf("%d",&t);
    while(t--)
    {
        iCase++;
        int n;
        scanf("%d",&n);
        for(int i = 1;i <= n;i++)
            scanf("%d",&a[i]);
        int m;
        int u,v;
        for(int i = 1;i <= n;i++)
            E[i].clear();
        scanf("%d",&m);
        while(m--)
        {
            scanf("%d%d",&u,&v);
             E[u].push_back(Edge(v,(a[v]-a[u])*(a[v]-a[u])*(a[v]-a[u])));
        }
        SPFA(1,n);
        printf("Case %d:\n",iCase);
        int q;
        scanf("%d",&q);
        while(q--)
        {
            scanf("%d",&u);
            if(cir[u] || dist[u] < 3 || dist[u] == INF)
                printf("?\n");
            else printf("%d\n",dist[u]);
        }
    }
    return 0;
}