luogu P4931 情侣？给我烧了！

Posted 2021-02-09 smyjr

双倍经验

传送门

首先坐在一起的cp和不坐在一起的cp是相对独立的,可以分开考虑,然后方案数相乘

坐在一起的cp,方案为(inom{n}{k}*inom{n}{k}*k!*2^k).首先选出k排座位,然后选出k对cp坐下,然后这k对可以任意打乱顺序,并且每一对可以换座位

不坐在一起的cp,一开始以为就是个错排,然后错排其实是没考虑一对cp坐在同一列的情况的.考虑递推,设(f[i])表示i个不坐在一起的cp的方案.首先可以选出两个不是一对cp的人,方案为(2i*(2i-2)),然后钦定他们坐第一排.然后考虑这两个人的另一半(严谨)

• 另外的那两个人坐一排,一个人可以坐(2x-2)个位置,第二个只能坐旁边,那么剩下的人的方案就是i-2对的方案,所以总方案为((2x-2)*f[i-2])

• 他们不坐一排,那么可以先假设他们是cp(逃,然后就是i-1对的方案,即(f[i-1])

然后不坐一起cp的方案为(2i*(2i-2)*(f[i-1]+(2x-2)*f[i-2]))

然后预处理阶乘以及逆元,还有f和2的次幂,乘起来就没了

#include<bits/stdc++.h>
#define LL long long
#define db double
#define il inline
#define re register

using namespace std;
const int N=5e6+10,mod=998244353;
il int rd()
{
    int x=0,w=1;char ch=0;
    while(ch<'0'||ch>'9') {if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') {x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
    return x*w;
}
il int fpow(int a,int b){int an=1;while(b){if(b&1) an=1ll*an*a%mod;a=1ll*a*a%mod,b>>=1;}return an;}
int fac[N],iac[N],d[N],pw[N];
il int C(int n,int m){return (m<0||n<m)?0:1ll*fac[n]*iac[m]%mod*iac[n-m]%mod;}

int main()
{
    //fffffffffffffffffffff
    fac[0]=1;
    for(int i=1;i<=N-10;++i) fac[i]=1ll*fac[i-1]*i%mod;
    iac[N-10]=fpow(fac[N-10],mod-2);
    for(int i=N-10;i;--i) iac[i-1]=1ll*iac[i]*i%mod;
    pw[0]=1;
    for(int i=1;i<=N-10;++i) pw[i]=(pw[i-1]<<1)%mod;
    d[0]=1;
    for(int i=2;i<=N-10;++i) d[i]=1ll*2*i%mod*2*(i-1)%mod*(d[i-1]+1ll*2*(i-1)*d[i-2]%mod)%mod;
    int T=rd();
    while(T--)
    {
        int n=rd(),k=rd();
        printf("%lld
",1ll*C(n,k)*C(n,k)%mod*fac[k]%mod*pw[k]%mod*d[n-k]%mod);
    }
    return 0;
}