1. 程式人生 > >ZOJ 4019 Schrödinger's Knapsack (dp)

ZOJ 4019 Schrödinger's Knapsack (dp)

傳送門

思路: dp題還是做的太少, 不夠靈活,這道題自己第一反應是揹包,但容量太大,不可行。按題意分析,每一類裡各自的價值是相同,所以當在一類中挑選的時候,當然是體積小的優先。所以對於兩類來說也是大趨勢是體積小的優先, 所以我們可以從小到大列舉A,B類的任意兩個區間, 只要這兩個區間的體積和小於總容量就行。但是A,B類的物品順序還有影響,我們用dp[i][j]代表列舉A類的1~i區間,B類的1~j區間。狀態轉移方程為 dp[i][j]=max(dp[i][j-1]+vb*(c-s), dp[i-1][j]+va*(c-s)); 表示我們可以先放A類的第i個物品,或者先放B類的第j個物品,取二者較大值。有一點貪心的感覺在裡面

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=2e3+10;
typedef long long ll;

int T;
ll dp[maxn][maxn];
ll va, vb, c, ca[maxn], cb[maxn], suma[maxn], sumb[maxn];

int main(){
	scanf("%d", &T);
	
	while(T--){
		scanf("%lld%lld%lld", &va, &vb, &c);
		int  n, m;
		scanf("%d%d", &n, &m);
		
		for(int i=0; i<=n; i++)
			for(int j=0; j<=m; j++) dp[i][j]=0;//用memset初始化會T;
		
		for(int i=1; i<=n; i++)
			scanf("%lld", &ca[i]);	
		for(int i=1; i<=m; i++)
			scanf("%lld", &cb[i]);
		
		sort(ca+1, ca+1+n); sort(cb+1, cb+1+m);
		for(int i=1; i<=n; i++)
			suma[i]=suma[i-1]+ca[i];
		for(int i=1; i<=m; i++)
			sumb[i]=sumb[i-1]+cb[i];
		
		ll ans=0;
		for(int i=1; i<=n; i++)
			if(suma[i]<=c){
				 dp[i][0]=va*(c-suma[i])+dp[i-1][0];
				 ans=max(ans, dp[i][0]);	
			}
		for(int j=1; j<=m; j++)
			if(sumb[j]<=c){
				dp[0][j]=vb*(c-sumb[j])+dp[0][j-1];
				ans=max(ans, dp[0][j]);	
			} 
		
		
		for(int i=1; i<=n; i++)
			for(int j=1; j<=m; j++)
			{
				ll s=suma[i]+sumb[j];
				if(s<=c)
				{
					dp[i][j]=max(dp[i-1][j]+va*(c-s), dp[i][j-1]+vb*(c-s));
					ans=max(ans, dp[i][j]);
				}
			}
		printf("%lld\n", ans);
	}
	
	
	return 0;
}