花店摆花问题

 

参考代码

public class baihua {
    private static int n,m,sum;
    private static int[] s;
    public static void main(String[] args) {
        String str = new Scanner(System.in).nextLine();
        n = Integer.parseInt(str.split(" ")[0]);
        m = Integer.parseInt(str.split(" ")[1]);
        s = new int[n];
        str = new Scanner(System.in).nextLine();
        for(int i=0;i<n;i++){
            s[i] = Integer.parseInt(str.split(" ")[i]);
        }
        dg(0,0);
        System.out.println(sum%1000007);
    }
    private static void dg(int index,int num) {
        if(num == m){
            sum++;
        }
        if(index >= n || num >= m){
            return;
        }else{
            for(int i=0;i<=s[index];i++){
                dg(index+1,num+i);
            }
        }
    }
}

运行结果

 

算法解析

dg方法使用了回溯算法，不断尝试。
其中第一个参数为当前花种，因为存储每种花摆放上限的数组s是从0开始的，所有0为第一种花
第二个参数所有花摆放数量总和

注意代码dg方法中代码“for(int i=0;i<=s[index];i++)”，因为有些花可以不摆，只要达到规定的摆放数量即可，所有循环开始时i=0,如果加上必须每种花都要摆的条件的话则可以改为i=1