题目 2662: 李白打酒加强版

题目描述:

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

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

无事街上走，提壶去打酒。

逢店加一倍，遇花喝一斗。

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

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

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

代码:

package lanqiao;

import java.math.BigInteger;
import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt(); //遇见店的次数，酒翻倍
        int M = sc.nextInt(); //遇见花的次数，喝一斗
        //状态表示:f(i, j, k) 表示 看遇到了i次花 j次店 时还剩k斗酒 的所有方案的集合

        int mod = (int)1e9 + 7;
        int[][][] f = new int[M + 1][N + 1][M + 1];
        //初始化1，只遇到了花
        f[0][0][2] = 1;
        f[1][0][1] = 1;
        f[2][0][0] = 1;

        //初始化2 只遇到了店(k不能超过m)
        for(int i = 1;i <= N && Math.pow(2,(i + 1)) <= M;i ++)
        {
            f[0][i][(int)Math.pow(2,(i + 1))] = 1;
        }

        for(int i = 1;i <= M;i ++)
        {
            for(int j = 1;j <= N;j ++)
            {
                for(int k = 0;k <= M;k ++)
                {
                    if(k < M)
                    {
                        f[i][j][k] += f[i - 1][j][k + 1];
                    }

                    if(k % 2 == 0)
                    {
                        f[i][j][k] += f[i][j - 1][k / 2];
                    }
                    f[i][j][k] %= mod;
                }
            }
        }
        System.out.println(f[M - 1][N][1]);
    }
}