結果

問題 No.703 ゴミ拾い Easy
コンテスト
ユーザー KowerKoint2010
提出日時 2023-06-18 20:25:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 361 ms / 1,500 ms
コード長 2,715 bytes
コンパイル時間 976 ms
コンパイル使用メモリ 78,844 KB
最終ジャッジ日時 2025-02-14 22:39:56
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

#line 1 "test/yukicoder-703.test.cpp"
#define PROBLEM "https://yukicoder.me/problems/no/703"

#line 2 "cpp/convex-hull-trick.hpp"

/**
 * @file convex-hull-tric.hpp
 * @brief Convex Hull Trick
 *
 * 一次関数で表される直線または線分の集合を管理し、あるxに対する最小値を求める
 */

#include <limits>
#include <type_traits>
#include <vector>

template <typename T, std::enable_if_t<std::is_scalar_v<T>, std::nullptr_t> = nullptr>
class LiChaoTree {
    int n, sz, height;
    std::vector<T> xs, as, bs;

    void update(T a, T b, int k, int h) {
        int l = (k << h) & (sz - 1);
        int r = l + (1 << h);
        
        while(1) {
            int m = (l + r) >> 1;
            T xl = xs[l], xm = xs[m];
            bool l_update = a*xl + b < as[k]*xl + bs[k];
            bool m_update = a*xm + b < as[k]*xm + bs[k];

            if(m_update) {
                std::swap(as[k], a);
                std::swap(bs[k], b);
            }
            if(h == 0) break;
            if(l_update != m_update) {
                k = k*2;
                r = m;
            } else {
                k = k*2+1;
                l = m;
            }
            h--;
        }
    }

public:
    LiChaoTree(const std::vector<T>& xs) : n(xs.size()), xs(xs) {
        sz = 1, height = 0;
        while(sz < (int)xs.size()) {
            sz <<= 1;
            height++;
        }
        this->xs.resize(sz, xs.back());
        as.assign(sz*2, 0);
        bs.assign(sz*2, std::numeric_limits<T>::max());
    }

    void add_line(T a, T b) {
        update(a, b, 1, height);
    }
    void add_segment(T a, T b, int l, int r) {
        if(l == r) return;
        l += sz, r += sz;
        int h = 0;
        while(l < r) {
            if(l & 1) update(a, b, l++, h);
            if(r & 1) update(a, b, --r, h);
            l >>= 1, r >>= 1, h++;
        }
    }

    T get_min(int i) const {
        T x = xs[i];
        int k = i + sz;
        T res = as[k]*x + bs[k];
        while(k > 1) {
            k >>= 1;
            T tmp = as[k]*x + bs[k];
            if(tmp < res) res = tmp;
        }
        return res;
    }
};
#line 4 "test/yukicoder-703.test.cpp"
#include <iostream>
#line 6 "test/yukicoder-703.test.cpp"

int main(void) {
    int n; std::cin >> n;
    std::vector<long long> a(n), x(n), y(n);
    for(int i = 0; i < n; i++) std::cin >> a[i];
    for(int i = 0; i < n; i++) std::cin >> x[i];
    for(int i = 0; i < n; i++) std::cin >> y[i];
    LiChaoTree<long long> cht(a);
    long long dp = 0;
    for(int i = 0; i < n; i++) {
        cht.add_line(-2*x[i], dp + x[i]*x[i] + y[i]*y[i]);
        dp = cht.get_min(i) + a[i]*a[i];
    }
    std::cout << dp << std::endl;
}
0