TOJ 1631 Naptime -- 环形DP

最新推荐文章于 2023-01-15 19:46:25 发布

randygx

最新推荐文章于 2023-01-15 19:46:25 发布

阅读量953

点赞数

分类专栏： dp ACM 文章标签：动态规划 dp

本文链接：https://blog.csdn.net/randygx/article/details/27566959

版权

ACM 同时被 2 个专栏收录

56 篇文章 0 订阅

订阅专栏

9 篇文章 0 订阅

订阅专栏

题目链接：http://acm.tzc.edu.cn/acmhome/problemdetail.do?&method=showdetail&id=1631

题目大意：从长度为n的非负数列中选取m个数，使选取的数和最大，其中选取的每一个连续子序列的第一个数不计入求和(比如选取a1, a2, a3，那么和为a2 + a3。特别地，选取的任意单个数求和为0)。此外，数列首尾视为相连。

分析：如果没有首尾相连的条件，这题就是简单的dp。加上首尾相连之后，第一想法是环形dp，即在数列后面接上m - 1个数(超过m - 1个数的部分都可以归为原数列的情况)。但是这样会导致一个问题：原数列的前面部分可能被选取2次，即接在原数列后面的数和原数列中的数同时被选取。只要前面某个数足够大就会发生这样的情况。所以这种做法是不对的。

正确的做法是：先忽略首尾相连的条件求出一个最大值，那么什么情况下这个最大值不对呢？就是当a0和a(n-1)同时被选的情况。用dp[i][j][1]表示从前i个数选j个数且a[i]必选时的最大值，在普通情况下dp[0][1][1] = 0，但选了a(n-1)之后，dp[0][1][1] = a[0]。这就是仅有的区别。所以可以这样处理这种情况：令dp[0][1][1] = a[0]并且dp[n - 1][m][1]就是这种情况下的最大值。注意dp[n - 1][m][1]可能并非从dp[0][1][1]转化而来，需要把除dp[0][1][1]之外的其他值预先置为-inf，这样强迫后面的所有值从dp[0][1][1]转化而来。当然，把其他所有值预置为-inf是不必要的，只需要预置部分值就行了(就是动态规划的初值，如dp[0][0][0]，在普通情况下这些值应该是0)，但是这样无疑可以简化操作与代码，反正这也花不了多少时间。

#include 
   
   
    
    
#include 
    
    
     
     
#include 
     
     
      
      
#include 
      
      
       
       
#include 
       
       
         #include 
         #include 
         
           #include 
          
            #include 
           
             #include 
            
              #include 
             
               #include 
              
                #include 
               
                 #include 
                
                  #include 
                 
                   #include 
                  
                    #define mp make_pair #define X first #define Y second #define MEMSET(a, b) memset(a, b, sizeof(a)) using namespace std; typedef unsigned int ui; typedef long long ll; typedef unsigned long long ull; typedef pair 
                   
                     pii; typedef vector 
                    
                      vi; typedef vi::iterator vi_it; typedef map 
                     
                       mii; typedef priority_queue 
                      
                        pqi; typedef priority_queue 
                       
                         , greater 
                        
                          > rpqi; typedef priority_queue 
                         
                           pqp; typedef priority_queue 
                          
                            , greater 
                           
                             > rpqp; const int MAX_N = 3830 + 2; int dp[2][MAX_N][2]; int a[MAX_N]; int main(int argc, char *argv[]) { // freopen("D:\\in.txt", "r", stdin); int n, b, i, j; cin >> n >> b; for (i = 1; i <= n; ++i) { scanf("%d", a + i); } for (i = 2; i <= n; ++i) { int id1 = i & 1, id2 = id1 ^ 1; for (j = 2; j <= i && j <= b; ++j) { dp[id1][j][0] = max(dp[id2][j][0], dp[id2][j][1]); dp[id1][j][1] = max(dp[id2][j - 1][0], dp[id2][j - 1][1] + a[i]); } } int ans = max(dp[n & 1][b][0], dp[n & 1][b][1]); for (j = 0; j <= b; ++j) { dp[0][j][0] = dp[0][j][1] = dp[1][j][0] = dp[1][j][1] = INT_MIN; } dp[1][1][1] = a[1]; for (i = 2; i <= n; ++i) { int id1 = i & 1, id2 = id1 ^ 1; for (j = 1; j <= i && j <= b; ++j) { dp[id1][j][0] = max(dp[id2][j][0], dp[id2][j][1]); dp[id1][j][1] = max(dp[id2][j - 1][0], dp[id2][j - 1][1] + a[i]); } } ans = max(ans, dp[n & 1][b][1]); cout << ans << endl; return 0; }