20180710 考试记录-优快云博客

本文提供两道算法题的解决方案：一是关于强迫症问题，利用动态规划与组合数学解决物品放置问题；二是数格子问题，采用矩阵快速幂优化状态压缩动态规划的方法。

T1 小a的强迫症

dp + 组合数学
code:

//By Menteur_Hxy
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#define LL long long
#define F(i,a,b) for(register int i=(a);i<=(b);i++)
#define R(i,a,b) for(register int i=(b);i>=(a);i--)
using namespace std;

LL rd() {
    LL x=0,f=1; char c=getchar();
    while(!isdigit(c)) {if(c=='-')f=-f; c=getchar();}
    while(isdigit(c)) x=(x<<1)+(x<<3)+c-48,c=getchar();
    return x*f;
}

const int MOD=998244353;
const int N=100010,M=500010;
int n;
int da[N];
LL fac[M],inv[M];

LL qpow(LL a,LL b) {
    LL ret=1;
    while(b) {
        if(b&1) ret=ret*a%MOD;
        a=a*a%MOD;
        b>>=1;
    }
    return ret;
}

void init() {
    fac[1]=1;
    F(i,2,M) fac[i]=fac[i-1]*i%MOD;
    inv[M]=qpow(fac[M],MOD-2);
    R(i,-1,M-1) inv[i]=(inv[i+1]*(i+1))%MOD;
}

LL C(int x,int y) {
    return fac[x]*inv[y]%MOD*inv[x-y]%MOD;
}

int main() {
    freopen("qiang.in","r",stdin);
    freopen("qiang.out","w",stdout);
    n=rd();
    F(i,1,n) da[i]=rd();
    init();
    LL ans=1,sum=da[1];
    F(i,2,n) ans=ans*C(sum+da[i]-1,da[i]-1)%MOD,sum+=da[i];
    printf("%lld",ans);
    return 0;
}

T2 数格子

矩阵快速幂优化状压dp
code:

//By Menteur_Hxy
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#define LL long long
#define F(i,a,b) for(register int i=(a);i<=(b);i++)
#define R(i,a,b) for(register int i=(b);i>=(a);i--)
using namespace std;

LL rd() {
    LL x=0,f=1; char c=getchar();
    while(!isdigit(c)) {if(c=='-')f=-f; c=getchar();}
    while(isdigit(c)) x=(x<<1)+(x<<3)+c-48,c=getchar();
    return x*f;
}

const int MOD=998244353;
const int N=100010,M=500010;
int n;
int da[N];
LL fac[M],inv[M];

LL qpow(LL a,LL b) {
    LL ret=1;
    while(b) {
        if(b&1) ret=ret*a%MOD;
        a=a*a%MOD;
        b>>=1;
    }
    return ret;
}

void init() {
    fac[1]=1;
    F(i,2,M) fac[i]=fac[i-1]*i%MOD;
    inv[M]=qpow(fac[M],MOD-2);
    R(i,-1,M-1) inv[i]=(inv[i+1]*(i+1))%MOD;
}

LL C(int x,int y) {
    return fac[x]*inv[y]%MOD*inv[x-y]%MOD;
}

int main() {
    freopen("qiang.in","r",stdin);
    freopen("qiang.out","w",stdout);
    n=rd();
    F(i,1,n) da[i]=rd();
    init();
    LL ans=1,sum=da[1];
    F(i,2,n) ans=ans*C(sum+da[i]-1,da[i]-1)%MOD,sum+=da[i];
    printf("%lld",ans);
    return 0;
}

转载于:https://www.cnblogs.com/Menteur-Hxy/p/9290314.html