【題目-任務安排2】斜率優化dp

題解

  • 首先,遞推關係如下:
    \(dp[i] = min(dp[i], dp[j] + sumt[i] * (sumc[i] - sumc[j]) + s * (sumc[n] - sumc[j]));\)
  • 顯然N太大,無法\(O(n^2)\)算法解決問題。考慮如何優化掉第二個j的循環,發現這個循環是找最優的j位置
  • 假設\(j\)就是最優位置,那麼可以先初步消掉min,接着如下分析:
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 3e5 + 5;

typedef long long ll;

ll dp[N];
ll sumt[N], sumc[N];

int q[N];

int main() {
    int n; ll s; scanf("%d%lld",&n, &s);
    for(int i = 1; i <= n; ++ i) {
        ll t, c; scanf("%lld%lld", &t, &c);
        sumt[i] = sumt[i - 1] + t;
        sumc[i] = sumc[i - 1] + c;
    }
    int h = 0, t = -1;
    
    for(int i = 1; i <= n; ++ i) dp[i] = 1e18;
    dp[0] = 0;
    
    q[++ t] = 0;
    for(int i = 1; i <= n; ++ i) {
        while(h < t && dp[q[h + 1]] - dp[q[h]] <= (sumt[i] + s) * (sumc[q[h + 1]] - sumc[q[h]])) h ++;
        dp[i] = dp[q[h]] + sumt[i] * (sumc[i] - sumc[q[h]]) + s * (sumc[n] - sumc[q[h]]);
        while (h < t && (dp[q[t]] - dp[q[t - 1]]) * (sumc[i] - sumc[q[t - 1]]) >= (dp[i] - dp[q[t - 1]]) * (sumc[q[t]] - sumc[q[t - 1]])) t --;
        q[++ t] = i; 
    }
    printf("%lld\n", dp[n]);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章