1. 程式人生 > >ytu oj 動態規劃進階題目之開餐館

ytu oj 動態規劃進階題目之開餐館

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;
}