ytu oj 動態規劃進階題目之開餐館
阿新 • • 發佈:2019-02-12
dp[i][0]代表第i個餐館沒有被選中,dp[i][1]代表第i個餐館被選中
所以狀態轉移方程看程式碼-_-
#include<stdio.h> #include<iostream> #include<string.h> #define maxn 1000010 using namespace std; int a[1009],b[1009],dp[1009][2]; int main() { int t; scanf("%d",&t); while(t--) { int n,k; scanf("%d %d",&n,&k); for(int i=1;i<=n;i++) scanf("%d",&a[i]); for(int i=1;i<=n;i++) scanf("%d",&b[i]); dp[0][0]=dp[0][1]=0; dp[1][0]=0; for(int i=1;i<=n;i++) { if(a[i]-k<a[1])dp[i][1]=b[i]; else { int j; for(j=i-1;j>=1;j--) { if(a[i]-k>a[j])break; } dp[i][1]=max(dp[j][0],dp[j][1])+b[i]; } dp[i][0]=max(dp[i-1][1],dp[i-1][0]); } int p; p=max(dp[n][0],dp[n][1]); printf("%d\n",p); } return 0; }