ZOJ 4019 Schrödinger's Knapsack (dp)
阿新 • • 發佈:2019-01-22
思路: 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; }