CodeForces 402E(强连通分量)

链接：点击打开链接

题意：给一个矩阵A,每一个元素都大于等于0，判断是否存在k使得A^k的所有元素都大于0

代码：

#include <set>
#include <map>
#include <queue>
#include <stack>
#include <string>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
int vis[2005];
vector<int> op,G[2005],rG[2005];
void dfs(int S){
    int i,tmp;
    vis[S]=1;
    for(i=0;i<G[S].size();i++){
        tmp=G[S][i];
        if(vis[tmp])
        continue;
        dfs(tmp);
    }
    op.push_back(S);
}
void rdfs(int S){
    int i,tmp;
    vis[S]=1;
    for(i=0;i<rG[S].size();i++){
        tmp=rG[S][i];
        if(vis[tmp])
        continue;
        rdfs(tmp);
    }
}
int main(){
    int n,i,j,u,ans;
    while(scanf("%d",&n)!=EOF){
        for(i=1;i<=n;i++){                      //其实就是相当于是否每两个点
            G[i].clear();                       //可以走k步互相到达，也就是
            rG[i].clear();                      //是强连通分量为1
        }
        for(i=1;i<=n;i++){
            for(j=1;j<=n;j++){
                scanf("%d",&u);
                if(u>0){
                    G[i].push_back(j);
                    rG[j].push_back(i);
                }
            }
        }
        op.clear();
        memset(vis,0,sizeof(vis));
        for(i=1;i<=n;i++){
            if(vis[i]==0)
            dfs(i);
        }
        ans=0;
        memset(vis,0,sizeof(vis));
        for(i=op.size()-1;i>=0;i--){
            if(vis[op[i]]==0){                  //判断是否能缩成一个点
                ans++;
                rdfs(op[i]);
            }
        }
        if(ans==1)
        puts("YES");
        else
        puts("NO");
    }
    return 0;
}