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

import java.util.*;
import java.io.*;

//借鉴了大佬的代码，加以理解
public class Main
{
    static BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
    static final int N = 100;
    static final int MOD = 1000_000_007;
    
    public static void main(String[] args) throws Exception
    {
        String[] strs = reader.readLine().split(" ");
        int n = Integer.parseInt(strs[0]);
        int m = Integer.parseInt(strs[1]);
        
        int[][][] f = new int[N][N][N];
        f[0][0][2] = 1;

        //n是遇到店的次数，m是遇到花的次数，k是酒的斗数
        for(int i = 0;i <= n;i++)
        {
            for(int j = 0;j <= m;j++)
            {
                if(i == 0 && j == 0) continue;
                //k不可能超过100，因为会喝不完，最多只能遇到100次花表示最多喝100斗酒
                for(int k = 0;k <= 100;k++)
                {
                    //遇到店，k必须为偶数，因为你前一次状态的斗数*2始终是偶数
                    if(k % 2 == 0 && i > 0){
                         //状态转移方程，需要自身状态加上前一次的状态
                         //因为此次遇到的是店，所以上一次是i-1，而遇到店是k*2，所以上一次是k / 2
                         f[i][j][k] += f[i - 1][j][k / 2];
                    }
                    //遇到花
                    //因为此次遇到的是花，所以上一次是j-1，而遇到店是k-1，所以上一次是k+1
                    if(j > 0){
                         f[i][j][k] += f[i][j - 1][k + 1];
                    }
                    f[i][j][k] %= MOD;
                }
            }
        }
        //因为最后一次遇到的是花，所以只需要得到遇到n次店，m-1次花时，酒的斗数为1（因为最后一次遇花会喝完）
        System.out.println(f[n][m - 1][1]);
    }
}