POJ 2796 Feel Good -- 单调栈,DP

9 篇文章 0 订阅

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

题目大意:给定一个数列a,记a[i..j]表示从i到j的子列,定义sum = min(k, i, j){a[k]} * sum[i..j],即子列a[i..j]中的最小值乘子列的和,求最大的sum。

分析:这题网上大多做为单调栈的例题,但是其实DP也可以(而且代码更简洁),两种代码我都给出来了。

重点在于求出以a[i]为最小值的最长子列,单调栈就不说了(网上太多了),说一下DP吧。以求左边最长子列为例,用dp[i]表示,看下面这一行代码

for (int j = i - 1; j > 0 && a[j] >= a[i]; dp[i] += dp[j], j -= dp[j]);
这就求出dp[i]了。因为当a[j] >= a[i]时,不小于a[j]的数都不小于a[i],但小于a[j]的数可能也不小于a[i],所以考察左边小于a[j]的第一个数即j - dp[j]。相比单调栈,这一行代码就显得很简洁了。时间上我用DP快一点,可能是因为我的stack实现直接用的STL,实际上两者是等价的,出栈的过程就相当于这里的j -= dp[j],单调栈中有一个new_node.pre += tmp.pre,这句话和dp[i] += dp[j]是一样的,tmp.pos和这里的j是一一对应的,只不过保存数据的结构不一样而已。

//单调栈
#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 = 100000 + 2; int a[MAX_N]; ll sum[MAX_N]; struct node { int num; int pre; int next; int pos; node() { } node(int _num, int _pre, int _next, int _pos) { num = _num; pre = _pre; next = _next; pos = _pos; } }; int main(int argc, char *argv[]) { // freopen("D:\\in.txt", "r", stdin); int n; stack 
                            
                              coll; cin >> n; for (int i = 1; i <= n; ++i) { scanf("%d", a + i); sum[i] = sum[i - 1] + a[i]; } ll ans(-1); int s, t; coll.push(node(a[1], 1, 1, 1)); for (int i = 2; i <= n; ++i) { node new_node(a[i], 1, 1, i); while (!coll.empty() && coll.top().num >= a[i]) { node tmp = coll.top(); coll.pop(); if (!coll.empty()) coll.top().next += tmp.next; new_node.pre += tmp.pre; ll _ans = tmp.num * (sum[tmp.pos + tmp.next - 1] - sum[tmp.pos - tmp.pre]); if (ans < _ans) { ans = _ans; s = tmp.pos - tmp.pre + 1; t = tmp.pos + tmp.next - 1; } } coll.push(new_node); } while (!coll.empty()) { node tmp = coll.top(); coll.pop(); if (!coll.empty()) coll.top().next += tmp.next; ll _ans = tmp.num * (sum[tmp.pos + tmp.next - 1] - sum[tmp.pos - tmp.pre]); if (ans < _ans) { ans = _ans; s = tmp.pos - tmp.pre + 1; t = tmp.pos + tmp.next - 1; } } printf("%I64d\n%d %d\n", ans, s, t); return 0; } //dp #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 = 100000 + 2; int a[MAX_N]; ll sum[MAX_N]; int dp[2][MAX_N]; int main(int argc, char *argv[]) { // freopen("D:\\in.txt", "r", stdin); int n; cin >> n; for (int i = 1; i <= n; ++i) { scanf("%d", a + i); sum[i] = sum[i - 1] + a[i]; dp[0][i] = dp[1][i] = 1; } ll ans(-1); int s, t; for (int i = 2; i <= n; ++i) { for (int j = i - 1; j > 0 && a[j] >= a[i]; dp[0][i] += dp[0][j], j -= dp[0][j]); } for (int i = n - 1; i > 0; --i) { for (int j = i + 1; j <= n && a[j] >= a[i]; dp[1][i] += dp[1][j], j += dp[1][j]); } for (int i = 1; i <= n; ++i) { ll tmp = a[i] * (sum[i + dp[1][i] - 1] - sum[i - dp[0][i]]); if (ans < tmp) ans = tmp, s = i - dp[0][i] + 1, t = i + dp[1][i] - 1; } printf("%I64d\n%d %d\n", ans, s, t); return 0; } 
                                                      
                                                     
                                                    
                                                   
                                                  
                                                 
                                                
                                               
                                              
                                             
                                            
                                           
                                          
                                         
                                        
                                       
                                      
                                     
                                    
                                  
                                 
                                
                               
                              
                             
                            
                           
                          
                         
                        
                       
                      
                     
                    
                   
                  
                 
                
               
              
             
            
           
          
       
      
      
     
     
    
    
   
   

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值