当前位置:编程学习 > C/C++ >>

poj 2228 Naptime dp

这个题目的状态还是比较好想的,dp[i][j]表示已经睡了i个时段,最后睡在j时段的最优值,但是需要处理环的情况,我的做法是算两次,第一次不处理环,第二次强制性要求第一个时段需要睡,然后查看dp[m][n]+a[1]的值是否更优。
 
#include <iostream>  
#include <cstdio>  
#include <cstring>  
using namespace std;  
const int maxn=4e3+9;  
int a[maxn],dp[maxn][maxn];  
  
inline int max(int a,int b)  
{  
    if(a>b) return a;  
    else return b;  
}  
  
int main()  
{  
//    freopen("in.txt","r",stdin);  
    int n,m;  
    while(scanf("%d %d",&n,&m)!=EOF)  
    {  
  
        for(int i=1;i<=m;i++)  
        for(int j=1;j<=n;j++)  
        dp[i][j]=0;  
        for(int i=1;i<=n;i++)  
        scanf("%d",&a[i]);  
        for(int k=2,ret;k<=m;k++)  
        {  
            ret=0;  
            for(int i=k;i<=n;i++)  
            {  
                dp[k][i]=max(ret,dp[k-1][i-1]+a[i]);  
                ret=max(ret,dp[k-1][i-1]);  
            }  
        }  
        int ans=0;  
        for(int i=m;i<=n;i++)  
        ans=max(dp[m][i],ans);  
  
        for(int i=1;i<=m;i++)  
        for(int j=1;j<=n;j++)  
        dp[i][j]=0;  
  
        dp[2][2]=a[2];  
        for(int k=3,ret;k<=m;k++)  
        {  
            ret=0;  
            for(int i=k;i<=n;i++)  
            {  
                dp[k][i]=max(ret,dp[k-1][i-1]+a[i]);  
                ret=max(ret,dp[k-1][i-1]);  
            }  
        }  
        if(m>=2)  
        ans=max(ans,dp[m][n]+a[1]);  
        cout<<ans<<endl;  
    }  
    return 0;  
}  

 


补充:软件开发 , C++ ,
CopyRight © 2012 站长网 编程知识问答 www.zzzyk.com All Rights Reserved
部份技术文章来自网络,