【AcWing算法提高课】1.9单调队列优化的DP问题

匿名博主

已于 2023-01-29 23:02:15 修改

阅读量85

点赞数

分类专栏： AcWing算法提高课文章标签：算法动态规划 c++

于 2022-09-19 23:26:24 首次发布

本文链接：https://blog.csdn.net/m0_71477690/article/details/126660942

版权

AcWing算法提高课专栏收录该内容

22 篇文章 2 订阅

订阅专栏

一、最大子序和

135.最大子序和题目链接

考察以 $i$ 结尾的，长度不超过 $m$ 的最大子序和：
对于长度为 $x\ (1\le x\le m)$ 的子序列，它的子序和为 $s [i] - s [i - x]$
因此要求以 $i$ 结尾的，长度不超过 $m$ 的最大子序和，即求 $1\le x\le m$ 中 $s [i] - s [i - x]$ 的最大值，只要求出 $s [i - x]$ 的最小值即可。
至此，滑动窗口的模型已经出现：对于每个 $i$ ，求出 $s$ 在长度为 $m$ 的区间 $[i - m, i - 1]$ 中的最小值。

代码实现：

#include <cstdio>
#include <algorithm>
using namespace std;

const int N = 300010, INF = 1e9;

int n, m;
int s[N], q[N];

int main(){
    scanf("%d %d", &n, &m);
    for (int i = 1; i <= n; i ++){
        scanf("%d", &s[i]);
        s[i] += s[i - 1];
    }
    
    int res = -INF;
    int hh = 0, tt = 0;  //注意初始化时队尾tt不是-1而是0,这使得单调队列在最初含有一个元素0
    
    for (int i = 1; i <= n; i ++){
        if (q[hh] < i - m) hh ++;
        res = max(res, s[i] - s[q[hh]]);
        while (hh <= tt && s[q[tt]] >= s[i]) tt --;
        q[++ tt] = i;
    }
    
    printf("%d", res);
    
    return 0;
}

二、旅行问题

1088.旅行问题题目链接

破环成链的方法已经在区间DP中提及，只要开两倍长的链即可。

在这里插入图片描述
先考虑顺时针怎么做：如何判断从 $i$ 号点顺时针能够周游一圈呢？
容易发现，从 $i$ 号点能够开到 $j$ 号点的充分必要条件是到达路径上每个点时车的剩余存油量都不小于零，即上个点剩余油量减去两点距离不小于零。
设 $s [i]$ 是加油站油量 $o [i]$ 减去到下点距离 $d [i]$ 的前缀和，那么上述条件可写作： $\forall i\le k<j,s[k]-s[i-1]\ge 0$ ，等价于 $min\{s[k]\}\ge s[i-1]\ (i\le k<j)$ 。
回到原问题，判断从 $i$ 号点顺时针能否周游一圈，即判断长度为 $n$ 的区间 $[i, i + n - 1]$ 中 $min\{s[k]\}$ 是否不小于 $s [i - 1]$ ，很明显这是一个滑动窗口的模型，用单调队列求解，注意求解过程是逆序枚举，因为要求每个点之后的区间最小值。

再来考虑逆时针怎么做：
与顺时针类似，用前缀和辅助判断，不过 $s [i]$ 是 $o [i] - d [i - 1]$ (到上点距离) 的前缀和，对于点 $i$ 是判断长度为 $n$ 的区间 $[i - n, i - 1]$ 中 $min\{s[k]\}$ 是否不大于 $s [i]$ (或者可以在之前求后缀和)。

代码实现：

#include <cstdio>
#include <algorithm>
using namespace std;

const int N = 2e6 + 10;

typedef long long LL;

int n;
int o[N], d[N];
LL s[N];
int q[N];
bool st[N];

