程序设计思维与实践 Week7 作业C SPFA

题意：

给定T组数据，每组数据有N个点和M条单向边。每个点有一个值，边的权值即为终点的值减去起点值得三次方，求出从起点开始到每个点得最小权值和。（由题意可知，图中有负边权，即有得点可能达不到，或者是处于负环路上。）如果某个点不能达到或者是处于负环路，则输出“?”。

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 号点的最少税费。

样例：

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

output:

每个例子输出一个Case i: 每个询问输出一行，如果不可达或税费小于 3 则输出 ‘?’。
样例：

Case 1:
3
4
Case 2:
?
?

思路:

首先根据题意肯定是求最短路径得题目，然后边权是隐式给出得，根据题目，可知，边权中有负边，故需要用SPFA算法来计算。即从起点开始，进行一边SPFA算法，算法和dij算法相似，多了一个cnt[]数组来存储到达每一个点的时候所走的边的数量。其余更新dis的路径长度和dij算法相似。在更新时，同步更新cnt。同时易知，在n个点的图中，若到达任一点所走的边数如果大于n，则一定是有负环路。则在此处，对所有在负环路上和负环路所影响的点都不能正确到达。即这些点要输出“？”。即对出现这种点的时候，从这个点开始进行一遍bfs，标记这个点所能到达的所有的点，（我标记为-1），此后在查询输出的时候，若这个点的dis值无穷大，说明不能到达这个点，若这个点的值小于3，（此题认为小于3即不符合要求），同样输出“？”，其余的输出查询所得的dis[]值即可。

代码：

#include <cstdio>
#include <algorithm> 
#include <iostream>
#include <string.h> 
#include <queue>
#include <cmath>
using namespace std;


const int inf = 1e8;
int T,N,M,Q,P;
bool vis[205];   //记录当前是否在队列中 
int dis[205];    //花费 
int cnt[205];    //记录边数 
int a[205];   //每个点的值 

struct Edge {
	int to, next, w;
}e[200010];
int head[205];
int tot;

queue<int > q;  //最大堆

void add_edge(int t, int f, int W) {
	e[tot].to = t;
	e[tot].next = head[f];
	head[f] = tot;
	e[tot].w = W;
	tot++;
}

queue <int> q2;
bool vis2[205]; 

void bfs(int x){
	while (q2.size()) q2.pop();
	
	memset(vis2,0,sizeof(vis2));
	dis[x]=-1;
	vis2[x] = 1;
	q2.push(x);
	while(q2.size()){
		int x = q2.front();
		q2.pop();
		
		int i = head[x];
		while (i != -1) {
			int t1 = e[i].to;
			if(vis2[t1]==0){
				q2.push(t1);
				vis2[t1]=1;
				dis[t1]=-1;
			}
			i = e[i].next;
		}
	}	
}


void spfa() {
	while (q.size()) q.pop();

	//初始化
	for (int i = 1; i <= N; i++) {
		vis[i] = 0, dis[i] = inf;
		cnt[i]=0;
	}
	dis[1] = 0, vis[1]=1;
	q.push(1);
	while (q.size()) {
		int x = q.front();
		q.pop();
		vis[x] = 0;  //x出队列，标记值为0,在spfa中可以被队列弹出多次 
		
		int i = head[x];
		while (i != -1) {
			int t1 = e[i].to;
			int w1 = e[i].w;
			if (dis[t1] > dis[x] + w1) {
				dis[t1] = dis[x] + w1;  //记录路径长度 
				cnt[t1]=cnt[x]+1;
				if(cnt[t1]>=N){
					bfs(t1);
				}else if(dis[t1]!=-1 && vis[t1]==0){
					q.push(t1);
					vis[t1]=1;
				}	
			}
			i = e[i].next;
		}
	}
}

int main() {
	ios::sync_with_stdio(0);
	cin>>T;
	for(int i=1; i<=T;i++){
		tot=0;
		cin>>N;
		for(int j=1;j<=N;j++){
			cin>>a[j];
			head[j]=-1; 
		}
		
		cin>>M;
		int x,y;
		for(int j=1;j<=M;j++){
			cin>>x>>y;
			add_edge(y,x,(a[y]-a[x])*(a[y]-a[x])*(a[y]-a[x]));
		}
		spfa();
		cout<<"Case "<<i<<":"<<endl;
		cin>>Q;
		for(int j=0;j<Q;j++){
			cin>>x;
			if(dis[x]<3 || dis[x]==1e8){
				cout<<"?"<<endl;
			}else{
				cout<<dis[x]<<endl;
			}
		}	
	}
	return 0;
}