AcWing.6 多重背包问题III（单调队列+滑动窗口）

AcWing.6 多重背包问题III

题目描述

有 N 种物品和一个容量是 V 的背包。
第 i 种物品最多有 si 件，每件体积是 vi，价值是 wi。
求解将哪些物品装入背包，可使物品体积总和不超过背包容量，且价值总和最大。
输出最大价值。

输入格式

第一行两个整数，N，V (0<N≤1000, 0<V≤20000)，用空格隔开，分别表示物品种数和背包容积。
接下来有 N 行，每行三个整数 vi,wi,si，用空格隔开，分别表示第 i 种物品的体积、价值和数量。

输出格式

输出一个整数，表示最大价值。

数据范围

0<N≤1000
0<V≤20000
0<vi,wi,si≤20000

输入样例

4 5
1 2 3
2 4 1
3 4 3
4 5 2

输出样例

10

在本题的数据范围下，暴力枚举和二进制优化均会TLE。
我们选择使用单调队列+滑动窗口。

dp[j]    =     dp[j]
dp[j+v]  = max(dp[j] +  w,  dp[j+v])
dp[j+2v] = max(dp[j] + 2w,  dp[j+v] +  w, dp[j+2v])
dp[j+3v] = max(dp[j] + 3w,  dp[j+v] + 2w, dp[j+2v] + w, dp[j+3v])
...
//滑动窗口最大为s(某一种物品的数量)
dp[j+kv] = max(dp[j+(k-s)] + sw,  dp[j+(k-s+1)v] + (s-1)w,...,dp[j+(k-1)v] + w,dp[j+kv])

C++代码：

使用数组模拟单调队列

#include<iostream>
#include<cstring>
#include<deque>
using namespace std;
const int maxn = 20010;
int dp[maxn], copy_dp[maxn],queue[maxn];

int main(){
	int N, V;
	cin >> N >> V;

	int v, w, s;
	int head = 0,tail = -1;
	for (int i = 1; i <= N; i++) {
		cin >> v >> w >> s;
		for (int j = 0; j < v; j++) {
			for (int k = j; k <= V; k += v) {
				while(tail>=head&&copy_dp[queue[tail]]+(k-queue[tail])/v*w<copy_dp[k])
					tail--;
				queue[++tail] = k;
				if(tail>=head&&queue[head]<k-s*v)
					head++;
				if(tail>=head)
					dp[k] = max(dp[k],copy_dp[queue[head]]+(k-queue[head])/v*w);		
			}
			head = 0;tail = -1;
		}
		memcpy(copy_dp, dp, sizeof(dp));
		
	}
	cout << dp[V] << endl;
	return 0;
} 

直接使用deque数据结构（会TLE）

#include<iostream>
#include<cstring>
#include<deque>
using namespace std;
const int maxn = 20010;

deque<int> order;
int dp[maxn], copy_dp[maxn];
int main() {
	int N, V;
	cin >> N >> V;

	int v, w, s;
	for (int i = 1; i <= N; i++) {
		cin >> v >> w >> s;
		deque<int> order;
		for (int j = 0; j < v; j++) {
			for (int k = j; k <= V; k += v) {
			    while (!order.empty() && copy_dp[order.back()] + (k - order.back()) / v * w < copy_dp[k]) {
					order.pop_back();
				}
				order.push_back(k);
				if (!order.empty() && order.front() < k - s * v) {
					order.pop_front();
				}
				if (!order.empty())
					dp[k] = max(copy_dp[k], copy_dp[order.front()] + (k - order.front()) / v * w);
			
				
			}
			order.clear();
		}
		memcpy(copy_dp, dp, sizeof(dp));
		
	}
	cout << dp[V] << endl;
	return 0;
}