最近练习了下算法题,里面有比较多的技巧,而一些思路转化为程序语言,也需要多练习实现方式。
动态规划,完全背包问题例子
在以前的选票问题讨论中,使用了0-1背包方式的规划实现,同样是数字和问题,如果每个数字可以多次使用,会是怎样的结果呢?
https://leetcode-cn.com/problems/combination-sum/
还是类似的处理,这里的转换公式需要理解一下,
f(i, j) = f(i-1, j) + f(i, j-v[i])
前i个数字组成和为j的总次数,分为2中情形,
1.组合中不包含第i个数字,即前i-1个数字组成和为j
2. 组合中肯定包含第i个数字,即前i个数字组成j-v[i], f(i, j-v[i])
这个分化处理很巧妙,画一画可以加深这个理解。
知道了组成方式,就可以按照一个个的v[i]数据来找到具体集合数据。
class Solution {
public:
int len;
int w;
int* d;
vector<vector<int>> ret;
int getv(int i, int j) {
if (i < 0) return 0;
if (j < 0) return 0;
return d[i*w + j];
}
void setv(int i, int j, int v) {
d[i*w + j] = v;
}
void print() {
/*
for (int i = 0; i < len; i++)
{
for (int j = 0; j < w; j++) {
printf("%d ", getv(i, j));
}
printf("\n");
}*/
}
void init(vector<int>& v) {
//
for (int i = 0; i < len; i++)
{
for (int j = 0; j < w; j++)
{
setv(i, j, 0);
}
}
for (int i = 0; i < len; i++)
{
setv(i, 0, 1);
}
print();
//f(i, j) = f(i-1, j) + f(i, j-v[i])
for (int i = 0; i < len; i++)
{
for (int j = 1; j < w; j++) {
int t = getv(i-1, j) + getv(i, j - v[i]);
setv(i, j, t);
}
}
print();
}
vector<vector<int>> join(int v, vector<vector<int>>& r) {
for (unsigned i = 0; i < r.size(); i++)
{
r[i].push_back(v);
}
return r;
}
void unionv(vector<vector<int>>& a, vector<vector<int>>& b) {
for (unsigned i = 0; i < b.size(); i++)
{
a.push_back(b[i]);
}
}
f(i, j) = f(i-1, j) + f(i, j-v[i])
vector<vector<int>> getr(int i, int j, vector<int>& v) {
vector<vector<int>> r;
if (j == 0)
{
vector<int> a;
r.push_back(a);
return r;
}
if (getv(i-1, j) > 0)
{
vector<vector<int>> t = getr(i-1, j, v);
unionv(r, t);
}
if (getv(i, j-v[i]) > 0)
{
vector<vector<int>> t = getr(i, j-v[i], v);
join(v[i], t);
unionv(r, t);
}
return r;
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
len = candidates.size();
w = target+1;
d = new int[len*(target+1)];
init(candidates);
if (getv(len-1, target) == 0)
{
return ret;
}
ret = getr(len-1, target, candidates);
return ret;
}
};