BZOJ 2208 [Jsoi2010] 连通数 (强连通分量+DFS (划掉) bitset)

版权声明：欢迎神犇指教 https://blog.csdn.net/sdxtcqs/article/details/82182835

https://www.lydsy.com/JudgeOnline/problem.php?id=2208
题意：求一张图的连通数，连通数的意思是可达顶点对的个数

对于100%的数据，N不超过2000，Time Limit: 20 Sec

先求一波强连通分量，显然对于每一个强连通块，对答案的贡献是连通块内节点个数的平方。
然后在缩点后的图上跑DFS就行了。
Tarjan+DFS的时间6940ms和空间24.9MB

然而坑爹的是，因为这题给了20秒，这个题不缩点，直接暴力跑DFS，竟然也能过。
直接暴力DFS的时间11628ms和空间40.5MB

当然出于一个程序员对时间复杂度的追求，应该选择尽量快的算法，还可以用bitset优化，我太菜了不太会用，参考这个博客https://blog.csdn.net/Jaihk662/article/details/76851404
Tarjan+bitset的时间4848ms和空间5.9MB

Tarjan缩点DFS

#include<iostream>
#include<cstdio>
#include<stack>
#include<algorithm>
#include<cstring>

using namespace std;
const int maxn=2010,maxm=2001000;
int dfn[maxn],low[maxn],head[maxn],color[maxn],sum[maxn];
int n,m,k,cnt,deep;
long long ans;
bool vis[maxn],f[maxn][maxn];
char mp[maxn][maxn];
struct edge{
    int v,next;
}e[maxm];

stack<int> s;

void addedge(int u,int v){e[k].v=v;e[k].next=head[u];head[u]=k++;}

void tarjan(int u)
{
    deep++;
    dfn[u]=deep;
    low[u]=deep;
    vis[u]=1;
    s.push(u);
    for(int i=head[u];~i;i=e[i].next)
    {
        int v=e[i].v;
        if(!dfn[v])
        {
            tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else
        {
            if(vis[v])
            {
                low[u]=min(low[u],low[v]);
            }
        }
    }
    if(dfn[u]==low[u])
    {
        cnt++;
        color[u]=cnt;
        vis[u]=0;
        while(s.top()!=u)
        {
            color[s.top()]=cnt;
            vis[s.top()]=0;
            s.pop();
        }
        s.pop();
    }
}
void dfs(int s,int now)
{
    vis[now]=1;
    for(int i=head[now];~i;i=e[i].next)
    {
        int v=e[i].v;
        //printf("%d %d\n",s,v);
        if(!vis[v])
        {
            ans+=sum[s]*sum[v];
            dfs(s,v);
        }
    }
}
int main()
{
    int x,y;
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%s",mp[i]+1);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(mp[i][j]=='1')
                addedge(i,j);
        }
    }
    for(int i=1;i<=n;i++)
    {
        if(!dfn[i])
            tarjan(i);
    }

    for(int i=1;i<=n;i++)
        sum[color[i]]++;
    for(int i=1;i<=cnt;i++)
        ans+=sum[i]*sum[i];
    memset(head,-1,sizeof(head));
    memset(e,0,sizeof(e));
    k=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(color[i]!=color[j]&&!f[color[i]][color[j]]&&mp[i][j]=='1')
            {
                addedge(color[i],color[j]);
                f[color[i]][color[j]]=1;
            }
        }
    }
    for(int i=1;i<=cnt;i++)
    {
        memset(vis,0,sizeof(vis));
        dfs(i,i);
    }
    printf("%d\n",ans);
    return 0;
}

直接暴力DFS

#include<iostream>
#include<cstdio>
#include<stack>
#include<algorithm>
#include<cstring>

using namespace std;
const int maxn=2010,maxm=2001000;
int dfn[maxn],low[maxn],head1[maxn],head2[maxn],color[maxn],sum[maxn];
int n,m,k,cnt,deep,ans;
bool vis[maxn],f[maxn][maxn];
char mp[maxn][maxn];
struct edge{
    int v,next;
}e1[maxm],e2[maxm];

stack<int> s;

void addedge1(int u,int v){e1[k].v=v;e1[k].next=head1[u];head1[u]=k++;}
void addedge2(int u,int v){e2[k].v=v;e2[k].next=head2[u];head2[u]=k++;}

void dfs(int now)
{
    vis[now]=1;
    ans++;
    for(int i=head1[now];~i;i=e1[i].next)
    {
        int v=e1[i].v;
        if(!vis[v])
        {
            dfs(v);
        }
    }
}
int main()
{
    int x,y;
    memset(head1,-1,sizeof(head1));
    memset(head2,-1,sizeof(head2));
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%s",mp[i]+1);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(mp[i][j]=='1')
                addedge1(i,j);
        }
    }
    for(int i=1;i<=n;i++)
    {
        memset(vis,0,sizeof(vis));
        dfs(i);
    }
    printf("%d\n",ans);
    return 0;
}