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

线段树和单调队列优化DP---POJ2373解题报告

在长为L(<=1000000)的草地(可看成线段)上装喷水头,喷射是以这个喷水头为中心,喷水头的喷洒半径是可调节的,

调节范围为[a,b]。要求草地的每个点被且只被一个喷水头覆盖,并且有些连续区间必须被某一个喷水头覆盖,

而不能由多个喷头分段完全覆盖,求喷水头的最小数目。

很容易想到,这可以用dp解决,定义dp[i]为覆盖[0,i]区间所需的的最小喷头数,

则dp[0]=0,dp[i]=min{dp[i-2*b]....dp[i-2*a]};因为喷头是向两边喷洒的,所以一个喷头覆盖的区间长度一定是偶数,

又由于题目要求喷头不能喷洒到[0,L]以外的区域,所以0开始的长度为奇数的子区间[0,L’]是不能被完全覆盖的。

还有一个问题是,某些连续区间必须被某一个喷水头覆盖这个限制该如何解决。我们可以这样想,

如果[s,e]这个区间只能被一个喷头覆盖,则[0,M](s<M<e)这一段子区间将不允许被完全覆盖,

因为如果[0,M]被完全覆盖会导致[s,e]区间被分割,所以我们可以对所以的[s,e]做上标记,

一种比较方便编码的方法是直接在dp这个数组上用一个特殊的值标记。我的做法是这样的:

[cpp] 
dp[0]=0; 
for(int i=1; i<=l; i++) dp[i]=inf; 
for(int i=0; i<n; i++) { 
        int s, e;   
        scanf("%d%d", &s, &e); 
        for(int j=s+1; j<e; j++) dp[j] = inf+1;//用inf+1表示不允许覆盖 

完整代码:
[cpp]
#include<cstdio> 
#include<cstring> 
#define L 1001000 
 
int a,b,n,l,inf,dp[L]; 
 
int dpro(void) 

     if(b<1) return -1; 
     dp[0]=0; 
                      
     for(int i=2; i<=l; i+=2) 
     { 
         if (dp[i]<=inf) { 
              int min = inf; 
               
              for(int j=a; j<=b; j++) { 
                    int idx = i-2*j; 
                    if(idx<0) break; 
                    if ( min>dp[idx] ) min=dp[idx]; 
              } 
               
              dp[i]=min+1; 
         } 
     } 
     if(dp[l]>=inf) return -1; 
     else return dp[l]; 

 
int main() 

    while (scanf("%d%d", &n, &l)!=EOF) { 
           
          scanf("%d%d", &a, &b); 
          inf = (l/a)+9; 
          for(int i=0; i<=l; i++) dp[i]=inf; 
           
          for(int i=0; i<n; i++) { 
                int s, e;   
                scanf("%d%d", &s, &e); 
                for(int j=s+1; j<e; j++) dp[j] = inf+1; 
          } 
           
          if(l&1==1) printf("-1\n"); 
          else printf("%d", dpro()); 
    } 
    return 0; 


从上面的程序中我们看到,在最坏的情况下,时间复杂度是O(L^2),而L的范围可达一百万,

所以在极端的数据下,这个程序是会超时的,所以我们需要对这个程序做一点优化。

在dp过程中的第二层for循环的作用是找出[i-2*b,i-2*a]这段区间的最小值,

这个查找一段区间的最小值的操作可以用线段树来优化,优化后的时间复杂度是O(LlgL)。


代码:

[cpp] 
#include<cstdio> 
#include<cstring> 
#define L 1001000 
 
int a,b,n,l,inf,dp[L]; 
int tree[4*L]; 
 
void updata(int *p, int rt, int l, int r,int pos, int k) 

     if (l==r) { 
          p[rt]=k; 
          return ; 
     } 
     int mid=(l+r)>>1; 
     if (pos<=mid) updata(p,rt<<1, l, mid, pos, k); 
     else updata(p, rt<<1|1, mid+1, r, pos, k); 
     int lv=p[rt<<1]; 
     int rv=p[rt<<1|1]; 
     p[rt]=lv<rv?lv:rv; 

 
int query(int *p,int rt, int l,int r, int s, int e) 

    if(l==s && e==r) return p[rt]; 
    int mid = (l+r)>>1; 
    if(e<=mid) return query(p, rt<<1, l, mid, s, e); 
    if(s>mid) return query(p, rt<<1|1, mid+1, r, s, e); 
    int lv=query(p,rt<<1, l, mid, s,mid); 
    int rv=query(p,rt<<1|1, mid+1, r, mid+1, e); 
    return lv<rv?lv:rv; 

 
int dpro(void) 

     if(b<1) return -1; 
     dp[0]=0; 
 
     for(int j=0; j<4*L; j++) tree[j]=L*2; 
 
     for(int j=a; j<=b; j++){ 
         if(dp[2*j]<=inf) dp[0+2*j] = dp[0]+1; 
         updata(tree,1, 1, l ,j,dp[2*j]); 
     } 
      
     for(int i=2*b+2; i<=l; i+=2) 
     { 
         int min; 
         int pos = (i>>1); 
         min = query(tree, 1, 1, l, pos-b, pos-a); 
 
         if (dp[i]<=inf) { 
     

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