9284:盒子与小球之二 —— C++

问题描述

代码

#include<iostream>
using namespace std;

int N,A,B;
int box[21];
long long cnt,ans;

int sum(int i){
	int temp=0;
	for(int k=1;k<=i;++k){
		temp+=box[k];
	}
	return temp;
}

void f(int i,int value){
	if(i==N+1){
		if(sum(N)<=value){
			++cnt;
		}
		return;
	}
	for(int k=0;k<=value;++k){
		if(sum(i-1)+k<=value){
			box[i]=k;
			f(i+1,value);
			box[i]=0;
		}
	}
}

int main(){
	cin>>N>>A>>B;
	f(1,A);
	ans=cnt;
	cnt=0;
	f(1,B);
	ans*=cnt;
	cout<<ans;
	return 0;
}