腾讯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);
}
}