分析:
求传递闭包,有向图,每个环内的点的连通数都一样,所以先缩点变成 DAG,然后按照拓扑排序倒着 dp ,用bitset维护走过的点。
bitset<N>dp[N],设 dp[i] 表示 以点 i 为起点的连通数,即走过的点
则 若存在边 i->j 则 dp[i] | = dp[j] 。复杂度 O ( n 2 / W ) O(n^{2}/W) O(n2/W)
代码:
#include<queue>
#include<bitset>
#include<cstdio>
#include<iostream>
using namespace std;
const int N = 2000+10;
const int M = N*N;
char str
[N];
struct node{
int to,nxt;
}e[M],E[M];
int hd1[N],tot1,hd2[N],tot2;
void add(node e[],int head[],int &tot,int u,int v){
e[++tot].to=v;
e[tot].nxt=head[u];
head[u]=tot;
}
bitset<N>dp[N];
int dfn[N],low[N],dfncnt,s[N],tp;
int scc[N],sc;
int sz[N],n;
void tarjan(int u)
{
low[u]=dfn[u]=++dfncnt,s[++tp]=u;
for(int i=hd1[u];i;i=e[i].nxt){
int v=e[i].to;
if(!dfn[v])
tarjan(v),low[u]=min(low[u],low[v]);
else if(!scc[v])
low[u]=min(low[u],dfn[v]);
}
if(dfn[u]==low[u]){
++sc;
while(s[tp]!=u) dp[sc].set(s[tp]),scc[s[tp]]=sc,sz[sc]++,--tp;
dp[sc].set(s[tp]),scc[s[tp]]=sc,sz[sc]++,--tp;
}
}
int a[N],p,in[N];
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%s",str+1);
for(int j=1;j<=n;j++){
if(str[j]=='1')
add(e,hd1,tot1,i,j);
}
}
for(int i=1;i<=n;i++)
if(!dfn[i])
tarjan(i);
for(int u=1;u<=n;u++)
for(int i=hd1[u];i;i=e[i].nxt){
int v=e[i].to;
if(scc[u]==scc[v]) continue;
add(E,hd2,tot2,scc[u],scc[v]);
in[scc[v]]++;
}
queue<int>que;
for(int i=1;i<=sc;i++)
if(!in[i])
que.push(i);
while(!que.empty()){
int u=que.front(); que.pop();
a[++p]=u;
for(int i=hd2[u];i;i=E[i].nxt){
int v=E[i].to;
if(--in[v]==0) que.push(v);
}
}
for(int t=p;t>=1;t--)
for(int i=hd2[a[t]];i;i=E[i].nxt){
int u=a[t],v=E[i].to;
dp[u]|=dp[v];
}
int res=0;
for(int i=1;i<=sc;i++) res+=dp[i].count()*sz[i];
cout<<res;
}