四种类型的物品,每一种类型物品数量都是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以内
#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
*/