CF gym102091 G.Communication(矩阵快速幂+biset/Floyd、并查集)

https://codeforces.com/gym/102091
题意：有n个办公室，问重新划分后，一共有几个部门。部门划分原则：①若A和B在同一个部门，那么A可以给B发送消息，B也可以给A发送消息；②若A可以给B发送消息，B也可以给A发送消息，那么A和B在同一个部门里

思路：先统计各办公室之间的连通情况，可以用矩阵快速幂或Floyd处理（因为只需要看是否能到达，所以可以用bitset优化下矩阵快速幂）。然后直接看，若A能到达B，B也能到达A，那么在AB在一个块内，用并查集统计连通块的个数。

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<bitset>
using namespace std;
//#define int long long
#define ll long long
#define ull unsigned long long
#define pii pair<int,int>
#define mid ((l + r)>>1)
#define chl (root<<1)
#define chr (root<<1|1)
#define lowbit(x) ( x&(-x) )
const int manx = 2e2 + 10;
const int manx2 = 4e7 + 10;
const int INF = 1e9;
const int mod = 244353;

int N,n,e;
int fa[manx];
struct mat
{
    bitset<manx>a[manx];
    bitset<manx>r[manx];
    mat(){
        for(int i=1;i<=n;++i)
            a[i].reset(),r[i].reset();
    }
    void print(){
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                cout<<a[i][j]<<' ';
            }
            puts("");
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                cout<<r[i][j]<<' ';
            }
            puts("");
        }
    }
    inline mat operator*(const mat &b)const{
        mat ans;
        for(int i=1;i<=n;++i)
            for(int j=1;j<=n;++j)ans.a[i][j]=a[i][j],ans.r[i][j]=r[i][j];
        for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j){
            ans.a[i][j]=ans.a[i][j]|((a[i]&b.r[j]).any());
            ans.r[i][j]=ans.r[i][j]|((a[j]&b.r[i]).any());
        }
        return ans;
    }
    inline mat operator^(ull x)const{
        mat ans,bas;
        for(int i=1;i<=n;++i)ans.a[i][i]=1;
        for(int i=1;i<=n;++i)
            for(int j=1;j<=n;++j)bas.a[i][j]=a[i][j],bas.r[i][j]=r[i][j];
        while(x){
            if(x&1)
                ans=ans*bas;
            x>>=1;
            bas=bas*bas;
        }
        return ans;
    }
};
int  ffind(int x)
{
    if(fa[x]==x)return x;
    return fa[x]=ffind(fa[x]);
}
int main()
{
    scanf("%d",&N);
    while(N--){
        int a,b;
        mat mp;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            fa[i]=i;
        scanf("%d",&e);
        for(int i=1;i<=e;i++){
            scanf("%d%d",&a,&b);
            ++a,++b;
            mp.a[a][b]=1;
            mp.r[b][a]=1;
        }
        mp=mp^n;
        for(int i=1;i<=n;i++)
        for(int j=i+1;j<=n;j++){
            if(mp.a[i][j]&&mp.a[j][i]){
                int fx=ffind(i);
                int fy=ffind(j);
                if(fx!=fy)
                    fa[fx]=fy;
            }
        }
        int ans=0;
        for(int i=1;i<=n;i++)
            if(fa[i]==i)ans++;
        printf("%d\n",ans);
    }
    return 0;
}