浅谈斜率优化
概论
列出状态转移方程,如果能化简为以下的形式:
\[dp[i]=\min/\max(c[i]+d[j]+C) \]
此时我们就可以利用单调队列优化从做 \(O(n^2)\) 到 \(O(n)\) 的复杂度。
现在考虑更一般的情况,如果化简为以下形式:
\[dp[i]=\min/\max(a[i]\cdot b[j]+c[i]+d[j]+C) \]
此时单调队列优化就不再适用,就需要使用斜率优化。
斜率优化的思想就是通过剔除一些不可能成为最优解的决策点,使剩下的决策点形成一个凸包,再使用单调队列、二分、平衡树、CDQ 分治或者李超线段树来快速转移。
理论讲解
斜率优化有两种理解方式:
代数法
设决策点 \(j_2\) 优于 \(j_1\)。
则有:(\(<\) 对应 \(\max\),\(>\) 对应 \(\min\))
\[a[i]\cdot b[j_1]+c[i]+d[j_1]+C</>a[i]\cdot b[j_2]+c[i]+d[j_2]+C \]
化简得:
\[a[i]\cdot b[j_1]+d[j_1]</>a[i]\cdot b[j_2]+d[j_2] \]
令 \(Y(x)=d[x]\),\(X(x)=b[x]\),\(k=-a[i]\),再让 \(X(j_2)>X(j_1)\)。
移项得:
\[k</>\frac{Y(j_2)-Y(j_1)}{X(j_2)-X(j_1)} \]
这就是说,将两个决策点在平面直角坐标系中用两点表示出来,如果两点间斜率小于(或大于)\(k\),那么后面的点优于前面的点。
现在我们考虑有如下位置关系的三个点:
![](https://free4.yunpng.top/2025/02/04/67a181a6a0aab.png)
如图,\(k_2<k_1\)。
下面我们讨论 \(\max\) 的情况。
- 当 \(k<k_2<k_1\) 时,则 \(C\) 比 \(A\) 优,\(B\) 比 \(C\) 优,\(B\) 最优。
- 当 \(k_2\le k\le k_1\) 时,则 \(A\)、\(B\) 至少有一个比 \(C\) 优。
- 当 \(k_2<k_1<k\) 时,则 \(A\) 比 \(C\) 优,\(C\) 比 \(B\) 优,\(A\) 最优。
综上,无论何种情况,\(C\) 都不可能最优,可以删去。
推广以上的性质,对于所有的决策点:
![](https://free4.yunpng.top/2025/02/04/67a184879df24.png)
删去所有不可能为最优的决策点,发现可能为最优解的点会形成如下的一个下凸包:
![](https://free4.yunpng.top/2025/02/04/67a185f8d8147.png)
同理,对于 \(\min\) 的情况,会形成一个上凸包。
几何法
我们先讨论 \(\max\) 的情况:
\[dp[i]=\max(a[i]\cdot b[j]+c[i]+d[j]+C) \]
去掉 \(\max\),再移项得:
\[-d[j]=a[i]\cdot b[j]+c[i]-dp[i]+C \]
我们可以将 \(-d[j]\) 看作纵坐标,\(a[i]\) 看作斜率,\(b[j]\) 看作横坐标,\(c[i]-dp[i]+C\) 看作常数。
那么此时就可以把状态转移方程看作一个形如 \(y=kx+b\) 的直线方程 ,其中对于同一个 \(i\) 斜率 \(k=a[i]\) 是不变的。
要使 \(dp[i]\) 最大,就要使方程的截距 \(b\) 尽可能小。
将所有决策点 \((b[j],-d[j])\) 在平面直角坐标系中表示出来:
![](https://free4.yunpng.top/2025/02/04/67a184879df24.png)
寻找最优决策点的过程就可以看作平移一条斜率不变的直线,自下往上遇到的第一个点必然使截距最小,如图:
![](https://free4.yunpng.top/2025/02/04/67a18a86e340d.png)
事实上,这个点就是和前面点斜率小于等于当前斜率,和后面点斜率大于等于当前斜率的点。
现在我们考虑改变直线斜率,发现可能是最优解的决策点当且仅当在下面的下凸包上:
![](https://free4.yunpng.top/2025/02/04/67a185f8d8147.png)
不在下面的下凸包上的,无论斜率如何改变,都不可能是从下往上平移最小的点。
同理,对于 \(\min\) 的情况,会形成一个上凸包。
个人更喜欢几何法,比较直观,所以下面的例题都会采用几何法论证。
代码实现
对于不同的情况,我们有不同的插入决策点及查询最优解的方式(具体的可以看下面的例题):
决策点横坐标(\(b[j]\))严格增,直线方程斜率(\(a[i]\))严格增
- 插入:可以维护一个队列存储决策点编号 \(j\),由于决策点横坐标(\(b[j]\))严格增,再根据凸包斜率严格增(或减)的性质,在插入时只要队列倒数第二元素和队尾的斜率小于(或大于)队尾和要插入的点的斜率,就弹出队尾,直到保证斜率严格增(或减)。
- 查询:根据最优决策点和前面点斜率小于(或大于)等于当前斜率,和后面点斜率大于(或小于)等于当前斜率的点,再根据直线方程斜率(\(a[i]\))严格增,所以最优决策点横坐标严格增,不用保存前面的决策点,因此只要队首的两个元素斜率小于(或大于)当前斜率,就弹出队首,直到不能弹为止,队首就是最优决策点。
- 复杂度:\(O(n)\)。
决策点横坐标(\(b[j]\))严格增,直线方程斜率(\(a[i]\))不严格增
- 插入:由于决策点横坐标(\(b[j]\))严格增不变,同上。
- 查询:由于直线方程斜率(\(a[i]\))不严格增,所以最优决策点没有单调性了,此时只能根据凸包斜率的单调性二分找到和前面点斜率小于(或大于)等于当前斜率,和后面点斜率大于(或小于)等于当前斜率的点的点。
- 复杂度:\(O(n\log n)\)。
决策点横坐标(\(b[j]\))不严格增
由于横坐标都不严格增,也就是说可能加的点在两点之间,上面的方法就行不通了。
可以考虑利用动态维护凸包或者采用 CDQ 分治离线等方法,在此先不予陈述。
这里介绍一种思维含量低、代码量小、常数小的优秀算法:李超线段树。
事实上,李超线段树并不是维护动态凸包,而是利用本身的特性直接维护答案。
将一些不变量提出:
\[dp[i]=c[i]+C+\min/\max(b[j]\cdot a[i]+d[j]) \]
我们将 \(\min/\max\) 中的式子看作一个直线方程 \(y=kx+b\),其中 \(b[j]\) 是斜率,\(a[i]\) 是此时横坐标,\(d[j]\) 是截距。
发现此时就是要求若干个直线 \(y=b[j]\cdot x+d[j]\) 在 \(x=a[i]\) 时的最值,而这恰好是李超线段树所维护的东西。
因此,每次我们只需将直线 \(y=b[j]\cdot x+d[j]\) 插入李超树,再查询所有直线在 \(x=a[i]\) 时的最值更新答案即可。
时间复杂度:\(O(n\log n)\)。
例题
P5785 [SDOI2012] 任务安排 link
题目描述
机器上有 \(n\) 个需要处理的任务,它们构成了一个序列。这些任务被标号为 \(1\) 到 \(n\),因此序列的排列为 \(1 , 2 , 3 \cdots n\)。这 \(n\) 个任务被分成若干批,每批包含相邻的若干任务。从时刻 \(0\) 开始,这些任务被分批加工,第 \(i\) 个任务单独完成所需的时间是 \(T_i\)。在每批任务开始前,机器需要启动时间 \(s\),而完成这批任务所需的时间是各个任务需要时间的总和。
注意,同一批任务将在同一时刻完成 。每个任务的费用是它的完成时刻乘以一个费用系数 \(C_i\)。
请确定一个分组方案,使得总费用最小。
数据范围
\(T1:1\leq n \leq 5000,0\leq S\leq 50,1\leq T_i,C_i \leq 100\)
\(T2:1\leq n \leq 300000,0\leq S\leq 512,1\leq T_i,C_i \leq 100\)
\(T3:1\leq n \leq 300000,0\leq S\leq 512,0\leq C_i \leq 512,-512\leq T_i \leq 512\)
解题思路
\(T1:\)
朴素 DP,不需要优化。
我们对 \(c\) 和 \(t\) 做前缀和,用前缀和替代原来的数组。
-
状态表示:\(dp[i]\) 表示完成前 \(i\) 个任务的最小费用。
-
初始化:\(dp[0]=0\)。
-
状态转移:考虑分配成的最后一组,设最后一组为 \(j+1\sim i\),首先要在 \(dp[j]\) 的基础上加上 \(t[i] \cdot (c[i] - c[j])\),但是 \(s\) 貌似比较难处理,发现 \(s\) 会对后面所有的任务产生影响,因此为方便处理,我们可以提前把它对后面的总贡献 \(s \cdot (c[n] - c[j])\) 直接算进当前区间 ,因此有状态转移方程:
\[dp[i] = \min(dp[j] + t[i] \cdot (c[i] - c[j]) + s \cdot (c[n] - c[j])) \]
-
答案:\(dp[n]\)。
时间复杂度:\(O(n^2)\)。
\(T2:\)
因为 \(1\leq n \leq 300000\),\(O(n^2)\) 无法通过,发现状态转移方程符合斜率优化的条件,考虑斜率优化。
化简、移项得:
\[dp[j] = (t[i]+s)\cdot c[j]+dp[i]-t[i]\cdot c[j]-s\cdot c[n] \]
要使 \(dp[i]\) 更小,那么就要使截距尽可能小,因此决策点会形成一个下凸包。
由于 \(T_i,C_i \geq 1\),因此前缀和严格增,所以横坐标 \(c[j]\) 严格增,截距 \(t[i]+s\) 严格增,应用第一种情况即可。
\(T3:\)
此时 \(-512\leq T_i \leq 512\),所以截距不严格增,其他不变,在查询时二分找决策点即可。
编码技巧:在处理斜率时为避免误差,可以写作乘积的形式。
代码
\(T1:\)
cpp
#include<bits/stdc++.h>
#define endl "\n"
#define ll long long
using namespace std;
const int N = 5010;
int n, s;
ll t[N], f[N];
ll dp[N];
int main()
{
ios :: sync_with_stdio(0);
cin.tie(0), cout.tie(0);
cin >> n >> s;
for (int i = 1; i <= n; i++)
{
cin >> t[i] >> f[i];
t[i] = t[i - 1] + t[i];
f[i] = f[i - 1] + f[i];
}
memset(dp, 0x7f, sizeof dp);
dp[0] = 0;
for (int i = 1; i <= n; i++)
{
for (int j = 0; j < i; j++)
{
dp[i] = min(dp[i], dp[j] + t[i] * (f[i] - f[j]) + s * (f[n] - f[j]));
}
}
cout << dp[n] << endl;
return 0;
}
\(T2:\)
cpp
#include<bits/stdc++.h>
#define endl "\n"
using namespace std;
const int N = 3e5 + 10;
int n, s;
int t[N], c[N];
int dp[N], q[N];
int main()
{
ios :: sync_with_stdio(0);
cin.tie(0), cout.tie(0);
cin >> n >> s;
for (int i = 1; i <= n; i++)
{
cin >> t[i] >> c[i];
t[i] += t[i - 1];
c[i] += c[i - 1];
}
int hh = 0, tt = 0;
q[0] = 0;
for (int i = 1; i <= n; i++)
{
while (hh < tt && (dp[q[hh + 1]] - dp[q[hh]]) <= (t[i] + s) * (c[q[hh + 1]] - c[q[hh]]))
{
hh++;
}
dp[i] = dp[q[hh]] + t[i] * (c[i] - c[q[hh]]) + s * (c[n] - c[q[hh]]);
while (hh < tt && (dp[q[tt]] - dp[q[tt - 1]]) * (c[i] - c[q[tt]]) >= (dp[i] - dp[q[tt]]) * (c[q[tt]] - c[q[tt - 1]]))
{
tt--;
}
q[++tt] = i;
}
cout << dp[n] << endl;
return 0;
}
\(T3:\)
cpp
#include<bits/stdc++.h>
#define endl "\n"
#define ll long long
using namespace std;
const int N = 3e5 + 10;
int n, s;
ll t[N], c[N];
ll dp[N], q[N];
int main()
{
ios :: sync_with_stdio(0);
cin.tie(0), cout.tie(0);
cin >> n >> s;
for (int i = 1; i <= n; i++)
{
cin >> t[i] >> c[i];
t[i] += t[i - 1];
c[i] += c[i - 1];
}
int hh = 0, tt = 0;
q[0] = 0;
for (int i = 1; i <= n; i++)
{
int l = hh, r = tt;
while (l < r)
{
int mid = (l + r) >> 1;
if (dp[q[mid + 1]] - dp[q[mid]] > (t[i] + s) * (c[q[mid + 1]] - c[q[mid]]))
{
r = mid;
}
else
{
l = mid + 1;
}
}
dp[i] = dp[q[r]] + t[i] * (c[i] - c[q[r]]) + s * (c[n] - c[q[r]]);
while (hh < tt && (dp[q[tt]] - dp[q[tt - 1]]) * (c[i] - c[q[tt]]) >= (dp[i] - dp[q[tt]]) * (c[q[tt]] - c[q[tt - 1]]))
{
tt--;
}
q[++tt] = i;
}
cout << dp[n] << endl;
return 0;
}
P4655 [CEOI 2017] Building Bridges link
题目描述
有 \(n\) 根柱子依次排列,每根柱子都有一个高度。第 \(i\) 根柱子的高度为 \(h_i\)。
现在想要建造若干座桥,如果一座桥架在第 \(i\) 根柱子和第 \(j\) 根柱子之间,那么需要 \((h_i-h_j)^2\) 的代价。
在造桥前,所有用不到的柱子都会被拆除,因为他们会干扰造桥进程。第 \(i\) 根柱子被拆除的代价为 \(w_i\),注意 \(w_i\) 不一定非负,因为可能政府希望拆除某些柱子。
现在政府想要知道,通过桥梁把第 \(1\) 根柱子和第 \(n\) 根柱子连接的最小代价。注意桥梁不能在端点以外的任何地方相交。
数据范围
\(2\le n\le 10^5,0\le h_i,\vert w_i\vert\le 10^6\)。
解题思路
先用 \(w\) 的前缀和替代原来的序列。
考虑先列出 \(O(n^2)\) 的状态转移方程:
-
状态表示:\(dp[i]\) 表示用桥梁把前 \(i\) 根柱子连接的最小代价。
-
初始化:\(dp[1]=0\)
-
状态转移:考虑柱子 \(i\) 和 \(j\) 相接,则有:
\[dp[i]=\min(dp[j]+(h[i]-h[j])^2+w[i-1]-w[j]) \]
-
答案:\(dp[n]\)。
现在考虑斜率优化,移项:
\[dp[j]=2h[i]h[j]-h[j]^2-w[j]+dp[i]-h[i]^2+w[i-1] \]
发现由于原来的 \(h[i]\) 可以为 \(0\),因此横坐标 \(h[j]\) 不严格增,只能用李超线段树优化。
将式子化为:
\[dp[i]=h[i]^2+w[i-1]+\min(-2h[j]h[i]+dp[j]+h[j]^2-w[j]) \]
问题转化为:每次插入直线 \(y=-2h[j]\cdot x+dp[j]+h[j]^2-w[j]\),求在 \(x=h[i]\) 时的最小值。
用李超线段树维护即可,时间复杂度 \(O(n\log n)\)。
代码
cpp
#include<bits/stdc++.h>
#define endl "\n"
using namespace std;
const int N = 1e5 + 10, M = 1e6 + 10;
int n;
long long h[N], w[N];
long long dp[N];
struct line
{
long long k, b;
inline long long calc(long long x)
{
return k * (x - 1) + b;
}
} p[N];
struct SGT
{
#define ls x << 1
#define rs x << 1 | 1
#define mid ((l + r) >> 1)
int dat[M << 2];
void update(int x, int l, int r, int k)
{
if (!dat[x])
{
dat[x] = k;
return;
}
if (p[k].calc(mid) < p[dat[x]].calc(mid))
{
swap(k, dat[x]);
}
if (p[k].calc(l) < p[dat[x]].calc(l))
{
update(ls, l, mid, k);
}
if (p[k].calc(r) < p[dat[x]].calc(r))
{
update(rs, mid + 1, r, k);
}
}
long long query(int x, int l, int r, int k)
{
if (r < k || l > k)
{
return 1e18;
}
long long res = p[dat[x]].calc(k);
if (l == r)
{
return res;
}
return min(res, min(query(ls, l, mid, k), query(rs, mid + 1, r, k)));
}
} t;
int main()
{
ios :: sync_with_stdio(0);
cin.tie(0), cout.tie(0);
p[0].b = 1e18;
cin >> n;
for (int i = 1; i <= n; i++)
{
cin >> h[i];
}
for (int i = 1; i <= n; i++)
{
cin >> w[i];
w[i] = w[i] + w[i - 1];
}
p[1].k = -2 * h[1], p[1].b = h[1] * h[1] - w[1];
t.update(1, 1, M, 1);
for (int i = 2; i <= n; i++)
{
dp[i] = h[i] * h[i] + w[i - 1] + t.query(1, 1, M, h[i] + 1);
p[i].k = -2 * h[i], p[i].b = dp[i] + h[i] * h[i] - w[i];
t.update(1, 1, M, i);
}
cout << dp[n] << endl;
return 0;
}
P4072 [SDOI2016] 征途 link
题目描述
Pine 开始了从 \(S\) 地到 \(T\) 地的征途。
从 \(S\) 地到 \(T\) 地的路可以划分成 \(n\) 段,相邻两段路的分界点设有休息站。
Pine 计划用 \(m\) 天到达 \(T\) 地。除第 \(m\) 天外,每一天晚上 Pine 都必须在休息站过夜。所以,一段路必须在同一天中走完。
Pine 希望每一天走的路长度尽可能相近,所以他希望每一天走的路的长度的方差尽可能小。
帮助 Pine 求出最小方差是多少。
设方差是 \(v\),可以证明,\(v\times m^2\) 是一个整数。为了避免精度误差,输出结果时输出 \(v\times m^2\)。
数据范围
\(1 \le n \le 3000\)。
保证从 \(S\) 到 \(T\) 的总路程不超过 \(3\times 10^4\)。
\(2 \leq m \leq n+1\),每段路的长度为不超过 \(3 \times 10^4\) 的正整数。
解题思路
考虑先化简最后答案的式子(\(a\) 表示输入的每段路的长度):
\[s^2=\frac{(\bar a-a_1)^2+(\bar a-a_2)^2+\cdots+(\bar a-a_m)^2}{m} \]
\[s^2=\frac{m\bar a^2-2\bar a(a_1+a_2+\cdots+a_m)+a_1^2+a_2^2+\cdots+a_m^2}{m} \]
\[s^2=\frac{m(\frac{\sum_{i=1}^ma_i}{m})^2-2(\frac{\sum_{i=1}^ma_i}{m})(\sum_{i=1}^ma_i)+\sum_{i=1}^ma_i^2}{m} \]
\[s^2=-\frac{{(\sum_{i=1}^ma_i)}^2}{m^2}+\frac{\sum_{i=1}^ma_i^2}{m} \]
\[s^2\cdot m^2={-(\sum_{i=1}^ma_i)}^2+m\sum_{i=1}^ma_i^2 \]
发现右边第一项是定值,因此只要右边第二项最小即可,也就是问题转化为求最小平方和。
不妨用 \(a\) 的前缀和替代原来的序列。
于是开始动态规划:
-
状态表示:由于分成几段也会影响答案,因此要开两维表示,\(dp[i][j]\) 表示前 \(i\) 项分成 \(j\) 段的最小平方和。
-
初始化:\(dp[0][i]=0\)。
-
状态转移:设最后一段为 \(k+1\sim i\),则有:
\[dp[i][j]=\min(dp[k][j-1]+(a[i]-a[k])^2) \]
-
答案:把 \(dp[n][m]\) 代入上面的式子即可。
时间复杂度 \(O(n^3)\),无法通过,考虑斜率优化。
将式子化为:
\[dp[k][j-1]+a[k]^2=2a[i]a[k]+dp[i][j]-a[i]^2 \]
要使 \(dp[i][j]\) 更小,则截距就要更小,所以最优决策点组成一个下凸包。
由于 \(a[i]\) 严格增,所以横坐标和斜率都严格增,应用第一种情况即可。
比较值得注意的是,因为这道题是二维的,我们优化的是第一维,所以可以先枚举第二维,然后跑 \(m\) 遍斜率优化。
时间复杂度 \(O(m\log n)\)。
代码
cpp
#include<bits/stdc++.h>
#define endl "\n"
#define ll long long
using namespace std;
const int N = 3e3 + 10;
ll n, m;
ll a[N];
ll dp[N][N], q[N];
int main()
{
ios :: sync_with_stdio(0);
cin.tie(0), cout.tie(0);
cin >> n >> m;
for (int i = 1; i <= n; i++)
{
cin >> a[i];
a[i] += a[i - 1];
dp[i][1] = a[i] * a[i];
}
for (int j = 2; j <= m; j++)
{
int h = 0, t = 0;
q[0] = 0;
for (int i = 1; i <= n; i++)
{
while (h < t && (dp[q[h + 1]][j - 1] - dp[q[h]][j - 1] + a[q[h + 1]] * a[q[h + 1]] - a[q[h]] * a[q[h]]) <= 2 * a[i] * (a[q[h + 1]] - a[q[h]]))
{
h++;
}
dp[i][j] = dp[q[h]][j - 1] + (a[i] - a[q[h]]) * (a[i] - a[q[h]]);
while (h < t && (dp[q[t]][j - 1] + a[q[t]] * a[q[t]] - dp[q[t - 1]][j - 1] - a[q[t - 1]] * a[q[t - 1]]) * (a[i] - a[q[t]]) >= (dp[i][j - 1] + a[i] * a[i] - dp[q[t]][j - 1] - a[q[t]] * a[q[t]]) * (a[q[t]] - a[q[t - 1]]))
{
t--;
}
q[++t] = i;
}
}
cout << m * dp[n][m] - a[n] * a[n] << endl;
return 0;
}
P4027 [NOI2007] 货币兑换 link
题目描述
小 Y 最近在一家金券交易所工作。该金券交易所只发行交易两种金券:A 纪念券(以下简称 A 券)和 B 纪念券(以下简称 B 券)。每个持有金券的顾客都有一个自己的帐户。金券的数目可以是一个实数。
每天随着市场的起伏波动,两种金券都有自己当时的价值,即每一单位金券当天可以兑换的人民币数目。我们记录第 \(K\) 天中 A 券和 B 券的价值分别为 \(A_K\) 和 \(B_K\)(元/单位金券)。
为了方便顾客,金券交易所提供了一种非常方便的交易方式:比例交易法。
比例交易法分为两个方面:
a) 卖出金券:顾客提供一个 \([0, 100]\) 内的实数 \(OP\) 作为卖出比例,其意义为:将 \(OP\%\) 的 A 券和 \(OP\%\) 的 B 券以当时的价值兑换为人民币;
b) 买入金券:顾客支付 \(IP\) 元人民币,交易所将会兑换给用户总价值为 \(IP\) 的金券,并且,满足提供给顾客的 A 券和 B 券的比例在第 \(K\) 天恰好为 \(\mathrm{Rate}_ K\);
例如,假定接下来 \(3\) 天内的 \(A_K,B_K,\mathrm{Rate}_ K\) 的变化分别为:
时间 | \(A_K\) | \(B_K\) | \(\mathrm{Rate}_ K\) |
---|---|---|---|
第一天 | \(1\) | \(1\) | \(1\) |
第二天 | \(1\) | \(2\) | \(2\) |
第三天 | \(2\) | \(2\) | \(3\) |
假定在第一天时,用户手中有 \(100\) 元人民币但是没有任何金券。
用户可以执行以下的操作:
时间 | 用户操作 | 人民币(元) | A 券的数量 | B 券的数量 |
---|---|---|---|---|
开户 | 无 | \(100\) | \(0\) | \(0\) |
第一天 | 买入 \(100\) 元 | \(0\) | \(50\) | \(50\) |
第二天 | 卖出 \(50\%\) | \(75\) | \(25\) | \(25\) |
第二天 | 买入 \(60\) 元 | \(15\) | \(55\) | \(40\) |
第三天 | 卖出 \(100\%\) | \(205\) | \(0\) | \(0\) |
注意到,同一天内可以进行多次操作。
小 Y 是一个很有经济头脑的员工,通过较长时间的运作和行情测算,他已经知道了未来 \(N\) 天内的 A 券和 B 券的价值以及 \(\mathrm{Rate}\)。他还希望能够计算出来,如果开始时拥有 \(S\) 元钱,那么 \(N\) 天后最多能够获得多少元钱。
数据范围
\(N \le 10^5\),\(0 < A_K \leq 10\),\(0 < B_K\le 10\),\(0 < \mathrm{Rate}_K \le 100\),\(\mathrm{MaxProfit} \leq 10^9\)。
必然存在一种最优的买卖方案满足:
每次买进操作使用完所有的人民币,每次卖出操作卖出所有的金券。
解题思路
考虑 DP。
-
状态表示:\(dp[i]\) 表示前 \(i\) 天最多能获得多少钱。
-
初始化:\(dp[0]=S\)。
-
状态转移:设现在是第 \(i\) 天,要使钱最多那么一定不买金券。
-
不卖金券:\(dp[i]=dp[i-1]\)。
-
卖金券:设现在卖出的金券是在第 \(j\) 天买入的,如果卖出有利润,那么当时就应该花光所有钱买金券以获得最大收益。
设第 \(i\) 天最多能获得 \(x_i\) 张 \(A\) 券, \(y_i\) 张 \(B\) 券,则有:
\[x_i=dp_i\frac{R_i}{A_iR_i+B_i} \]
\[y_i=dp_i\frac{1}{A_iR_i+B_i} \]
所以:\(dp[i]=\max(x_jA_i+y_jB_i)\)。
综上,状态转移方程为:
\[dp[i]=\max(dp[i-1],\max(x_jA_i+y_jB_i)) \]
-
-
答案:\(dp[n]\)。
现在来考虑斜率优化,我们把它化成斜率优化能处理的式子(先不考虑不卖,最后把它算上即可):
\[y_j=-\frac{A_i}{B_i}x_j+\frac{1}{B_i}dp[i] \]
发现 \(x_j\) 没有单调性,因此考虑使用李超线段树。
变形得:
\[dp[i]=b[i]\max(x_j\frac{a_i}{b_i}+y_j) \]
问题转化为:每次插入直线 \(y=x_j\cdot x+y_j\),求在 \(x=\frac{a_i}{b_i}\) 时的最大值。
用李超线段树维护即可,时间复杂度 \(O(n\log n)\)。
值得注意的是,这道题需要查询实数位置的最小值,可以离散化,再将李超线段树的 clac()
函数改写即可。
代码
cpp
#include<bits/stdc++.h>
#define endl "\n"
using namespace std;
const int N = 1e5 + 10;
const double eps = 1e-10;
int n;
double a[N], b[N], c[N], d[N], r[N];
double ans;
struct line
{
double k, b;
int id;
inline double calc(int x)
{
return k * c[x] + b;
}
} p[N];
struct SGT
{
#define ls x << 1
#define rs x << 1 | 1
#define mid ((l + r) >> 1)
int dat[N << 2];
void update(int x, int l, int r, int k)
{
if (!dat[x])
{
dat[x] = k;
return;
}
if (p[k].calc(mid) - p[dat[x]].calc(mid) > eps)
{
swap(k, dat[x]);
}
if (p[k].calc(l) - p[dat[x]].calc(l) > eps || (p[k].calc(l) == p[dat[x]].calc(l) && k < dat[x]))
{
update(ls, l, mid, k);
}
if (p[k].calc(r) - p[dat[x]].calc(r) > eps || (p[k].calc(r) == p[dat[x]].calc(r) && k < dat[x]))
{
update(rs, mid + 1, r, k);
}
}
double query(int x, int l, int r, int k)
{
if (r < k || l > k)
{
return 0;
}
double res = p[dat[x]].calc(k);
if (l == r)
{
return res;
}
return max(res, max(query(ls, l, mid, k), query(rs, mid + 1, r, k)));
}
} t;
int main()
{
ios :: sync_with_stdio(0);
cin.tie(0), cout.tie(0);
cin >> n >> ans;
for (int i = 1; i <= n; i++)
{
cin >> a[i] >> b[i] >> r[i];
c[i] = d[i] = a[i] / b[i];
}
sort(c + 1, c + n + 1);
for (int i = 1; i <= n; i++)
{
int sum = lower_bound(c + 1, c + n + 1, d[i]) - c;
ans = max(ans, b[i] * t.query(1, 1, n, sum));
p[i].k = ans * r[i] / (a[i] * r[i] + b[i]), p[i].b = ans / (a[i] * r[i] + b[i]);
t.update(1, 1, n, i);
}
cout << fixed << setprecision(3) << ans << endl;
return 0;
}