7-1 自然数拆分的方案数 (10 分)
题目
输入格式
第一行自然整数T,表示之后测试数据组数, 以后T行,每行一个自然数N,(1<N<=4000)
注意:本题规模为4000,回溯法还合适吗?仔细思考应该如何设计算法
建议自行学习完全背包相关知识。
输出格式
T行,每行输出一个整数,表示拆分的方案数,结果对2147483648取模。
输入样例
2
6
7
输出样例
11
15
基本思路
这个问题可以等价转换成完全背包问题:将n个物品放入背包(容量为n的方案数)
- dp[i][j]表示:从前i个数中选取,使得体积刚好为j的方案数量
- dp[i][j]=dp[i-1][j](第i个物品选择0次)+dp[i-1][j-Vi](第i个物品选择1次)+dp[i-1][j-2Vi](第i个物品选择2次)+…+dp[i-1][j-mVi](第i间物品选择m次,m=j/vi)
用j-Vi代替j,得到:dp[i][j-Vi]=dp[i-1][j-Vi](第i个物品选择1次)+dp[i-1][j-2Vi](第i个物品选择2次)+…+dp[i-1][j-mVi](第i间物品选择m次,m=j/vi)
上面两式相减,得到:dp[i][j] = dp[i-1][j]+dp[i][j-vi] - 从上式可以看出,求解dp[i][j]时,只会用到dp[i-1][j]和dp[i][j-vi]。只要正向枚举i和j,那么求解dp[i][j]就和一维的i无关了,所以,可以把第一维省略(这一技巧也叫做滚动数组)。 得到:dp[j] = dp[j] +dp[j-vi](vi= =i)
代码
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
int k;
int n[4001];//存储每一组测试样例的背包容量
long long dp[4001];//dp[j]表示在前i个物品中选取体积恰好为j时所获得方案数量(滚动数组,i隐含)
int main(){
cin>>k;
//读入所有的测试样例,记录下最大的背包容量max,对它进行动态规划
//这样,在求解最大背包容量max的方案数量的过程中,顺便就把更小背包容量的方案数量求出来了
int max=0;
for(int i=0;i<k;i++){
cin>>n[i];
if(max<n[i]) max=n[i];
}
//递推边界
dp[0]=1;
//状态转移方程
//因为最大背包容量为max,说明物品编号为1~max(物品体积等于编号),所以i从第一个物品开始枚举(第一个状态),到第max个物品结束(第max个状态)
//在第i个状态下,说明在此状态下能选的物品编号为1~i(物品体积等于编号)
for(int i=1;i<=max;i++){//枚举物品从1到max
//因为能选体积为i的物品,所以容量j从i开始枚举(确保能放下体积为i的物品),到最大背包容量max结束
for(int j=i;j<=max;j++){//枚举容量从i到max
dp[j]=(dp[j]+dp[j-i])%2147483648;
}
}
//输出
for(int i=0;i<k;i++){
cout<<dp[n[i]]<<endl;
}
}