缩点之后dfs求连通块个数,水题
#include <stdio.h>
#include <string.h>
#include <vector>
#include <stack>
using namespace std;
#define maxn 100005
vector<int> G[maxn];
vector<int> H[maxn];
stack<int> S;
int fa[maxn];
int pre[maxn],lowlink[maxn],sccno[maxn],dfs_clock,scc_cnt;
int vis[100005];
void tarjan(int u)
{
pre[u]=lowlink[u]=++dfs_clock;
S.push(u);
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i];
if(!pre[v])
{
tarjan(v);
lowlink[u]=min(lowlink[u],lowlink[v]);
}
else if(!sccno[v])
{
lowlink[u]=min(lowlink[u],pre[v]);
}
}
if(lowlink[u]==pre[u])
{
scc_cnt++;
for(;;)
{
int x=S.top();
S.pop();
sccno[x]=scc_cnt;
if(x==u)break;
}
}
}
void find_scc(int n)
{
dfs_clock=scc_cnt=0;
memset(sccno,0,sizeof(sccno));
memset(pre,0,sizeof(pre));
for(int i=0;i<n;i++)
if(!pre[i])tarjan(i);
}
int dfs(int x)
{
for(int i=0; i<H[x].size(); i++)
{
int u=H[x][i];
if(!vis[u])
{
vis[u]=1;
dfs(u);
}
}
}
int main()
{
int T,n,m,x,y,ct;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
for(int i=0; i<n; i++)
{
G[i].clear();
H[i].clear();
}
while(m--)
{
scanf("%d%d",&x,&y);
x--;y--;
G[x].push_back(y);
}
find_scc(n);
memset(fa,0,sizeof(fa));
for(int i=0;i<n;i++)
{
for(int j=0;j<G[i].size();j++)
{
int u=G[i][j];
if(sccno[i]!=sccno[u])
{
H[sccno[i]].push_back(sccno[u]);
fa[sccno[u]]++;
}
}
}
memset(vis,0,sizeof(vis));
ct=0;
for(int i=0;i<scc_cnt;i++)
{
if(!fa[i])
{
vis[i]=1;
dfs(i);
ct++;
}
}
printf("%d\n",ct);
}
return 0;
}