POJ-1976(01背包)三个火车头

题目大意:有3个车头和一个车厢序列,每节车厢有一定的重量,每个车头最多可以拉连续的k节车厢,问3个车头最多可以拉的重量 

 解题思路:dp[i][j]表示第i节车厢,第j个车头能拉的最多的人数,由于车厢的重量不可能是负的因此,最优解一定满足每个车头要拉k节车厢 

代码如下

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
const int N=50005;
int dp[N][4],a[N];
int main()
{
    int T;
    cin>>T;
    while(T--){
        int n,k;
        a[0]=0;
        cin>>n;
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++){
            cin>>a[i];
            a[i]=a[i]+a[i-1];
        }
        cin>>k;
        /*for(int i=1;i<=n;i++){
            for(int j=1;j<=3;j++){
                int p=max(0,i-k);
                dp[i][j]=max(dp[i-1][j],dp[p][j-1]+a[i]-a[p]);
            }
        }*/
        for(int i=1;i<=n;i++){
            for(int j=1;j<=3;j++){
                if(i<j*k)
                    dp[i][j]=0;
                else
                    dp[i][j]=max(dp[i-1][j],dp[i-k][j-1]+a[i]-a[i-k]);
            }
        }
        cout<<dp[n][3]<<endl;
    }
    return 0;
}

没有更多推荐了,返回首页

私密
私密原因:
请选择设置私密原因
  • 广告
  • 抄袭
  • 版权
  • 政治
  • 色情
  • 无意义
  • 其他
其他原因:
120
出错啦
系统繁忙,请稍后再试

关闭