int main(){
    scanf("%d", &n);
    for (int i = 1; i <= n; i ++)
        scanf("%d %d", &o[i], &d[i]);
    
    //先顺时针    
    for (int i = 1; i <= n; i ++)
        s[i] = s[i + n] = o[i] - d[i];
    for (int i = 1; i <= n * 2; i ++)
        s[i] += s[i - 1];
        
    int hh = 0, tt = -1;
    for (int i = n * 2; i; i --){
        if (hh <= tt && q[hh] >= i + n) hh ++;
        while (hh <= tt && s[q[tt]] >= s[i]) tt --;
        q[++ tt] = i;
        if (i <= n && s[q[hh]] >= s[i - 1]) st[i] = 1;
    }
    
    //再逆时针
    d[0] = d[n];
    for (int i = 1; i <= n; i ++)
        s[i] = s[i + n] = o[i] - d[i - 1];
    for (int i = 1; i <= n * 2; i ++)
        s[i] += s[i - 1];
        
    hh = 0, tt = -1;
    for (int i = 1; i <= n * 2; i ++){
        if (hh <= tt && q[hh] < i - n) hh ++;
        if (i > n && s[q[hh]] <= s[i]) st[i - n] = 1;
        while (hh <= tt && s[q[tt]] <= s[i]) tt --;
        q[++ tt] = i;
    }
    
    for (int i = 1; i <= n; i ++){
        if (st[i]) puts("TAK");
        else puts("NIE");
    }
    
    return 0;
}

三、烽火传递

1089.烽火传递题目链接

设 $f [i]$ 表示所有在前 $i$ 个烽火台中传递，且第 $i$ 个烽火台发出信号的方案中总代价的最小值，容易得出：
$f[i]=min\{f[k]\}+w[i]\ (i-m+1\le k<i)$ (即求长度为 $m - 1$ 的区间 $[i - m + 1, i - 1]$ 中 $f [k]$ 的最小值)
滑动窗口问题，用单调队列求解。
最终答案为 $min\{f[k]\}\ (n-m+1\le k \le n)$ 。

代码实现：

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;

const int N = 2e5 + 10, INF = 1e9;

int n, m;
int a[N], q[N];
int f[N];

int main(){
    scanf("%d %d", &n, &m);
    for (int i = 1; i <= n; i ++) scanf("%d", &a[i]);
    
    int hh = 0, tt = 0;
    for (int i = 1; i <= n; i ++){
        if (q[hh] < i - m) hh ++;
        f[i] = f[q[hh]] + a[i];
        while (hh <= tt && f[q[tt]] >= f[i]) tt --;
        q[++ tt] = i;
    }
    
    int res = INF;
    for (int i = n - m + 1; i <= n; i ++) res = min(res, f[i]);
    
    printf("%d", res);
    
    return 0;
}

四、绿色通道

1090.绿色通道题目链接

容易证明，当最长的空题段的长度减少时，抄完合法的题目所需的时间会增加，因此具有二分性质，可用二分答案求解。
而二分答案的判断即是上题的DP过程，要注意上题区间长度是 $m - 1$ (连续 $m$ 个中至少有一个)，而本题区间长度是 $m + 1$ (相邻两个之间的空段是 $m$ )。

代码实现：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 5e4 + 10, INF = 1e9;

int n, m;
int w[N];
int f[N], q[N];

bool check(int k){
    f[0] = 0;
    int hh = 0, tt = 0;
    for (int i = 1; i <= n; i ++){
        if (q[hh] < i - k - 1) hh ++;
        f[i] = f[q[hh]] + w[i];
        while (hh <= tt && f[q[tt]] >= f[i]) tt --;
        q[++ tt] = i;
    }
    
    int res = INF;
    for (int i = n - k; i <= n; i ++) res = min(res, f[i]);
    
    return res <= m;
}

