BZOJ-2208-[Jsoi2010]连通数（bitset+floyd）

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

第一轮

 

题解

这道题刚看到就想到暴力，时限有20s，dfs可以A

但是dfs有些慢，我们想一些更优的算法

这道题就是让我们求传递闭包

所以我们可以用bitset来优化floyd

#include<bits/stdc++.h>
#define N 2005
using namespace std;
int n,x,ans;
bitset<N>flag[N];
char st[N];
int main(){
    scanf("%d",&n);
    for (int i=1;i<=n;i++){
        scanf("%s",st+1);
        for (int j=1;j<=n;j++)
            if (st[j]=='1'||i==j) flag[i][j]=true;
    }
    for (int j=1;j<=n;j++)
        for (int i=1;i<=n;i++)
            if (flag[i][j]) flag[i]|=flag[j];
    for (int i=1;i<=n;i++)
        ans+=flag[i].count();
    printf("%d\n",ans);
    return 0;
}