bzoj 2208 [Jsoi2010]连通数 bitset

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。

传送门
沙伯题……数据还水……
就是求个传递闭包，用floyd优化一下就好了。
看见评论里有人说dfs是O($n^2$)的，切记切记不是！
是O($n*(n+m)$)的！而大多时候m接近于$n^2$，
所以bitset稳健。

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