鱼C论坛

 找回密码
 立即注册
查看: 75|回复: 2

[技术交流] 斜率优化

[复制链接]
发表于 2025-8-10 21:56:47 | 显示全部楼层 |阅读模式

马上注册,结交更多好友,享用更多功能^_^

您需要 登录 才可以下载或查看,没有账号?立即注册

x
自学一下斜率优化,有一定的理解难度,我理了好久思路
P2356 任务安排
  1. // 易推出 dp 式: dp[i] = dp[j] + (sumc[i] - sumc[j]) * sumt[i] + s * (sumc[n] - sumc[j])
  2. // dp[j] = (s + sumt[i]) * sumc[j] + dp[i] - sumc[i] * sumt[i] - s * sumc[n]
  3. // 转化为一次函数形式 y = k * x + b, y 为 dp[j], k 为(s + sumt[i]), x 为 sumc[i],  其余为 b
  4. // 那么题意转化为 ---> 求 b 的最小值  --->   求直线 y 的最小截距
  5. // 对于所有已知点(sumc[j], dp[j]), y 的 k 固定, y 的最小截距就是直线 y 向上平移第一个碰到的点产生的截距
  6. // 因此维护一个下凸包即可, 即维护 (dp[i] - dp[j]) / (sumc[i] - sumc[j]) >= k 这么一个不等式,由于其的单调性易联想到单调队列
  7. // 由于 t 可以为负数,因此 k 不一定单调递增,但是我们维护的 (dp[i] - dp[j]) / (sumc[i] - sumc[j]) 具有单调性,因此二分跳跃至正确决策点即可
  8. // Created on 2025/8/10 20:37.
  9. #include <bits/stdc++.h>
  10. #define int long long
  11. #define rep(i, j, k) for(int i = j; i <= k; i++)
  12. #define reps(i, j, k) for(int i = j; i >= k; i--)
  13. using namespace std;

  14. const int _ = 3e5 + 7;
  15. int n, s, c[_], t[_], f[_], st[_], l = 0, r = 0;

  16. int x(int a, int b){
  17.     return c[b] - c[st[a]];
  18. }

  19. int y(int a, int b){
  20.     return f[b] - f[st[a]];
  21. }

  22. int get_pos(int i){
  23.     int k1 = l, k3 = r, k = t[i] + s;
  24.     while (k1 < k3){
  25.         int k2 = (k1 + k3) >> 1;
  26.         if (y(k2, st[k2 + 1]) >= x(k2, st[k2 + 1]) * k) k3 = k2;
  27.         else k1 = k2 + 1;
  28.     }
  29.     return k1;
  30. }

  31. signed main() {
  32.     ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);

  33.     cin >> n >> s;
  34.     rep(i, 1, n){
  35.         cin >> t[i] >> c[i];
  36.         t[i] += t[i - 1], c[i] += c[i - 1];
  37.     }

  38.     rep(i, 1, n){
  39.         int pos = st[get_pos(i)];
  40.         f[i] = f[pos] + (c[i] - c[pos]) * t[i] + s * (c[n] - c[pos]);
  41.         while (l < r && y(r - 1, st[r]) * x(r, i) >= y(r, i) * x(r - 1, st[r])) r--;
  42.         st[++r] = i;
  43.     }

  44.     cout << f[n];

  45.     return 0;
  46. }
复制代码


小小记录一下做的紫题,属于是很大进步了, 个人感觉这题从代码角度比 严格次小生成树 简单很多, 但是思想上难不少,考场还是喜欢 严格次小生成树 这类题目一点
小甲鱼最新课程 -> https://ilovefishc.com
回复

使用道具 举报

发表于 2025-8-10 22:02:20 | 显示全部楼层
根据你的描述,你正在学习斜率优化,并尝试解决任务安排问题(P5785)。你的推导思路正确,但代码存在一些问题。以下是修正后的简洁解答:

### 斜率优化核心思路
1. **状态转移方程**:  
   `dp[i] = min{ dp[j] + sumT[i]*(sumC[i]-sumC[j]) + s*(sumC[n]-sumC[j]) }`  
   其中 `sumT[i] = Σt[1..i]`, `sumC[i] = Σc[1..i]`。

