POJ 1664 求m个苹果放入n个盘子的不同放法数目
题目描述 Description
把M个同样的苹果放在N个同样的盘子里,允许有的盘子空着不放,问共有多少种不同的分法?(用K表示)5,1,1和1,5,1 是同一种分法。
输入描述 Input Description
第一行是测试数据的数目t(0 <= t <= 20)。以下每行均包含二个整数M和N,以空格分开。1<=M,N<=10。
输出描述 Output Description
对输入的每组数据M和N,用一行输出相应的K。
样例输入 Sample Input
14 2
样例输出 Sample Output
3
解法一、递归
[解题思路]
设f(m,n)为m个苹果,n个盘子的放法数目,则先对n作讨论,当n>m:则必定有n-m个盘子永远空着,去掉它们对摆放苹果方法数目不产生影响。即 if(n>m) f(m,n) = f(m,m)
当n <= m,不同的放法可以分成两类:含有0的方案数,不含有0的方案数
1、含有0的方案数,即有至少一个盘子空着,即相当于 f(m,n)=f(m,n-1);
2、不含有0的方案数,即全部盘子都有苹果,那么先从m个苹果中抽取出n个出来,各个盘子分一个,考虑剩下的m-n个苹果放到n个盘子里的放法,即 f(m,n)=f(m-n,n).
而总的放苹果的放法数目等于两者的和,即 f(m,n)=f(m,n-1)+f(m-n,n)
递归出口条件说明:
当n=1时,所有苹果都必须放在一个盘子里,所以返回1;
当m==0(没有苹果可放)时,定义为1种放法;
[代码实现]
int f(int m, int n) //m个苹果放在n个盘子中共有几种方法
{
if(m==0 || n==1)
return 1;
if(n>m)
return f(m,m);
else
return f(m,n-1)+f(m-n,n);
}
int main()
{
int m,n;
cin>>m>>n;
cout<<f(m,n)<<endl;
return 0;
}
解法二、深度搜索
[解题思路]
可以把M个苹果放在N个盘子的所有方案都保存起来(其中N1+N2+...+N=M,Ni为第i个盘子中苹果的数),通过排序,去重,统计有多少种方案即为答案,同样通过STL里的vector会方便很多。
[代码实现]
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int M,N,k;/* M苹果 N盘子 k种排法 */
int a[10]; /* 每个盘子放的苹果数量 */
vector<vector<int>> ans;/* 存放结果 去重 */
bool IsM() /* 判断是sum == M */
{
int sum=0;
for(int i=0;i<N;i++) sum+=a[i];
return sum==M? true:false;
}
bool IsRepeat() /* 检查是否重复 */
{
vector<int> Temp(N); /* 存放N个盘子里的 */
copy(begin(a),a+N,Temp.begin());
std::sort(Temp.begin(),Temp.end());
for(vector<vector<int>>::iterator it=ans.begin();it!=ans.end();it++)
{
if(Temp==(*it)) return false;
}
ans.push_back(Temp); /* 否则插入ans中 */
return true;
}
void dfs(int step)
{
if(step == N)
{
if(IsM() && IsRepeat()) k++;
return ;
}
for(int i=0;i<=M;i++)
{
a[step] = i;
dfs(step + 1);
}
}
int main()
{
cin>>M>>N;
dfs(0);
cout<<k<<endl;
return 0;
}