DP的优化

参考资料:

李煜东《算法竞赛进阶指南》

斜率优化

形如:

\(f[i] = min\{f[j]+val(i,j)\}\)的dp,多项式\(val(i,j)\)包含\(i,j\)的乘积项

引入一个例题:

[HNOI2008]玩具装箱TOY

\(dp[i] = min\{dp[j] + (sum[i] + i - sum[j] - j-L-1)^2\}(j < i)\)

定义:\(a[i]=sum[i]+i,b[i]=sum[i]+i+L+1\)

\[dp[i]=dp[j]+(a[i]-b[j])^2\\ dp[i]=dp[j]+a[i]^2-2*a[i]*b[j]+b[j]^2\\ 2*a[i]*b[j]+dp[i]-a[i]^2=dp[j]+b[j]^2\\ \]

\(b[j]\)看作x,\(dp[j]+b[j]^2\)看作y

那么这个式子就是一个\(k=2*a[i]\)的直线

那么我们要最小化这个过\((b[j],dp[j]+b[j]^2)\)的直线的\(b\)

根据定义,\(a[i]\)单调递增,那么直线的斜率越来越大

我们可以用单调队列维护下凸壳,不断用\(a[i]\)去切这个凸包

习题:

[APIO2014]序列分割

[APIO2010]特别行动队

[CEOI2004]锯木厂选址

wqs二分/带权二分/凸优化

解决n个东西恰好选m个的最优问题

普通dp方法一般都是\(O(nm)\)

wqs二分可以优化成\(O(nlogW)\)

引入例题:

有n个物品,每个物品有一个价值,

问:选m个物品可以获得的最大价值。

这显然可以贪心,但是我们为了引出wqs二分,我们想dp做法

\(f[i][j]=max(f[i-1][j-1]+val[i],f[i-1][j])\)

考虑如果没有m的限制,那么dp式子长这样

\(f[i]=f[i-1]+max(v[i],0)\)

wqs二分又称带权二分

那么我们二分这个带权x

给每个物品价值增加x

显然x越大,我们会选的东西不会变少

如果选的个数大于m,那么x要减小,否则增大

最后\(ans=f[n]-x*m\)

整数二分x可能会有些精度问题,最好是写实数二分

update:
\(wqs\)二分的本质是求一个凸函数某个位置的值
但是这有时候并不是很好求。我们转化成用一条斜率为\(mid\)的直线去切这个函数。
关于整数二分的问题,对于一个切不到刚好选出\(m\)个的函数,那么显然存在多个相同的,
例如上述的贪心\(5,3,3\)中选\(2\)
如果\(mid=3\),则变成\(2,0,0\)选一个
如果\(mid=2\),则变成\(3,1,1\)选三个
这个时候我们用\(mid=3\)作为答案是对想想为什么?

习题:

[八省联考2018]林克卡特树lct

四边形不等式优化

这个东西比较难,看不懂的可以记住结论。(证明由于篇幅过长,有兴趣的可以去查找一下资料)

​ 设\(w(x,y)\)是定义在整数集合的二元函数。若对于定义域上的任意整数\(a,b,c,d\),其中\(a\leq b \leq c \leq d\),都有\(w(a,d)+w(b,c)≥w(a,c)+w(b,d)\)成立,则称函数\(w\)满足四边形不等式。

定理(四边形不等式的另一种定义):对于定义域上任意整数\(a,b\),其中\(a<b\),都有\(w(a,b+1)+w(a+1,b)≥w(a,b)+w(a+1,b+1)\)成立,则函数满足四边形不等式。

一维线性dp的四边形不等式优化

​ 形如:\(f[i]=min\{f[j]+val(j,i)\}\)的dp,记\(p[i]\)为令\(f[i]\)取到最小值的j的值,即\(p[i]\)\(f[i]\)的最优决策。若\(p\)\([1,N]\)上单调不减,则称\(f\)具有决策单调性

定理(决策单调性):在上述dp中,若\(val\)满足四边形不等式,则\(f\)满足决策单调性。

那么,当\(F\)有决策单调性时,我们可以把复杂度降低到\(O(nlogn)\)

考虑维护\(p\)数组。最初\(p\)数组全部为0。在\(i\)循环进行的任意时刻,根据\(p[i]\)的单调性,情况应该如下图所示:

\(p:[j1,j1,j2,j3,j3,j3,j4,j4,j5,j5,j5]\) \((j1<j2<j3<j4<j5)\)

当求出一个新的\(f[i]\)时,我们应该考虑\(i\)可以作为哪些\(F[i'](i'>i)\)的最优决策。根据决策单调性,最终我们会找到一个位置,在该位置之前,p数组原来的决策比i好,之后的比i差。这个位置可以二分求得。

怎么修改?

直接修改数组效率肯定不行,我们可以建立一个队列保存三元组\((j,l,r)\)\(j\)表示决策,\(l,r\)表示 \(p[ l-r ]\) 值都为 \(j\)

然后像单调队列一样维护即可。

例题:诗人小G

按照上面的模拟即可,注意用long double(精度高) 代替long long。

#include<bits/stdc++.h>

#define LL long long
#define RG register

