腾讯面试题之小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 取模的结果。
输入示例:
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存储。

猜你喜欢

转载自blog.csdn.net/u010592112/article/details/81568168