背包问题 – 有限制的选择问题
链接: 差分
一、背包DP
1. 01背包
问题描述
有 N件物品和一个容量是 V的背包。每件物品只能使用一次。
第 i件物品的体积是 vi,价值是 wi。
求解将哪些物品装入背包,可使这些物品的总体积不超过背包容量,且总价值最大。
输出最大价值。
输入格式
第一行两个整数,N,V,用空格隔开,分别表示物品数量和背包容积。
接下来有 N 行,每行两个整数 vi,wi,用空格隔开,分别表示第 i 件物品的体积和价值。
输出格式
输出一个整数,表示最大价值。
数据范围
0<N,V≤1000
0<vi,wi≤1000
输入样例
4 5
1 2
2 4
3 4
4 5
输出样例:
8
解题思路:
f(i, j)是集合的某种属性(最大值)-- 数。
代码
#include<iostream>
using namespace std;
const int N = 1010;
int n, m;
int v[N], w[N];
int f[N][N];
int main()
{
cin >> n >> m;
//初始状态f[0][0] = 0,从i = 1开始决策。
for(int i = 1; i <= n; i++)
cin >> v[i] >> w[i];
for(int i = 1; i <= n; i++)
for(int j = 1; j <= m; j++)
{
if(v[i] > j) f[i][j] = f[i - 1][j];
else f[i][j] = max(f[i - 1][j], f[i - 1][j - v[i]] + w[i]);
}
cout << f[n][m] << endl;
return 0;
}
使用滚动数组优化
观察递推公式不难发现,dp 数组中第 i 行的元素仅由第 i − 1 行的元素得来,即第 0 行元素的更新值放到第 1 行,第 1 行元素的更新值放到第 2 行,以此类推。与其把一行的更新值放到新的一行,不如直接就地更新,因此我们的 dp 数组只需要一行来存储,即一维数组。
原先 j 是从 1 遍历至 m 的,现在只需从 w[i] 遍历至 m 。但,这个遍历顺序真的对吗?
到此为止,可以总结出,当 dp 数组是二维数组时,j 既可以从小到大遍历也可以从大到小遍历,但当 dp 数组是一维数组时,j只能从大到小遍历。
背包问题的优化只是再原代码上做出改动,大概逻辑不变。为什么要倒过来遍历j,因为如果正序遍历,由于j - v[i] < j,当我们更新f[j]时我们已经把f[j - v[i]]更新了。也就是说我们使用的是第[i]个循环里面的f[j - v[i]],而我们实际要用的是第[i-1]的个循环里面的f[j - v[i]]
顺序的话,j - v[i]是刚刚i层遍历的j - v[i],而不是i - 1层的j - v[i],如果倒序的和j - v[i]第i层没被算过,所以存的是i - 1层的j - v[i]。
代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 1010;
int n, m;
int v[N], w[N];
int f[N];
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i++) cin >> v[i] >> w[i];
for(int i = 1; i <= n; i++)
for(int j = m; j >= v[i]; j--)
f[j] = max(f[j], f[j - v[i]] + w[i]);
cout << f[m] << endl;
return 0;
}
2. 完全背包
有 N 种物品和一个容量是 V 的背包,每种物品都有无限件可用。
第 i 种物品的体积是 vi,价值是 wi。
求解将哪些物品装入背包,可使这些物品的总体积不超过背包容量,且总价值最大。输出最大价值。
输入格式
第一行两个整数,N,V,用空格隔开,分别表示物品种数和背包容积。
接下来有 N 行,每行两个整数 vi,wi,用空格隔开,分别表示第 i 种物品的体积和价值。
输出格式
输出一个整数,表示最大价值。
数据范围
0<N,V≤1000
0<vi,wi≤1000
输入样例
4 5
1 2
2 4
3 4
4 5
输出样例:
10
完全背包是根据f[i][j - v]+ w来决定的,所以j从小到大遍历就行。
代码
#include<iostream>
using namespace std;
const int N = 1010;
int n, m;
int v[N], w[N];
int f[N][N];
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i++)
cin >> v[i] >> w[i];
for(int i = 1; i <= n; i++)
for(int j = 1; j <= m; j++)
{
if(v[i] > j) f[i][j] = f[i - 1][j];
else f[i][j] = max(f[i - 1][j], f[i][j - v[i]] + w[i]);
}
cout << f[n][m] << endl;
return 0;
}
一维优化代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 1010;
int n, m;
int v[N], w[N];
int f[N];
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i++) cin >> v[i] >> w[i];
for(int i = 1; i <= n; i++)
for(int j = v[i]; j <= m; j++)
f[j] = max(f[j], f[j - v[i]] + w[i]);
cout << f[m] << endl;
return 0;
}
3. 多重背包
有 N 种物品和一个容量是 V 的背包。
第 i 种物品最多有 si 件,每件体积是 vi,价值是 wi。
求解将哪些物品装入背包,可使物品体积总和不超过背包容量,且价值总和最大。
输出最大价值。
输入格式
第一行两个整数,N,V,用空格隔开,分别表示物品种数和背包容积。
接下来有 N 行,每行三个整数 vi,wi,si,用空格隔开,分别表示第 i种物品的体积、价值和数量。
输出格式
输出一个整数,表示最大价值。
数据范围
0<N≤1000
0<V≤2000
0<vi,wi,si≤2000
提示:
本题考查多重背包的二进制优化方法。
输入样例
4 5
1 2 3
2 4 1
3 4 3
4 5 2
输出样例:
10
二进制的优化方式:
把第i个物品打包1, 2, 4, 8,… ,每组只能选一个,然后把所有可能性都存了一个数组中,构造01背包问题。
代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 25000;
int n, m;
int v[N], w[N];
int f[N];
int main()
{
int cnt = 0;
cin >> n >> m;
for(int i = 1; i <= n; i++)
{
int a, b, s;
cin >> a >> b >> s;
int k = 1;
while(k <= s)
{
cnt++;
v[cnt] = a * k;
w[cnt] = b * k;
s -= k;
k *= 2;
}
if(s > 0)
{
cnt++;
v[cnt] = a * s;
w[cnt] = b * s;
}
}
n = cnt;
for(int i = 1; i <= n; i++)
for(int j = m; j >= v[i]; j--)
f[j] = max(f[j], f[j - v[i]] + w[i]);
cout << f[m] << endl;
return 0;
}
4. 分组背包
有 N 组物品和一个容量是 V 的背包。
每组物品有若干个,同一组内的物品最多只能选一个。
每件物品的体积是 vij,价值是 wij,其中 i 是组号,j 是组内编号。
求解将哪些物品装入背包,可使物品总体积不超过背包容量,且总价值最大。
输出最大价值。
输入格式
第一行有两个整数 N,V,用空格隔开,分别表示物品组数和背包容量。
接下来有 N 组数据:
每组数据第一行有一个整数 Si,表示第 i 个物品组的物品数量;每组数据接下来有 Si 行,每行有两个整数 vij,wij,用空格隔开,分别表示第 i 个物品组的第 j 个物品的体积和价值;
输出格式
输出一个整数,表示最大价值。
数据范围
0<N,V≤100
0<Si≤100
0<vij,wij≤100
输入样例
3 5
2
1 2
2 4
1
3 4
1
4 5
输出样例:
8
代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 110;
int n, m;
int v[N][N], w[N][N], s[N];
int f[N];
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i++)
{
cin >> s[i];
for(int j = 1; j <= s[i]; j++)
cin >> v[i][j] >> w[i][j];
}
for(int i = 1; i <= n; i++)
for(int j = m; j >= 0; j--)
{
for(int k = 1; k <= s[i]; k++)
{
if(j >= v[i][k]) //j >= 0所以要判断是否能装进去
f[j] = max(f[j], f[j - v[i][k]] + w[i][k]);
}
}
cout << f[m] << endl;
return 0;
}
二、 背包问题的应用
1. 包子凑数
小明几乎每天早晨都会在一家包子铺吃早餐。
他发现这家包子铺有 N 种蒸笼,其中第 i 种蒸笼恰好能放 Ai 个包子。
每种蒸笼都有非常多笼,可以认为是无限笼。
每当有顾客想买 X 个包子,卖包子的大叔就会迅速选出若干笼包子来,使得这若干笼中恰好一共有 X 个包子。
比如一共有 3 种蒸笼,分别能放 3、4 和 5 个包子。
当顾客想买 11 个包子时,大叔就会选 2 笼 3 个的再加 1 笼 5个的(也可能选出 1 笼 3 个的再加 2 笼 4 个的)。
当然有时包子大叔无论如何也凑不出顾客想买的数量。
比如一共有 3 种蒸笼,分别能放 4、5 和 6 个包子。
而顾客想买 7 个包子时,大叔就凑不出来了。
小明想知道一共有多少种数目是包子大叔凑不出来的。
输入格式
第一行包含一个整数 N。
接下来 N
行,每行包含一个整数 Ai。
输出格式
输出一个整数代表答案。
如果凑不出的数目有无限多个,输出INF。
数据范围
1≤N≤100,
1≤Ai≤100
输入样例1:
2
4
5
输出样例1:
6
输入样例2:
2
4
6
输出样例2:
INF
样例解释
对于样例1,凑不出的数目包括:1, 2, 3, 6, 7, 11。
对于样例2,所有奇数都凑不出来,所以有无限多个。
解题思路:
- 无限个数被凑不出来,说明这些数的gcd不是1。
gcd最大公约数两个数和的最大公约数 (Greatest Common Divisor) 是指同时整除a和b的最大因子,记为gcd(a, b) 。特殊的,当gcd(a, b) = 1,我们称a和b互素。
例如,1,2,4 均为 8 和 12 的公约数,最大的公约数就是 4。- 所以要两两求下gcd,看gcd是否大于1。
gcd(a, b) = gcd(b, a % b) 当余数为0时,当前算式的除数就是a和b的gcd。
gcd为1,最大不能表现出来的数必定有上界为(a - 1)(b - 1) - 1。而98,99是100内最大的互质数,所以上界选择10000。
下面就是看10000以内有多少个不能被组合出来。
代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 110;
int n;
int v[N];
bool f[10010];
int gcd(int a, int b)
{
return b ? gcd(b, a % b) : a;
}
int main()
{
int d = 0;
cin >> n;
for(int i = 1; i <= n; i++)
{
cin >> v[i];
d = gcd(d, v[i]);
}
if(d != 1)
{
cout << "INF" << endl;
return 0;
}
int cnt = 0;
f[0] = true;
for(int i = 1; i <= n; i++)
for(int j = v[i]; j <= 10000; j++)
f[j] = f[j] || f[j - v[i]];
for(int i = 1; i <= 10000; i++)
if(!f[i])
cnt++;
cout << cnt << endl;
return 0;
}
2.波动数列
观察这个数列:
1 3 0 2 -1 1 -2 …
这个数列中后一项总是比前一项增加2或者减少3,且每一项都为整数。
栋栋对这种数列很好奇,他想知道长度为 n 和为 s 而且后一项总是比前一项增加 a 或者减少 b 的整数数列可能有多少种呢?
输入格式
共一行,包含四个整数 n,s,a,b,含义如前面所述。
输出格式
共一行,包含一个整数,表示满足条件的方案数。
由于这个数很大,请输出方案数除以 100000007 的余数。
数据范围
1≤n≤1000,
−109≤s≤109,
1≤a,b≤106
输入样例:
4 10 2 3
输出样例:
2
样例解释
两个满足条件的数列分别是2 4 1 3和7 4 1 -2。
解题思路:
代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 1010;
int n, s, a, b;
int f[N][N];
int get_mod(int a, int b)
{
return (a % b + b) % b;
}
int main()
{
f[0][0] = 1;
cin >> n >> s >> a >> b;
for(int i = 1; i < n; i++)
for(int j = 0; j < n; j++)
{
f[i][j] = (f[i - 1][get_mod(j - i * a, n)] + f[i - 1][get_mod(j + i * b, n)]) % 100000007;
}
cout << f[n - 1][get_mod(s, n)] << endl;
return 0;
}
3. 砝码称重
你有一架天平和 N 个砝码,这 N 个砝码重量依次是 W1,W2,⋅⋅⋅,WN。
请你计算一共可以称出多少种不同的正整数重量?
注意砝码可以放在天平两边。
输入格式
输入的第一行包含一个整数 N。
第二行包含 N 个整数:W1,W2,W3,⋅⋅⋅,WN。
输出格式
输出一个整数代表答案。
数据范围
对于 50% 的评测用例,1≤N≤15。
对于所有评测用例,1≤N≤100,N 个砝码总重不超过 105。
输入样例:
3
1 4 6
输出样例:
10
样例解释
能称出的 10 种重量是:1、2、3、4、5、6、7、9、10、11。
1 = 1;
2 = 6 − 4 (天平一边放 6,另一边放 4);
3 = 4 − 1;
4 = 4;
5 = 6 − 1;
6 = 6;
7 = 1 + 6;
9 = 4 + 6 − 1;
10 = 4 + 6;
11 = 1 + 4 + 6。
解题思路:
代码
#include<iostream>
using namespace std;
const int N = 105, M = 1e5 + 10;
int n;
int a[N];
bool f[N][M];
int main()
{
int m = 0;
cin >> n;
for(int i = 1; i <= n; i++)
{
cin >> a[i];
m += a[i];
}
f[0][0] = true;
for(int i = 1; i <= n; i++)
for(int j = 0; j <= m; j++)
{
f[i][j] = f[i - 1][j] || f[i - 1][abs(j - a[i])];
if(j + a[i] <= m) f[i][j] = f[i][j] || f[i - 1][j + a[i]];
}
int cnt = 0;
for(int i = 1; i <= m; i++)
{
if(f[n][i])
cnt++;
}
cout << cnt << endl;
return 0;
}