BZOJ 2208: [Jsoi2010]连通数

2208: [Jsoi2010]连通数

Time Limit: 20 Sec Memory Limit: 512 MB
Description

Input

输入数据第一行是图顶点的数量，一个正整数N。 接下来N行，每行N个字符。第i行第j列的1表示顶点i到j有边，0则表示无边。

Output

输出一行一个整数，表示该图的连通数。

Sample Input

3

010

001

100
Sample Output

9
HINT

对于100%的数据，N不超过2000。

Source

第一轮

据说暴力可A

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
#define maxn 2020
char s[maxn];
int n,cnt,ans;
bool vis[maxn];
vector<int>G[maxn];
void DFS(int u){
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if(!vis[v]){
            cnt++; vis[v]=1;
            DFS(v);
        }
    }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",s);
        for(int j=1;j<=n;j++)
            if(s[j-1]=='1')
                G[i].push_back(j);
    }
    for(int i=1;i<=n;i++){
        memset(vis,0,sizeof vis );
        vis[i]=1;cnt=1;
        DFS(i);
        ans+=cnt;
    }
    printf("%d",ans);
    return 0;
}