int main(){
    scanf("%d %d", &n, &m);
    for (int i = 1; i <= n; i ++) scanf("%d", &w[i]);
    
    int l = 0, r = n;
    while (l < r){  //注意二分模板的采用
        int mid = l + r >> 1;
        if (check(mid)) r = mid;
        else l = mid + 1;
    }
    
    printf("%d", r);
    
    return 0;
}

五、修剪草坪

1087.修建草坪题目链接

设 $f [i]$ 表示所有在前 $i$ 头奶牛中选的方案的最大效率， $s [i]$ 是前 $i$ 头奶牛效率的和，有：

不选第 $i$ 头牛， $f [i - 1]$
选第 $i$ 头牛，且包括第 $i$ 头牛在内选了连续 $k\ (1\le k\le K)$ 头牛： $min\{f[i-k-1]-s[i-k]\}+s[i]$

代码实现：

#include <cstdio>
#include <algorithm>
using namespace std;

const int N = 1e5 + 10;

typedef long long LL;

int n, m;
LL s[N];
LL f[N];
int q[N];

LL g(int i){
    return f[i - 1] - s[i];
}

int main(){
    scanf("%d %d", &n, &m);
    for (int i = 1; i <= n; i ++)
        scanf("%lld", &s[i]), s[i] += s[i - 1];
      
    int hh = 0, tt = 0; 
    for (int i = 1; i <= n; i ++){
        if (q[hh] < i - m) hh ++;
        f[i] = max(f[i - 1], g(q[hh]) + s[i]);
        while (hh <= tt && g(q[tt]) <= g(i)) tt --;
        q[++ tt] = i;
    }
    
    printf("%lld", f[n]);
    
    return 0;
}

六、理想的正方形

1091.理想的长方形题目链接

二维上的滑动窗口 (滑动矩阵)。

只要分别求出每个 $n\times n$ 的正方形区域中的最大值和最小值，整个题目也就迎刃而解。

先对矩阵的每行做一次单调队列，某行第 $i$ 列元素存储同一行中区间 $[i - n + 1, i]$ 的最值；再对矩阵的每一列做一次单调队列，同上。

这样处理之后，以 $(i, j)$ 为右下顶点，边长为 $n$ 的正方形区域中的最值就是 $(i, j)$ 所存储的值。

代码实现：

#include <cstdio>
#include <algorithm>
using namespace std;

const int N = 1010;

int n, m, k;
int w[N][N];
int row_min[N][N], row_max[N][N];
int q[N];

void get_min(int a[], int b[], int tot){
    int hh = 0, tt = -1;
    for (int i = 1; i <= tot; i ++){
        if (hh <= tt && q[hh] <= i - k) hh ++;
        while (hh <= tt && a[q[tt]] >= a[i]) tt --;
        q[++ tt] = i;
        b[i] = a[q[hh]];
    }
}

void get_max(int a[], int b[], int tot){
    int hh = 0, tt = -1;
    for (int i = 1; i <= tot; i ++){
        if (hh <= tt && q[hh] <= i - k) hh ++;
        while (hh <= tt && a[q[tt]] <= a[i]) tt --;
        q[++ tt] = i;
        b[i] = a[q[hh]];
    }
}

int main(){
    scanf("%d %d %d", &n, &m, &k);
    
    for (int i = 1; i <= n; i ++)
        for (int j = 1; j <= m; j ++)
            scanf("%d", &w[i][j]);
            
    for (int i = 1; i <= n; i ++){
        get_min(w[i], row_min[i], m);
        get_max(w[i], row_max[i], m);
    }
    
    int a[N], b[N], c[N];
    int res = 1e9;
    for (int i = k; i <= m; i ++){
        for (int j = 1; j <= n; j ++) a[j] = row_min[j][i];
        get_min(a, b, n);
        
        for (int j = 1; j <= n; j ++) a[j] = row_max[j][i];
        get_max(a, c, n);
        
        for (int j = k; j <= n; j ++) res = min(res, c[j] - b[j]);
    }
    
    printf("%d", res);
    
    return 0;
}