BZOJ 2208: [Jsoi2010]连通数 tarjan bitset

2208: [Jsoi2010]连通数

Time Limit: 20 Sec

Memory Limit: 256 MB

题目连接

http://www.lydsy.com/JudgeOnline/problem.php?id=2208

Description

Input

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

Output

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

Sample Input

3
010
001
100

Sample Output

9

HINT

对于100%的数据，N不超过2000。

题意

题解：

先缩点，变成一个有向无环图之后，再直接跑dp就好了

可以用bitset做

代码：

//qscqesze
#include <cstdio>
#include <cmath>
#include <cstring>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <set>
#include <bitset>
#include <vector>
#include <sstream>
#include <queue>
#include <typeinfo>
#include <fstream>
#include <map>
#include <stack>
typedef long long ll;
using namespace std;
//freopen("D.in","r",stdin);
//freopen("D.out","w",stdout);
#define sspeed ios_base::sync_with_stdio(0);cin.tie(0)
#define maxn 200051
#define mod 10007
#define eps 1e-9
int Num;
//const int inf=0x7fffffff;   //нчоч╢С
const int inf=0x3f3f3f3f;
inline ll read()
{
    ll x=,f=;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
    return x*f;
}
//**************************************************************************************
 
vector<int> Q[maxn];
char s[];
int dfn[],low[],_clock=;
int sta[],top;
bool in_sta[];
int changed[],scc,num[];
bitset<> have[];
void tarjan(int x)
{
    dfn[x]=low[x]=++_clock;
    sta[++top]=x;
    in_sta[x]=;
    for(int i=;i<Q[x].size();i++)
    {
        int v = Q[x][i];
        if(!dfn[v])
            tarjan(v),low[x]=min(low[x],low[v]);
        else if(in_sta[v])
            low[x]=min(low[x],dfn[v]);
    }
    if(dfn[x]==low[x])
    {
        int temp;
        ++scc;
        do{
            temp = sta[top--];
            in_sta[temp]=;
            changed[temp]=scc;
            ++num[scc];
        }while(temp!=x);
    }
}
int main()
{
    int n=read();
    for(int i=;i<=n;i++)
    {
        scanf("%s",s+);
        for(int j=;j<=n;j++)
        {
            if(s[j]=='')
            Q[i].push_back(j);
        }
    }
    for(int i=;i<=n;i++)
        if(!dfn[i])tarjan(i);
    for(int i=;i<=n;i++)
        have[changed[i]][i]=;
    int ans=;
    for(int i=;i<=scc;i++)
    {
        ans+=num[i]*num[i];
        bitset<>temp;
        for(int x = ;x<=n;x++)
        {
            if(changed[x]==i)
            {
                for(int j=;j<Q[x].size();j++)
                {
                    int v = Q[x][j];
                    if(changed[v]!=i)
                    temp|=have[changed[v]];
                }
            }
        }
        ans+=num[i]*temp.count();
        have[i]|=temp;
 
    }
    printf("%d\n",ans);
}