题意:
多组输入,给定a,b两个数组,长度分别为n,m。每个元素有两个值:wi,vi,要求从两个数组中分别选出一个子数组,使得两个子数组的wi和相同,并使总的vi之和最大。
数据范围:n,m<=1e3,wi<=1e3,vi<=1e9,∑(n+m)<=1e4
解法
参考了大佬——> 传送门
大佬不愧是大佬,看了以后我深受启发,不过其中的dp过程开始看有点费解。
后来我看我队友捣鼓出来的过程,个人感觉比上文好懂一点。
总的思路是一样的。一支队伍从左往右dp,另外一支从右往左dp
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll inf = 1e15;
const int maxn = 1000007;
const int maxdp = 1000007;
const int base = 50000;
const int lim = 100000;
struct node{
ll w,v;
}s[maxn];
ll w[maxn],v[maxn];
int t,n,m;
ll f[maxdp];
int main()
{
scanf("%d",&t);
while(t--)
{
//init
fill(f, f + maxdp, -inf);
f[base] = 0;
scanf("%d%d",&n,&m);
for(int i=1;i<=n+m;i++)
{
scanf("%lld%lld",&s[i].w,&s[i].v);
if(i > n) s[i].w = -s[i].w;
}
random_shuffle(s + 1, s + 1 + m + n);
for(int i=1;i<=n+m;i++)
{
w[i] = s[i].w;
v[i] = s[i].v;
}
ll ans = 0;
for(int i=1;i<=n+m;i++)
{
if(w[i] > 0)
for(int j=lim;j-w[i]>=0;j--)
f[j] = max(f[j], f[j - w[i]] + v[i]);
else
for(int j=0;j-w[i]<=lim;j++)
f[j] = max(f[j], f[j - w[i]] + v[i]);
ans = max(ans, f[base]);
}
printf("%lld\n",ans);
}
return 0;
}