POJ3259——SPFA的DFS优化判断负环

题目描述：

耳机楼里有很多教室，这些教室由双向走廊连接。另外，还存在一些单向的秘密通道，通过它们可以回到过去。现在有 N (1 ≤ N ≤ 500) 个教室，编号 1..N, M (1 ≤ M ≤ 2500) 条走廊，和 W (1 ≤ W≤ 200) 条秘密通道。

DY在养猫之余，还是一个时间旅行爱好者。她希望从一间教室出发，经过一些走廊和秘密通道，回到她出发之前的某个时间。

共有F (1 ≤ F ≤ 5) 组数据。对每组数据，判断DY是否有回到过去的可能性。不存在耗时超过10,000秒的走廊，且不存在能带DY回到10,000秒之前的秘密通道。

Input

首先是一个整数F，表示接下来会有F组数据。

每组数据第1行：分别是三个空格隔开的整数：N，M和W

第2行到M+1行：三个空格分开的数字（S，E，T）描述双向走廊：从S到E需要耗费T秒。两个教室可能由一个以上的路径来连接。

第M +2到M+ W+1行：三个空格分开的数字（S，E，T）描述秘密通道：从S到E可以使时间倒流T秒。

OutputF行，每行对应一组数据。 每组数据输出单独的一行，” YES”表示能满足要求，”NO”表示不能满足要求。Sample Input

2
3 3 1
1 2 2
1 3 4
2 3 1
3 1 3
3 2 1
1 2 3
2 3 4
3 1 8

Sample Output

NO
YES

代码：

//
//  main.cpp
//  POJ3259
//
//  Created by jinyu on 2018/7/13.
//  Copyright © 2018年 jinyu. All rights reserved.
//

#include <iostream>
#include <queue>


using namespace std;


const int INF = 0x3f3f3f3f;
int visited[500+2];
int arc[500+2][500+2];
int dist[500+2];
int b[500+2][500+2];
int flag = 0;
int N;


void SPFAdfs(int start){
    if(flag) return;
    visited[start] = 1;
    for(int i = 1;i<=b[start][0];i++){
        if(dist[b[start][i]] > dist[start] + arc[start][b[start][i]]){
            dist[b[start][i]] = dist[start] + arc[start][b[start][i]];
            if(visited[b[start][i]]){
                flag = 1;
            }
            else{
                SPFAdfs(b[start][i]);
            }
        }
    }
    visited[start] = 0;
}

int main(){
    int F;
    cin>>F;
    while(F--){
        int M,W;
        cin>>N>>M>>W;
        for(int i = 1;i<=N;i++){
            dist[i] = INF;
            visited[i] = 0;
            b[i][0] = 0;
            for(int j = 1;j<=N;j++){
                if(i==j)
                    arc[i][j] = 0;
                else
                    arc[i][j] = INF;
            }
        }
        int mm = M;
        while(mm--){
            int S,E,T;
            cin>>S>>E>>T;
            if(T<arc[S][E]){
                arc[S][E] = T;
                arc[E][S] = T;
                b[S][0]++;
                b[E][0]++;
                b[S][b[S][0]] = E;
                b[E][b[E][0]] = S;
                
            }
        }
        int ww = W;
        while(ww--){
            int S,E,T;
            cin>>S>>E>>T;
            arc[S][E] = -T;
            b[S][0]++;
            b[S][b[S][0]] = E;
        }
        
        
        dist[1] = 0;
        SPFAdfs(1);

        if(flag)
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
        flag = 0;
    }
    return 0;
}

题目本身不难，主要是SPFA的dfs优化来判断负环。但是有很多细节：

1. 第一次打时数组初始化敲错，也没有注意到题目中的点下标是从1开始的。

2.习惯性采用邻接矩阵去做，但其实会TLE，应该采用邻接表（这里我只是简单加了数组b来存储邻接信息）。

3.测试数据貌似都是连通图，如果是非连通的似乎应该对每个点SPFA。

4.memset的问题。