每日一题-动态规划(从不同类型的物品中各挑选一个,使得最后花费总和等于1000)

四种类型的物品,每一种类型物品数量都是n,先要从每种类型的物品中挑选一件,使得最后花费总和等于1000

暴力做法10000^4

看到花费总和是1000,很小且固定的数字,肯定有玄机,从这里想应该是用dp,不难想到用dp[i][j]表示前i种类型的物品花费为j的方案数量,思考转移方程:

dp[i][j] = dp[i-1][j-A] * js[i][A],js[i][A]表示i类型的物件花销为A的方案数量,如此只需要枚举j和A,它们的范围就是1000以内

cpp 复制代码
#include <iostream>
#include <vector>
#define ios ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
using namespace std;

int dp[5][1100], js[5][11000];
int n;
vector<int> ve[5];
int main() {
    ios;
    cin >> n;
    for(int i = 1; i <= n; i ++) {
    	int a, b , c, d;
    	cin >> a >> b >> c >> d;
    	ve[1].push_back(a);
    	ve[2].push_back(b);
    	ve[3].push_back(c);
    	ve[4].push_back(d);
	}
	for(int i = 1; i <= 4; i++) {
		for(int j = 0; j < ve[i].size(); j++) {
			js[i][ve[i][j]] ++;
		}
	}
	for(auto p : ve[1]) {
		dp[1][p] ++;
	}
	for(int i = 2; i <= 4; i++) {
		for(int j = 1; j <= 1000; j++) {
			if(js[i][j]) {
				for(int k = j; k <= 1000; k++) {
					dp[i][k] += dp[i-1][k-j] * js[i][j];
				}
			}
		}
	}
	cout << dp[4][1000];
    return 0;
}
/*
3
250 250 250 250
156 201 205 400
205 190 100 250
*/
相关推荐
-dzk-6 小时前
【代码随想录】LC 59.螺旋矩阵 II
c++·线性代数·算法·矩阵·模拟
风筝在晴天搁浅6 小时前
hot100 78.子集
java·算法
Jasmine_llq6 小时前
《P4587 [FJOI2016] 神秘数》
算法·倍增思想·稀疏表(st 表)·前缀和数组(解决静态区间和查询·st表核心实现高效预处理和查询·预处理优化(提前计算所需信息·快速io提升大规模数据读写效率
超级大只老咪6 小时前
快速进制转换
笔记·算法
m0_706653237 小时前
C++编译期数组操作
开发语言·c++·算法
故事和你917 小时前
sdut-Java面向对象-06 继承和多态、抽象类和接口(函数题:10-18题)
java·开发语言·算法·面向对象·基础语法·继承和多态·抽象类和接口
qq_423233907 小时前
C++与Python混合编程实战
开发语言·c++·算法
TracyCoder1237 小时前
LeetCode Hot100(19/100)——206. 反转链表
算法·leetcode
m0_715575347 小时前
分布式任务调度系统
开发语言·c++·算法