【洛谷U20626】gemo 容斥 FWT 高斯消元

题目大意

  给你一个无向图,有 m m 个询问,每次给你一个点x和一个点集 S S ,问你从x开始走,每次从一个点随机的走到与这个点相邻的点,问你访问 S S 中每个点至少一次的期望步数是多少。

  n18,m100000

题解

  有个东西叫min-max容斥:

max(S)=TS(1)|T|+1min(T) max ( S ) = ∑ T ⊆ S ( − 1 ) | T | + 1 min ( T )

  这道题中, min(S) min ( S ) 是从点 x x 开始走,走到S中任意一个点的期望步数。 max(S) max ( S ) 是从 x x 开始走,走完S中全部点的期望步数。

  我们先枚举所有 S S ,用高斯消元算出min(S)

  总共有 2n 2 n 个集合,消一次是 O(n3) O ( n 3 ) 的。

  这样就算出了所有起点为 i i ,遍历S中每个点至少一次的期望步数 fi,S f i , S

  然后带入这个式子里,用FWT算出 max(S) max ( S )

  总共有 n n 个不同的起点,FWT一次是O(n2n)的。

  所以总的时间复杂度是 O(n32n) O ( n 3 2 n ) 的。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath>
using namespace std;
typedef long long ll;
const ll p=998244353;
vector<int> g[20];
int n,m;
int d[20];
int a[19][20];
ll inv[10010];
int c[10010];
ll fp(ll a,ll b)
{
    ll s=1;
    for(;b;b>>=1,a=a*a%p)
        if(b&1)
            s=s*a%p;
    return s;
}
int f[19][1<<18];
int bitcnt[1<<18];
void add(int &a,int b)
{
    if((a+=b)>p)
        a-=p;
}
void fwt(int *a,int n)
{
    int i,j,k;
    for(i=2;i<=1<<n;i<<=1)
        for(j=0;j<1<<n;j+=i)
            for(k=j;k<j+i/2;k++)
                add(a[k+i/2],a[k]);
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("gemo.in","r",stdin);
    freopen("gemo.out","w",stdout);
#endif
    scanf("%d%d",&n,&m);
    int x,y;
    int i,j;
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        d[x]++;
        d[y]++;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    inv[0]=inv[1]=1;
    for(i=2;i<=10000;i++)
        inv[i]=-p/i*inv[p%i]%p;
    ll v;
    int k;
    int l;
    int all=(1<<n)-1;
    bitcnt[0]=0;
    for(i=1;i<all;i++)
        bitcnt[i]=bitcnt[i>>1]+(i&1);
    for(l=1;l<all;l++)
    {
        memset(a,0,sizeof a);
        for(i=1;i<=n;i++)
        {
            a[i][i]=1;
            if(l&(1<<(i-1)))
                continue;
            a[i][n+1]=1;
            for(auto v:g[i])
                a[i][v]=-inv[d[i]];
        }
        for(i=1;i<=n;i++)
        {
            for(j=i;j<=n;j++)
                if(a[i][j])
                    break;
            if(j>n)
                continue;
            if(i!=j)
                for(k=i;k<=n+1;k++)
                    swap(a[i][k],a[j][k]);
            if(a[i][i])
            {
                v=fp(a[i][i],p-2);
                for(j=i;j<=n+1;j++)
                    a[i][j]=a[i][j]*v%p;
            }
            for(j=1;j<=n;j++)
                if(i!=j&&a[j][i])
                {
                    v=a[j][i];
                    for(k=i;k<=n+1;k++)
                        a[j][k]=(a[j][k]-a[i][k]*v)%p;
                }
        }
        for(i=1;i<=n;i++)
            f[i][l]=a[i][n+1];
    }
    for(i=1;i<=n;i++)
    {
        for(j=0;j<all;j++)
        {
            if(!(bitcnt[j]&1))
                f[i][j]=-f[i][j];
            if(f[i][j]<0)
                f[i][j]+=p;
        }
        fwt(f[i],n);
    }
    int m;
    scanf("%d",&m);
    for(i=1;i<=m;i++)
    {
        scanf("%d",&x);
        int s=0;
        for(j=1;j<=x;j++)
        {
            scanf("%d",&y);
            s|=1<<(y-1);
        }
        scanf("%d",&y);
        printf("%d\n",f[y][s]);
    }
    return 0;
}
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值