小Q的歌单

题目描述：

小Q有X首长度为A的不同的歌和Y首长度为B的不同的歌，现在小Q想用这些歌组成一个总长度正好为K的歌单，每首歌最多只能在歌单中出现一次，在不考虑歌单内歌曲的先后顺序的情况下，请问有多少种组成歌单的方法。

 

输入描述:

每个输入包含一个测试用例。
每个测试用例的第一行包含一个整数，表示歌单的总长度K(1<=K<=1000)。
接下来的一行包含四个正整数，分别表示歌的第一种长度A(A<=10)和数量X(X<=100)以及歌的第二种长度B(B<=10)和数量Y(Y<=100)。保证A不等于B。

输出描述:

输出一个整数,表示组成歌单的方法取模。因为答案可能会很大,输出对1000000007取模的结果。

 

示例1

输入

5
2 3 3 3

输出

9

解析：

本题的做法可采用01背包的思路。不妨假设歌单的编号为i,所要求的的总歌单长度为j。采用动态规划的思想，对于长度要求为j的总歌单无非只有两种情况即包含歌单i的总歌单与不包含歌单i的总歌单。

可以定义二维数组record[i][j],来存储在前i个歌单中，组成总长度为j的总歌单的方法数。即：

record[i][j]=record[i-1][j-歌单i的长度](包含歌单i)+record[i-1][j](不包含歌单i)

由于整个算法中数组record只用到了i与i-1时的两个状态，因此可以将数组优化成一维数组record[j]来处理。此处内循环中用逆序遍历的原因也是为了在record被更新之前及时获得上一状态的数据。

import java.util.Arrays;
import java.util.Scanner;

public class Qplaylist {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int K = sc.nextInt();
        int A = sc.nextInt();
        int X = sc.nextInt();
        int B = sc.nextInt();
        int Y = sc.nextInt();
        int[] record = new int[K + 1];//记录歌单长度对应的组合数
        int[] size = new int[X + Y + 1];//记录各个歌单的长度
        int mod=1000000007;
        Arrays.fill(size, 1, X+1, A);//初始化各个歌单的长度
        Arrays.fill(size, X+1, X + Y + 1, B);
        record[0]=1;
        for (int i = 1; i <= X + Y; i++) {
            for (int j = K; j >= size[i]; j--) {
                record[j] = (record[j] + record[j - size[i]])%mod;
            }
        }
        System.out.println(record[K]);
    }
}

原题地址