题解 | #称砝码#
称砝码
https://www.nowcoder.com/practice/f9a4c19050fc477e9e27eb75f3bfd49c
#include <iostream> #include <unordered_set> #include <vector> using namespace std; int main() { int n; cin >> n; vector<int> weight(n); vector<int> num(n); vector<int> vec; int sum = 0; for(int i = 0; i < n; i++) cin >> weight[i]; for(int i = 0 ; i < n; i ++){ cin >> num[i]; for(int j = 0; j < num[i]; j++) vec.push_back(weight[i]); sum += weight[i] * num[i]; } vector<bool> dp(sum+1, false); dp[0] = true; for(int it : vec){ for(int i = sum; i >= it; i--){ if(dp[i - it]) dp[i] = true; } } int count = 0; for(bool it : dp){ if(it) count++; } cout << count << endl; return 0; }