CF1697E Coloring

注意到很多人题解都涉及到了，点集最多为4这个性质。

而我读完题解后想到的写法并没有涉及这一点，所以写下这篇

bitset预处理与i距离最近点集，然后check点集内所有点是不是都满足条件

最后做一个01背包，算出颜色种类为k的方案数，最后给答案是sigma dp[k]*fac[n]*finv[n-k].

复杂度瓶颈是 (n^3)/64 其中n为100,可以粗略当作为n^2

#include<bits/stdc++.h>
using namespace std;
const int P=998244353,N=105,inf=1e9+10;
using ll=long long;
struct DSU
{
    vector<int>f,siz;
    DSU(int n):f(n),siz(n,1){iota(f.begin(),f.end(),0);}
    int leader(int x)
    {
        while(x!=f[x])x=f[x]=f[f[x]];
        return x;
    }
    bool same(int x,int y){return leader(x)==leader(y);}
    bool merge(int x,int y)
    {
        x=leader(x);
        y=leader(y);
        if(x==y)return false;
        siz[x]+=siz[y];
        f[y]=x;
        return true;
    }
    int size(int x) {return siz[leader(x)]; }
};
ll finv[N],fac[N];
ll POW(ll x,int k=P-2,ll rs=1){while(k){if(k&1)rs=rs*x%P;x=x*x%P;k>>=1;}return rs;}
void init()
{
    finv[1]=finv[0]=fac[0]=1;
    for(int i=1;i<N;++i)fac[i]=fac[i-1]*i%P;
    finv[N-1]=POW(fac[N-1]);
    for(int i=N-2;i;--i)finv[i]=finv[i+1]*(i+1)%P;
}
int A(int a,int b){return fac[a]*finv[a-b]%P;}
array<int,2>p[N];
bitset<N>bit[N];
int cal(int i,int j){return abs(p[i][0]-p[j][0])+abs(p[i][1]-p[j][1]);}
void add(int &a,int b){a+=b;if(a>=P)a-=P;}


int f[N],g[N];
void solve()
{
    init();
    int n;scanf("%d",&n);
    for(int i=1,x,y;i<=n;++i)scanf("%d%d",&x,&y),p[i]={x,y};
    for(int i=1;i<=n;++i)
    {
        int dis=inf;
        for(int j=1;j<=n;++j)if(i!=j)dis=min(dis,cal(i,j));
        
        for(int j=1;j<=n;++j)if(i!=j&&dis==cal(i,j))
        bit[i][j]=1;
        bit[i][i]=1;
    }
    DSU dsu(n+1);
    for(int i=1;i<=n;++i)
    {
        if(dsu.size(i)>1)continue;
        bool fg=1;
        for(int j=1;j<=n;++j)
        if(bit[i][j]){if(bit[i]!=bit[j]){fg=0;break;}}
        
        if(fg)for(int j=1;j<=n;++j)if(bit[i][j])dsu.merge(i,j);
    }
    vector<int>vec;
    for(int i=1;i<=n;++i)if(dsu.leader(i)==i)vec.push_back(dsu.size(i));
    
    int rs=0;
    f[0]=1;
    for(auto t:vec)
    {
        swap(f,g);memset(f,0,sizeof(f));
        for(int i=1;i<=n;++i)
        {if(i>=t)add(f[i],g[i-t]);if(t!=1)add(f[i],g[i-1]);}
    }
    for(int i=1;i<=n;++i)add(rs,1ll*f[i]*A(n,i)%P);
    cout<<rs;
}
int main()
{
    int T=1;//scanf("%d",&T);
    while(T--)solve();
}