选课时间(题目已修改,注意读题)
Problem Description
又到了选课的时间了,xhd看着选课表发呆,为了想让下一学期好过点,他想知道学n个学分共有多少组合。你来帮帮他吧。(xhd认为一样学分的课没区别)
Input
输入数据的第一行是一个数据T,表示有T组数据。
每组数据的第一行是两个整数n(1 <= n <= 40),k(1 <= k <= 8)。
接着有k行,每行有两个整数a(1 <= a <= 8),b(1 <= b <= 10),表示学分为a的课有b门。
Output
对于每组输入数据,输出一个整数,表示学n个学分的组合数。
Sample Input
2
2 2
1 2
2 1
40 8
1 1
2 2
3 2
4 2
5 8
6 9
7 6
8 8
Sample Output
2
445
代码:
# include <iostream>
# include <cstring>
using namespace std;
int main(){
int T,n,k,i,j,p;
int a[10],b[10];//学分为a的b门
int c1[10000],c2[10000];
while(cin>>T){
while(T--){
cin>>n>>k;//统计组合学分为n的,k行
for(i=0;i<k;i++){
cin>>a[i]>>b[i];
}
memset(c1,0,sizeof(c1));
memset(c2,0,sizeof(c2));
for(i=0;i<=a[0]*b[0];i+=a[0]){
c1[i] = 1;
}
int len = a[0]*b[0];
for(i=1;i<k;i++){
for(j=0;j<=len;j++){//第i-1项
for(p=0;p<=b[i]*a[i];p+=a[i]){//第i项
c2[j+p] += c1[j];
}
}
len += b[i]*a[i];
for(p=0;p<=len;p++){
c1[p] = c2[p];
c2[p] = 0;
}
}
for(i=1;i<len;i++){
if(i==n){
cout<<c1[i]<<endl;
break;
}
}
}
}
return 0;
}