腾讯2018实习编程题第二题

问题：现有m首长度为a的歌曲，和n首长度为b的歌曲。现有组成一个歌单，歌单中所有的歌的总长度为k，同一首歌在一个歌单中只能出现一次。问有多少种满足条件的不同组合的歌单（不考虑歌单中歌曲顺序）。

/**
 * 输入 5
 *     2 3 3 3
 *    第一行5，代表要求的和，也就是歌单中所有的歌的总长度k
 *    第二行第一个数2，表示歌的长度为2，第二个数表示这样的歌有3首，
 *    第二行第三个数3，表示歌的长度为3，第四个数表示这样的有3首
 *    
 * 输出：有多少种满足条件的不同组合的歌单
 * 
 * 思路：类似于求abc的所有组合，如a,b,c,ab,ac,bc,abc
 * 求222333的所有组合中满足歌单总长度为5的组合共有9种，如有一个组合是23，是由数2和3组成，和为5，满足条件
 *
 */

public class test2 {
	static int result=0;//记录满足条件的组合数
	static int sum=0;//一个组合的和，如一个组合“23”，和为2+3=5
	public static void combine(int[] array,int begin,int len,StringBuilder sb,int k) {
		if(len==0) {//当len为0时，表示得到一个组合，接下来计算这个组合的和
			char[] charArray = sb.toString().toCharArray();
			for(int i=0;i<charArray.length;i++) {
				sum+=Integer.valueOf(charArray[i]+"");
			}
			if(sum==k) {
				result++;
			}
			sum =0;//初始化，好计算下一个组合的和
			return;
		}
		if(begin==array.length)//搜索完长度为len的所有组合了
			return;
		sb.append(array[begin]);
		combine(array, begin+1, len-1, sb,k);
		sb.deleteCharAt(sb.length()-1);
		combine(array, begin+1, len, sb,k);
	}
	
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int alen =0;
		int aNum =0;
		int blen =0;
		int bNum =0;
		StringBuilder sb = new StringBuilder();
		int k =Integer.valueOf(scanner.nextLine());
		while(scanner.hasNext()) {
			alen = scanner.nextInt();
			aNum = scanner.nextInt();
			blen = scanner.nextInt();
			bNum = scanner.nextInt();
			break;
		}
		int[] array = new int[aNum+bNum];
		for(int i=0;i<aNum;i++) {
			array[i]=alen;
		}
		for(int i=aNum;i<aNum+bNum;i++) {
			array[i]=blen;
		}

		int len = array.length;
		for(int i=1;i<=len;i++) 
			combine(array,0,i,sb,k);
		System.out.println(result);
	}
}