【题目描述】小 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 取模的结果。
输入示例:
5
2 3 3 3
输出示例:
9
思路:使用排列组合的思想,分情况讨论。
代码(已通过全部测试用例)
import java.math.BigInteger;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
//已经通过所有测试用例
public class QSongsSheet {
final static int mod = 1000000007;
public static void main(String[] args) {
Scanner input = new Scanner(System.in);
int len = input.nextInt();
int k1 = 0;
int v1 = 0;
int k2 = 0;
int v2 = 0;
while(input.hasNext()) {
k1 = input.nextInt();
v1 = input.nextInt();
k2 = input.nextInt();
v2 = input.nextInt();
}
List<Integer> list1 = new ArrayList<Integer>();
List<Integer> list2 = new ArrayList<Integer>();
for(int i=0; i<v1; i++) {
list1.add(k1);
}
for(int j=0; j<v2; j++) {
list2.add(k2);
}
int res = getRes(list1,list2, len);
System.out.println(res);
}
private static int getRes(List<Integer> list1, List<Integer> list2, int len) {
BigInteger res = new BigInteger(0 + "");
int sum = 0;
int count1 = 0;
int count2 = 0;
int tmp = 0;
for(int i=0; i<=list1.size(); i++) {
if(i > 0) {
tmp += list1.get(i-1);
count1 = i;
}
if(tmp == len) {
count1 = i;
res = res.add(getRes1(count1, list1));
break;
}
count2 = 0;
sum = tmp;
for(int j=0; j<list2.size(); j++) {
sum += list2.get(j);
if(sum > len) {
break;
}
if(sum == len) {
count2 = (j + 1);
if(i == 0) {
res = res.add(getRes1(count2, list2));
} else {
res = res.add(getRes2(count1, list1, count2, list2));
}
}
}
}
return Integer.valueOf(res.mod(new BigInteger(mod + "")) + "");
}
private static BigInteger getRes1(int flag, List<Integer> list) {
BigInteger b1 = new BigInteger(1 + "");
BigInteger b2 = new BigInteger(1 + "");
BigInteger res = new BigInteger(0 + "");
int v = list.size();
if(flag == 0) {
res = new BigInteger(0 + "");
} else {
for(int i=0; i<flag; i++) {
b1 = b1.multiply(new BigInteger((v - i) + ""));
b2 = b2.multiply(new BigInteger((i + 1) + ""));
}
res = b1.divide(b2);
}
return res;
}
private static BigInteger getRes2(int flag1, List<Integer> list1, int flag2, List<Integer> list2) {
BigInteger b1 = new BigInteger((getRes1(flag1, list1)) + "");
BigInteger b2 = new BigInteger((getRes1(flag2, list2)) + "");
BigInteger tmp = new BigInteger(b1.multiply(b2) + "");
return tmp;
}
}
注意之处:有的测试用例计算中间值就已经很大,超出了long的范围,可以用Java自带的BigInteger存储。