2. **斜率优化转换**:  
   - 将方程变形为:`dp[j] = (sumT[i] + s) * sumC[j] + (dp[i] - sumT[i]*sumC[i] - s*sumC[n])`  
   - 令 `y = dp[j]`, `x = sumC[j]`, `k_i = sumT[i] + s`  
   - 目标:找到点 `(x, y)` 使截距最小(即 `dp[i]` 最小)

3. **维护下凸包**:  
   - 用单调队列维护候选点集,保证相邻点斜率递增(下凸包性质)。  
   - 由于 `k_i` 不单调(`t_i` 可能为负),需在凸包上二分查找第一个斜率 `≥k_i` 的线段左端点。

### 修正代码(附注释)
  1. #include <iostream>
  2. #define int long long
  3. #define rep(i, a, b) for (int i = a; i <= b; i++)
  4. using namespace std;

  5. const int N = 3e5 + 10;
  6. int n, s;
  7. int sumT[N], sumC[N], dp[N], q[N];

  8. // 斜率比较:判断 (q[r-1],q[r]) 与 (q[r],i) 的斜率大小(避免浮点)
  9. bool check(int a, int b, int c) {
  10.     int x1 = sumC[b] - sumC[a], y1 = dp[b] - dp[a];
  11.     int x2 = sumC[c] - sumC[b], y2 = dp[c] - dp[b];
  12.     return y1 * x2 >= y2 * x1; // 若成立,则删除队尾b点
  13. }

  14. signed main() {
  15.     cin >> n >> s;
  16.     rep(i, 1, n) {
  17.         int t, c;
  18.         cin >> t >> c;
  19.         sumT[i] = sumT[i - 1] + t;
  20.         sumC[i] = sumC[i - 1] + c;
  21.     }

  22.     int l = 0, r = 0; // 队列初始化(q[0]=0)
  23.     q[0] = 0;

  24.     rep(i, 1, n) {
  25.         int k = sumT[i] + s; // 当前斜率k_i
  26.         
  27.         // 二分查找最优决策点
  28.         int L = l, R = r;
  29.         while (L < R) {
  30.             int mid = (L + R) >> 1;
  31.             // 比较 mid 与 mid+1 的斜率
  32.             if (dp[q[mid + 1]] - dp[q[mid]] >= k * (sumC[q[mid + 1]] - sumC[q[mid]]))
  33.                 R = mid;
  34.             else
  35.                 L = mid + 1;
  36.         }
  37.         int j = q[L]; // 最优决策点j
  38.         
  39.         // 更新dp[i]
  40.         dp[i] = dp[j] + sumT[i] * (sumC[i] - sumC[j]) + s * (sumC[n] - sumC[j]);
  41.         
  42.         // 维护下凸包(删除队尾不满足凸性的点)
  43.         while (l < r && check(q[r - 1], q[r], i)) r--;
  44.         q[++r] = i; // 将i加入队尾
  45.     }
  46.     cout << dp[n] << endl;
  47.     return 0;
  48. }
复制代码


### 关键点说明
1. **凸包维护**:  
   - 队列 `q` 存储决策下标,保证相邻点斜率递增(下凸包)。  
   - 插入新点 `i` 时,若队尾两点 `(a,b)` 与 `(b,i)` 的斜率不满足凸性(即 `check()` 返回 `true`),则删除 `b`。

2. **二分查找**:  
   - 在队列中二分找到第一个满足 `(dp[j+1]-dp[j])/(sumC[j+1]-sumC[j]) >= k_i` 的 `j`,此时 `j` 是最优决策点。

3. **复杂度**:`O(n log n)`,适用于 `n ≤ 3e5`。

> 提示:若 `t_i ≥ 0`(`k_i` 单调递增),可直接用单调队列将复杂度降至 `O(n)`。本题因 `t_i` 可能为负,故需二分。

以上回复来自 FishC AI 小助理,如未能正确解答您的问题,请继续追问。
小甲鱼最新课程 -> https://ilovefishc.com
回复 支持 反对

使用道具 举报

发表于 2025-8-11 00:05:37 | 显示全部楼层
厉害
小甲鱼最新课程 -> https://ilovefishc.com
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

小黑屋|手机版|Archiver|鱼C工作室 ( 粤ICP备18085999号-1 | 粤公网安备 44051102000585号)

GMT+8, 2025-8-20 06:23

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

快速回复 返回顶部 返回列表