zoukankan      html  css  js  c++  java
  • 用单调队列优化DP,写给自己

    单调队列是一种严格单调的队列,可以单调递增,也可以单调递减。队首位置保存的是最优解,第二个位置保存的是次优解,ect。。。

    单调队列可以有两个操作:

    1、插入一个新的元素,该元素从队尾开始向队首进行搜索,找到合适的位置插入之,如果该位置原本有元素,则替换它。

    2、在过程中从队首删除不符合当前要求的元素。

    单调队列实现起来可简单,可复杂。简单的一个数组,一个head,一个tail指针就搞定。复杂的用双向链表实现。

    用处:

    1、保存最优解,次优解,ect。

    2、利用单调队列对dp方程进行优化,可将O(n)复杂度降至O(1)。也就是说,将原本会超时的N维dp降优化至N-1维,以求通过。这也是我想记录的重点

    是不是任何DP都可以利用单调队列进行优化呢?答案是否定的。

    记住!只有形如 dp[i]=max/min (f[k]) + g[i]  (k<i && g[i]是与k无关的变量)才能用到单调队列进行优化。

    优化的对象就是f[k]。

    通过例题来加深感受

    http://www.acm.uestc.edu.cn/problem.php?pid=1685

                                        我要长高

    Description

    韩父有N个儿子,分别是韩一,韩二…韩N。由于韩家演技功底深厚,加上他们间的密切配合,演出获得了巨大成功,票房甚至高达2000万。舟子是名很有威望的公知,可是他表面上两袖清风实则内心阴暗,看到韩家红红火火,嫉妒心遂起,便发微薄调侃韩二们站成一列时身高参差不齐。由于舟子的影响力,随口一句便会造成韩家的巨大损失,具体亏损是这样计算的,韩一,韩二…韩N站成一排,损失即为C*(韩i与韩i+1的高度差(1<=i<N))之和,搞不好连女儿都赔了.韩父苦苦思索,决定给韩子们内增高(注意韩子们变矮是不科学的只能增高或什么也不做),增高1cm是很容易的,可是增高10cm花费就很大了,对任意韩i,增高Hcm的花费是H^2.请你帮助韩父让韩家损失最小。

    Input

    有若干组数据,一直处理到文件结束。 每组数据第一行为两个整数:韩子数量N(1<=N<=50000)和舟子系数C(1<=C<=100) 接下来N行分别是韩i的高度(1<=hi<=100)。

    首先建立方程,很容易想到的是,dp[i][j]表示第 i 个儿子身高为 j 的最低花费。分析题目很容易知道,当前儿子的身高花费只由前一个儿子影响。因此,

    dp[i][j]=min(dp[i-1][k] + abs(j-k)*C + (x[i]-j)*(x[i]-j));其中x[i]是第i个儿子原本的身高

    我们分析一下复杂度。

    首先有N个儿子,这需要一个循环。再者,每个儿子有0到100的身高,这也需要一维。再再者,0到100的每一个身高都可以有前一位儿子的身高0到100递推而来。

    所以朴素算法的时间复杂度是O(n^3)。题目只给两秒,难以接受!

    分析方程:

    当第 i 个儿子的身高比第 i-1 个儿子的身高要高时,

    dp[i][j]=min(dp[i-1][k] + j*C-k*C + X);   ( k<=j ) 其中 X=(x[i]-j)*(x[i]-j)。

    当第 i 个儿子的身高比第 i-1 个儿子的身高要矮时,

    dp[i][j]=min(dp[i-1][k] - j*C+k*C + X);   ( k>=j )

    对第一个个方程,我们令 f[i-1][k]=dp[i-1][k]-k*C,  g[i][j]=j*C+X; 于是 dp[i][j] = min (f[i-1][k])+ g[i][j]。转化成这样的形式,我们就可以用单调队列进行优化了。

    第二个方程同理。

    接下来便是如何实现,实现起来有点技巧。具体见下

    View Code
     1 #include<iostream>
     2 #include<string>
     3 #include<stdio.h>
     4 #include<memory.h>
     5 using namespace std;
     6 #define inf 0xfffffff
     7 #define min(a,b) a<b?a:b
     8 #define max(a,b) a>b?a:b
     9 
    10 int dp[2][101];
    11 int n,c;
    12 int q[101];
    13 int head,tail,cur;
    14 
    15 int main()
    16 {
    17     int i,j,x,nowf;
    18     freopen("D:\\in.txt","r",stdin);
    19     while(scanf("%d%d",&n,&c)==2)
    20     {
    21         scanf("%d",&x);
    22         cur=0;
    23         for(i=0;i<x;i++)
    24             dp[cur][i]=inf;
    25         for(i=x;i<=100;i++)
    26             dp[cur][i]=(x-i)*(x-i);
    27         for(i=1;i<n;i++)
    28         {
    29             scanf("%d",&x);
    30             cur=1-cur;
    31             //比前一个人高
    32             head=tail=0;
    33             for(j=0;j<=100;j++) //当身高为j时候,队列里便已经保存了0~j-1的信息,注意,是第i-1个人的信息
    34             {
    35                 nowf=dp[1-cur][j]-j*c;
    36                 while(head<tail && q[tail-1]>nowf)
    37                     tail--;
    38                 q[tail++]=nowf;
    39                 if(j<x)
    40                     dp[cur][j]=inf;    
    41                 else
    42                     dp[cur][j]=q[head]+j*c+(x-j)*(x-j);
    43             }
    44             //比前一个人矮
    45             head=tail=0; 
    46             for(j=100;j>=0;j--) //当身高为j时候,队列里便已经保存了100~j+1的信息,正写反写是有技巧的
    47             {
    48                 nowf=dp[1-cur][j]+j*c;
    49                 while(head<tail && q[tail-1]>nowf)
    50                     tail--;
    51                 q[tail++]=nowf;
    52                 if(j>=x)
    53                     dp[cur][j]=min(dp[cur][j],q[head]-j*c+(x-j)*(x-j));
    54             }
    55         }
    56         int ans=inf;
    57         for(i=0;i<=100;i++)
    58             ans=min(ans,dp[cur][i]);
    59         printf("%d\n",ans);
    60     }
    61     return 0;
    62 }

    还有一个比较适合理解该优化方法的题目是HDU 3401http://acm.hdu.edu.cn/showproblem.php?pid=3401

    大概题目便是:一个人知道接下来T天的股市行情,想知道最终他能赚到多少钱。

    构造状态dp[i][j]表示第i 天拥有 j只股票的时候,赚了多少钱

    状态转移有:

    1、从前一天不买不卖:

    dp[i][j]=max(dp[i-1][j],dp[i][j])

    2、从前i-W-1天买进一些股:

    dp[i][j]=max(dp[i-W-1][k]-(j-k)*AP[i],dp[i][j])

    3、从i-W-1天卖掉一些股:

    dp[i][j]=max(dp[i-W-1][k]+(k-j)*BP[i],dp[i][j])

    这里需要解释一下为什么只考虑第i-W-1天的买入卖出情况即可。想想看,i-W-2天是不是可以通过不买不卖将自己的最优状态转移到第i-W-1天?以此类推,之前的都不需要考虑了,只考虑都i-W-1天的情况即可。

    对买入股票的情况进行分析,转化成适合单调队列优化的方程形式

    dp[i][j]=max(dp[i-W-1][k]+k*AP[i])-j*AP[i]。令f[i-W-1][k]=dp[i-W-1][k]+k*AP[i],则dp[i][j]=max(f[i-W-1][k]) - j*AP[i]。

    这便可以用单调队列进行优化了。卖股的情况类似分析。

    View Code
     1 #include<iostream>
     2 #include<string>
     3 using namespace std;
     4 #define inf 0xfffffff
     5 #define min(a,b) a<b?a:b
     6 #define max(a,b) a>b?a:b
     7 
     8 struct node
     9 {
    10     int pos;
    11     int f;
    12 };
    13 
    14 int AP[2001],BP[2001],AS[2001],BS[2001];
    15 int dp[2001][2001];
    16 node q[2001];
    17 int head,tail;
    18 int n,MaxP,W;
    19 
    20 int main()
    21 {
    22     int i,j,cas,nowf;
    23     freopen("D:\\in.txt","r",stdin);
    24     scanf("%d",&cas);
    25     while(cas--)
    26     {
    27         scanf("%d%d%d",&n,&MaxP,&W);
    28         for(i=1;i<=n;i++)
    29         {
    30             scanf("%d%d%d%d",&AP[i],&BP[i],&AS[i],&BS[i]);
    31         }
    32         for(i=0;i<=n;i++)
    33             for(j=0;j<=MaxP;j++)
    34                 dp[i][j]=-inf;
    35         for(i=1;i<=W+1;i++)
    36         {
    37             for(j=0;j<=(min(MaxP,AS[i]));j++)
    38                 dp[i][j]=-AP[i]*j;
    39         }
    40         dp[0][0]=0;
    41         for(i=1;i<=n;i++)
    42         {
    43             for(j=0;j<=MaxP;j++)
    44                 dp[i][j]=max(dp[i][j],dp[i-1][j]);
    45             if(i<=W+1)
    46                 continue;
    47             int pre=i-W-1; //队列里保存的是第i-W-1天的信息
    48             head=tail=0;
    49             for(j=0;j<=MaxP;j++) //因为是买,所以肯定越来越多,求DP[i][j]的时候,之前的0~~j-1的信息就存在在队列了
    50             {
    51                 nowf=dp[pre][j]+j*AP[i];
    52                 while(head<tail && q[tail-1].f<nowf)
    53                     tail--;
    54                 q[tail].f=nowf;q[tail++].pos=j;
    55                 while(head<tail && q[head].pos+AS[i]<j)
    56                     head++;
    57                 dp[i][j]=max(dp[i][j],q[head].f-j*AP[i]);
    58             }
    59             head=tail=0;
    60             for(j=MaxP;j>=0;j--) //因为是卖,道理同上
    61             {
    62                 nowf=dp[pre][j]+j*BP[i];
    63                 while(head<tail && q[tail-1].f<nowf)
    64                     tail--;
    65                 q[tail].f=nowf;q[tail++].pos=j;
    66                 while(head<tail && q[head].pos-BS[i]>j)
    67                     head++;
    68                 dp[i][j]=max(dp[i][j],q[head].f-j*BP[i]);
    69             }
    70         }
    71         int ans=0;
    72         for(i=0;i<=MaxP;i++)
    73             ans=max(ans,dp[n][i]);
    74         printf("%d\n",ans);
    75     }
    76     return 0;
    77 }

     最后再说一个应用,用单调队列来优化多重背包问题   hdu 2191

    如果有n个物品,每个物品的价格是w,重量是c,且每个物品的数量是k,那么用这样的一些物品去填满一个容量为m的背包,使得得到的背包价值最大化,这样的问题就是多重背包问题。

    对于多重背包的问题,有一种优化的方法是使用二进制优化,这种优化的方法时间复杂度是O(m*∑log k[i]),具体可以见

    http://www.cnblogs.com/ka200812/archive/2011/08/06/2129505.html

    而利用单调队列的优化,复杂度是O(mn)

    首先,对于第i件物品,如果已知体积为V,价值为W,数量为K,那么可以按照V的余数,将当前的体积J分成V组(0,1,....V-1)。

    对于任意一组,可以得到转移方程:f[i*V+c]=f[k*V+c]+(i-k)*W,其中c是V组分组中的任意一个

    令f[i*V+c]=dp[i],那么就得到dp[i]=dp[k]+(i-k)*W (k>=i-K)

    将dp[k]-k*W看做是优化函数,那么就可以运用单调队列来优化了

    View Code
     1 #include<iostream>
     2 #include<string>
     3 #include<algorithm>
     4 using namespace std;
     5 
     6 int c,w,num;
     7 int a[101],b[101],f[101];
     8 int n,m;
     9 int q[101];
    10 int head,tail;
    11 
    12 int main()
    13 {
    14     int i,cas,k,j;
    15     freopen("in.txt","r",stdin);
    16     scanf("%d",&cas);
    17     while(cas--)
    18     {
    19         scanf("%d%d",&m,&n);
    20         memset(f,0,sizeof(f));
    21         for(i=0;i<n;i++)
    22         {
    23             scanf("%d%d%d",&c,&w,&num);
    24             if(m/c<num) //取最小值
    25                 num=m/c;
    26             for(k=0;k<c;k++) //分成c份
    27             {
    28                 head=tail=0;
    29                 for(j=0;j<=(m-k)/c;j++)
    30                 {
    31                     int x=j;
    32                     int y=f[j*c+k]-j*w; //将f[j*c+k]-j*w入队
    33                     while(head<tail && y>=b[tail-1])
    34                         tail--;
    35                     a[tail]=x; //记录下标
    36                     b[tail++]=y; //记录dp值
    37                     while(a[head]<j-num) //如果超出了范围
    38                         head++;
    39                     f[j*c+k]=b[head]+j*w;
    40                 }
    41             }
    42         }
    43         printf("%d\n",f[m]);
    44     }
    45     return 0;
    46 }
  • 相关阅读:
    codevs 3160 最长公共子串(SAM)
    noi2018d2t1
    后缀自动机
    [Scoi2014]方伯伯的玉米田
    【NOI2017】泳池
    【BZOJ2560】串珠子
    POJ 1737 统计有n个顶点的连通图有多少个 (带标号)
    【BZOJ3590】[Snoi2013]Quare 状压DP
    【BZOJ1095】【ZJOI2007】捉迷藏 [动态点分治]
    BZOJ1316: 树上的询问
  • 原文地址:https://www.cnblogs.com/ka200812/p/2585950.html
Copyright © 2011-2022 走看看