其他分享
首页 > 其他分享> > 抽屉上锁(计数dp+高精度)

抽屉上锁(计数dp+高精度)

作者:互联网

题目描述
现在我们来考虑对一个有N层抽屉的柜子上锁。该柜子中的所有抽屉都整齐地排布在一列上,并且相邻上下两个抽屉之间没有木板分隔。也就是说,如果第一层的抽屉没有锁上,即使第二层的抽屉锁上了,我们也能够拿到第二层抽屉里面的东西(将第一层的抽屉抽出来)。我们的问题是,如果要求恰好只有M层抽屉里面的东西拿不到,总共有多少种不同的锁抽屉方式呢?
输入
输入包括两个数N和M,意义如题目中所描述。
输出
输出仅包含一个整数,即答案。
样例输入
6 4
样例输出
6
提示
对于20%的数据,有1≤N, M≤10;
对于50%的数据,有1≤N, M≤65;
对于100%的数据,有1≤N, M≤300。
解题思路:每个抽屉有锁和不锁两种状态,而且这层抽屉能否拿出只与自己和上一层抽屉的状态有关,所以想到dp。
dp[i][j][0]表示i层抽屉有j个拿不到并且这层抽屉是没锁的,dp[i][j][1]表示表示i层抽屉有j个拿不到并且这层抽屉是上锁的。
状态转移方程位(根据方程自己看一下很容易就理解了):
dp[i][j][1]=dp[i-1][j-1][1],dp[i-1][j][0];
dp[i][j][0]=dp[i-1][j][0],dp[i-1][j][1];
注意:在计数过程中,因数太大会导致爆炸,所以采用高精度

#include<bits/stdc++.h>
//#pragma GCC optimize(2)
#define N 100005
using namespace std;
typedef long long int ll;
typedef unsigned long long int ull;
const int mod=1e9+7;
const int INF=0x7fffffff;
const int inf=0x3f3f3f3f;
const ull base=131;
//高精加
string jia(string a,string b){
    string s="";
    int aa[1100],bb[1100],cc[1100];
    memset(aa,0,sizeof(aa));
    memset(bb,0,sizeof(bb));
    memset(cc,0,sizeof(cc));
    int lena=a.size(),lenb=b.size();
    for(int i=1;i<=lena;i++) aa[i]=a[lena-i]-'0';
    for(int i=1;i<=lenb;i++) bb[i]=b[lenb-i]-'0';
    int len=1;
    while(len<=lena||len<=lenb){
        cc[len]+=aa[len]+bb[len];
        if(cc[len]>=10){
            cc[len+1]=cc[len]/10;
            cc[len]%=10;
        }
        len++;
    }
    while(!cc[len]&&len>1) len--;
    for(int i=len;i>=1;i--) s+=cc[i]+'0';
    return s;
}

string dp[310][310][2];
int n,m;
int main(){
    cin>>n>>m;
    dp[0][0][1]=dp[1][0][0]=dp[1][1][1]="1";
    for(int i=2;i<=n;i++) dp[i][0][0]=jia(dp[i-1][0][0],dp[i-1][0][1]),dp[i][0][1]=dp[i-1][0][0];//初始化边界
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m&&i>=j;j++){
            dp[i][j][1]=jia(dp[i-1][j-1][1],dp[i-1][j][0]);
            dp[i][j][0]=jia(dp[i-1][j][0],dp[i-1][j][1]);
        }
    }
    cout<<jia(dp[n][m][0],dp[n][m][1]);
    return 0;
}

标签:string,高精度,int,len,cc,抽屉,上锁,dp
来源: https://blog.csdn.net/qq_50382040/article/details/119207996