90pts求助
查看原帖
90pts求助
763878
Jerry_heng楼主2024/3/11 20:36
#include<bits/stdc++.h>
using namespace std;
const int mxn=2010;
int n,low[mxn],head[mxn],dfn[mxn],be[mxn],d[mxn],cnt,col,cnt2,siz[mxn],ans;
bool vis[mxn];
bitset<mxn>g[mxn];
stack<int>st;
queue<int>q;
struct node{
	int frm,to,nxt;
}edge[mxn*mxn];
void add(int u,int v){
	edge[++cnt]=(node){u,v,head[u]};
	head[u]=cnt;
}
void tarjan(int u){
	dfn[u]=low[u]=++cnt2;
	st.push(u);
	vis[u]=1;
	for(int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].to;
		if(!dfn[v]){
			tarjan(v);
			low[u]=min(low[u],low[v]);
		}
		else if(vis[v])low[u]=min(low[u],dfn[v]);
	}
	if(low[u]==dfn[u]){
		col++;
		while(st.top()!=u){
			int x=st.top();st.pop();
			vis[x]=0,be[x]=col;
			siz[col]++;
		}
		int x=st.top();st.pop();
		vis[x]=0,be[x]=col;
		siz[col]++;
	}
}
void topo(){
	for(int i=1;i<=col;i++)
		if(!d[i])q.push(i);
	while(!q.empty()){
		int u=q.front();
		q.pop();
		for(int v=1;v<=col;v++){
			if(!g[u][v]||u==v)continue;
			g[v]|=g[u];
			if(!(--d[v]))q.push(v);
		}
	}
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++){
			int x;
			scanf("%1d",&x);
			if(x)add(i,j);
		}
	for(int i=1;i<=n;i++)
		if(!dfn[i])tarjan(i);
	for(int i=1;i<=cnt;i++){
		int u=edge[i].frm,v=edge[i].to;
		if(be[u]!=be[v]){
			if(!g[be[v]][be[u]])d[be[u]]++;
			g[be[v]][be[u]]=1;
		}
	}
	for(int i=1;i<=col;i++)g[i][i]=1;
	topo();
	for(int i=1;i<=col;i++)
		for(int j=1;j<=col;j++)
			ans+=g[i][j]*siz[i]*siz[j];
	printf("%d\n",ans);
	return 0;
}
2024/3/11 20:36
加载中...