using namespace std;
template<class T> inline void read(T &x) {
    x = 0; RG char c = getchar(); bool f = 0;
    while (c != '-' && (c < '0' || c > '9')) c = getchar(); if (c == '-') c = getchar(), f = 1;
    while (c >= '0' && c <= '9') x = x*10+c-48, c = getchar();
    x = f ? -x : x;
    return ;
}
template<class T> inline void write(T x) {
    if (!x) {putchar(48);return ;}
    if (x < 0) x = -x, putchar('-');
    int len = -1, z[20]; while (x > 0) z[++len] = x%10, x /= 10;
    for (RG int i = len; i >= 0; i--) putchar(z[i]+48);return ;
}
#define ld long double
const int N = 100010;
int n, L, P;
ld sum[N], f[N];
char str[N][35];
ld fpow(ld a, int b) {
    ld res = 1;
    for (; b; b >>= 1, a = a * a) if (b & 1) res = res * a;
    return res;
}
ld getans(int l, int r) {
    return f[l] + fpow(abs(sum[r] - sum[l] + (r - l - 1) - L), P);
}
struct node {
    int j, l, r;
} q[N];
int find(int l, int r, int j, int i) {
    int ans = -1;
    while (l <= r) {
        int mid = (l + r) >> 1;
        getans(j, mid) >= getans(i, mid) ? ans = mid, r = mid - 1 : l = mid + 1;
    }
    return ans;
}
int p[N], que[N];
void solve() {
    read(n), read(L), read(P);
    for (int i = 1; i <= n; i++) {
        scanf("%s", str[i]);
        sum[i] = sum[i - 1] + (int) strlen(str[i]);
    }
    int l = 0, r = 0;
    q[l] = (node) {0, 1, n};
    for (int i = 1; i <= n; i++) {
        if (q[l].r < i) l++;
        p[i] = q[l].j; q[l].l = i;
        f[i] = getans(q[l].j, i);
        while (l <= r && getans(q[r].j, q[r].l) >= getans(i, q[r].l)) r--;
        if (getans(q[r].j, q[r].r) < getans(i, q[r].r)) { if (q[r].r < n) q[r + 1] = (node) {i, q[r].r + 1, n}, r++; }
        else {
            int pos = find(q[r].l, q[r].r, q[r].j, i);
            q[r].r = pos - 1;
            q[++r] = (node) {i, pos, n};
        }
    }
    if (f[n] > 1e18) puts("Too hard to arrange");
    else {
        printf("%lld\n", (LL) f[n]);
        int now = n, tot = 0;
        while (now) que[++tot] = now, now = p[now];
        que[tot + 1] = 0;
        for (int i = tot; i; i--) {
            for (int j = que[i + 1] + 1; j < que[i]; j++)
                printf("%s ", str[j]);
            puts(str[que[i]]);
        }
            
    }
    puts("--------------------");
    return ;
}
int main() {
    int T;
    read(T);
    while (T--) solve();
    return 0;
}

二维区间DP的四边形不等式优化

区间DP问题

一类区间问题的转移方程:

\(f[i][j]=min\{f[i][k]+f[k+1][j]+w(i,j)\}\)

定理

​ 在上述方程中,如果\(w\)满足四边形不等式,且\(w(a,d)≥w(b,c)[a≤b≤c≤d]\)

那么\(f\)也满足四边形不等式

二维决策单调性

​ 记\(p[i][j]\)为令\(f[i][j]\)取到最小值的\(k\)值,如果\(f\)满足四边形不等式,那么对于任意\(i<j\),有\(p[i][j-1]≤p[i][j]≤p[i+1][j]\)

例题:[IOI2000]邮局

先写出转移方程:\(f[i][j]=min\{f[i][j],f[k][j−1]+dis(k+1,i)\}\)

显然这个\(dis\)满足四边形不等式。

那么这个\(f\)满足决策点单调性

#include<bits/stdc++.h>

#define LL long long
#define RG register

using namespace std;
template<class T> inline void read(T &x) {
	x = 0; RG char c = getchar(); bool f = 0;
	while (c != '-' && (c < '0' || c > '9')) c = getchar(); if (c == '-') c = getchar(), f = 1;
	while (c >= '0' && c <= '9') x = x*10+c-48, c = getchar();
	x = f ? -x : x;
	return ;
}
template<class T> inline void write(T x) {
	if (!x) {putchar(48);return ;}
	if (x < 0) x = -x, putchar('-');
	int len = -1, z[20]; while (x > 0) z[++len] = x%10, x /= 10;
	for (RG int i = len; i >= 0; i--) putchar(z[i]+48);return ;
}
const int N = 3010;
int a[N], n, m, p[N][N], sum[N], f[N][N], w[N][N];

int main() {
	read(n), read(m);
	for (int i = 1; i <= n; i++) read(a[i]);
	sort(a + 1, a + 1 + n);
	for (int i = 1; i <= n; i++) sum[i] = sum[i - 1] + a[i];
	for (int i = 1; i < n; i++)
		for (int j = i + 1; j <= n; j++) {
			int mid = (i + j) >> 1;
			w[i][j] = a[mid] * (mid - i) - (sum[mid - 1] - sum[i - 1]);
			w[i][j] += (sum[j] - sum[mid]) - a[mid] * (j - mid);
		}
	memset(f, 63, sizeof(f));
	for (int i = 1; i <= n; i++)
		f[i][1] = w[1][i];
	for (int j = 2; j <= m; j++) {
		p[n + 1][j] = n;
		for (int i = n; i >= 1; i--)
			for (int k = p[i][j - 1]; k <= p[i + 1][j]; k++)
				if (f[i][j] > f[k][j - 1] + w[k + 1][i]) {
					f[i][j] = f[k][j - 1] + w[k + 1][i];
					p[i][j] = k;
				}
	}
	printf("%lld\n", f[n][m]);
	return 0;
}
posted @ 2019-07-01 17:38  zzy2005  阅读(234)  评论(0编辑  收藏  举报