【强联通分量缩点】【搜索】bzoj2208 [Jsoi2010]连通数

两次dfs缩点，然后n次dfs暴搜。

#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;
#define N 2001
vector<int>G[N],rG[N],vs,G2[N];
typedef vector<int>::iterator ITER;
char s[N+1][N+1];
int cmp[N],sum,cnt[N],ans,n;
bool vis[N];
void dfs(int U)
{
    vis[U]=1;
    for(ITER it=G[U].begin();it!=G[U].end();it++)
      if(!vis[*it])
        dfs(*it);
    vs.push_back(U);
}
void dfs2(int U)
{
    vis[U]=1;
    cmp[U]=sum;
    cnt[sum]++;
    for(ITER it=rG[U].begin();it!=rG[U].end();it++)
      if(!vis[*it])
        dfs2(*it);
}
void scc()
{
    for(int i=1;i<=n;i++)
      if(!vis[i])
        dfs(i);
    memset(vis,0,sizeof(vis));
    ITER it=vs.end(); it--;
    for(;;it--)
      {
        if(!vis[*it])
          {
            sum++;
            dfs2(*it);
          }
        if(it==vs.begin()) break;
      }
    memset(vis,0,sizeof(vis));
}
void dfs3(int U)
{
    vis[U]=1; ans+=cnt[U];
    for(ITER it=G2[U].begin();it!=G2[U].end();it++)
      if(!vis[*it])
        dfs3(*it);
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
      {
        scanf("%s",s[i]+1);
        for(int j=1;j<=n;j++)
          if(s[i][j]=='1')
            {
              G[i].push_back(j);
              G[j].push_back(i);
            }
      }
    scc();
    for(int i=1;i<=n;i++)
      for(int j=1;j<=n;j++)
        if(s[i][j]=='1'&&cmp[i]!=cmp[j])
          G2[cmp[i]].push_back(cmp[j]);
    for(int i=1;i<=sum;i++)
      {
        memset(vis,0,sizeof(vis));
        dfs3(i);
      }
    printf("%d\n",ans);
    return 0;
}