李白打酒加强版

P8786 [蓝桥杯 2022 省 B] 李白打酒加强版

题目描述

话说大诗人李白，一生好饮。幸好他从不开车。

一天，他提着酒壶，从家里出来，酒壶中有酒 $2$ 斗。他边走边唱：

无事街上走，提壶去打酒。
逢店加一倍，遇花喝一斗。

这一路上，他一共遇到店 $N$ 次，遇到花 $M$ 次。已知最后一次遇到的是花，他正好把酒喝光了。

请你计算李白这一路遇到店和花的顺序，有多少种不同的可能?

注意：壶里没酒（$0$ 斗）时遇店是合法的，加倍后还是没酒；但是没酒时遇花是不合法的。

输入格式

第一行包含两个整数 $N$ 和 $M$。

输出格式

输出一个整数表示答案。由于答案可能很大，输出模 $1000000007$（即 $10^9+7$）的结果。

输入输出样例 #1

输入 #1

5 10

输出 #1

14

说明/提示

【样例说明】

如果我们用 0 代表遇到花，1 代表遇到店，$14$ 种顺序如下:

010101101000000
010110010010000
011000110010000
100010110010000
011001000110000
100011000110000
100100010110000
010110100000100
011001001000100
100011001000100
100100011000100
011010000010100
100100100010100
101000001010100

【评测用例规模与约定】

对于 $40\%$ 的评测用例：$1\le N,M\le 10$。

对于 $100\%$ 的评测用例：$1\le N,M\le 100$。

蓝桥杯 2022 省赛 B 组 I 题。

代码内容

// #include <iostream>
// #include <algorithm>
// #include <cstring>
// #include <stack>//栈
// #include <deque>//堆
// #include <queue>//队列
// #include <map>//映射
// #include <unordered_map>//哈希表
// #include <vector>//容器，存数组的数，表数组的长度
#include <bits/stdc++.h>
using namespace std;

#define endl '\n'
typedef long long ll;
const ll mod=1e9+7;
const ll N=2e2+10;
ll a[N][N][N];
 
void solve()
{
    ll n,m;
    cin>>n>>m;
    
    a[0][0][2]=1;
    for(ll i=0;i<n+m;i++)
        for(ll j=0;j<m;j++)
            for(ll k=0;k<=m;k++)
            {
                if(a[i][j][k])
                {
                    if(k) (a[i+1][j+1][k-1]+=a[i][j][k])%=mod;//遇花，酒减一
                    if(k<=50) (a[i+1][j][2*k]+=a[i][j][k])%=mod;//遇店，酒双倍
                }
            }
            
    cout<<a[n+m][m][0]<<endl;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);

    ll t=1;
    //cin>>t;
    while(t--) solve();
    
    return 0;
}