BZOJ2208 [JSOI2010] 连通数

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=2208

Description

Input

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

Output

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

 

Tarjan缩点+拓扑排序+bitset，mark[i][j]中的i表示第i个强连通分量，每一个点j对应mark[i][j/30+1]从右边数起第j%30位。

每一个点与一个强连通分量连通的答案贡献值等于该强连通分量的大小

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#define rep(i,l,r) for(int i=l; i<=r; i++)
#define clr(x,y) memset(x,y,sizeof(x))
#define travel(x) for(int i=last[x]; i; i=edge[i].pre)
#define travel2(x) for(int i=last2[x]; i; i=edge2[i].pre)
using namespace std;
const int maxn = 2010;
struct Edge{
    int pre,to;
}edge[maxn*maxn],edge2[maxn*maxn];
int n,m,x,now,maxx,tot=0,tot2=0,ans=0,scc=0,dfsclock=0;
int last[maxn],last2[maxn],dfn[maxn],low[maxn],belong[maxn],cnt[maxn],ind[maxn],mark[maxn][70];
char ch[maxn];
bool isin[maxn];
stack <int> s;
queue <int> q;
inline void addedge(int x,int y){
    edge[++tot].pre = last[x];
    edge[tot].to = y;
    last[x] = tot;
}
inline void addedge2(int x,int y){
    edge2[++tot2].pre = last2[x];
    edge2[tot2].to = y;
    last2[x] = tot2;
    ind[y]++;
}
void tarjan(int x){
    dfn[x] = low[x] = ++dfsclock;
    isin[x] = 1; s.push(x);
    travel(x){
        if (!dfn[edge[i].to]){
            tarjan(edge[i].to);
            low[x] = min(low[x],low[edge[i].to]);
        }
        else if (isin[edge[i].to]) low[x] = min(low[x],dfn[edge[i].to]);
    }
    if (dfn[x] == low[x]){
        scc++;
        while (s.top() != x){
            isin[s.top()] = 0;
            belong[s.top()] = scc;
            cnt[scc]++;
            s.pop();
        }
        isin[x] = 0; belong[x] = scc; cnt[scc]++; s.pop();
    }
}
void toposort(){
    clr(mark,0);
    rep(i,1,n) mark[belong[i]][i/30+1] |= (1 << (i % 30));
    rep(i,1,scc) if (!ind[i]) q.push(i);
    while (!q.empty()){
        now = q.front(); q.pop();
        travel2(now){
            rep(j,1,maxx) mark[edge2[i].to][j] |= mark[now][j];
            ind[edge2[i].to]--;
            if (!ind[edge2[i].to]) q.push(edge2[i].to);
        }
    }
}
int main(){
    scanf("%d",&n); clr(last,0); clr(last2,0); clr(mark,0); maxx = n / 30 + 1;
    rep(i,1,n){
        scanf("%s",ch);
        rep(j,1,n) if (ch[j-1] - '0') addedge(i,j);
    }
    clr(isin,0); clr(dfn,0); clr(cnt,0);
    rep(i,1,n) if (!dfn[i]) tarjan(i);
    clr(ind,0);
    rep(now,1,n) travel(now) if (belong[now] != belong[edge[i].to])
        addedge2(belong[now],belong[edge[i].to]);
    toposort();
    rep(i,1,scc) rep(j,1,n)
    if (mark[i][j/30+1] & (1 << (j % 30))) ans += cnt[i];
    printf("%d\n",ans);
    return 0;
}