#326555 (C++14) No.703 ゴミ拾い Easy

提出ソース

結果

問題	No.703 ゴミ拾い Easy
ユーザー	Haar
提出日時	2019-03-20 17:56:33
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	124 ms / 1,500 ms
コード長	4,047 bytes
コンパイル時間	2,767 ms
コンパイル使用メモリ	186,188 KB
実行使用メモリ	15,052 KB
最終ジャッジ日時	2023-08-30 19:04:40
合計ジャッジ時間	7,045 ms
ジャッジサーバーID （参考情報）	judge15 / judge12

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 4,376 KB
testcase_01	AC	1 ms 4,380 KB
testcase_02	AC	2 ms 4,376 KB
testcase_03	AC	2 ms 4,376 KB
testcase_04	AC	2 ms 4,376 KB
testcase_05	AC	1 ms 4,384 KB
testcase_06	AC	1 ms 4,384 KB
testcase_07	AC	2 ms 4,376 KB
testcase_08	AC	2 ms 4,380 KB
testcase_09	AC	1 ms 4,376 KB
testcase_10	AC	1 ms 4,380 KB
testcase_11	AC	1 ms 4,376 KB
testcase_12	AC	1 ms 4,380 KB
testcase_13	AC	2 ms 4,380 KB
testcase_14	AC	2 ms 4,376 KB
testcase_15	AC	2 ms 4,376 KB
testcase_16	AC	2 ms 4,376 KB
testcase_17	AC	2 ms 4,376 KB
testcase_18	AC	2 ms 4,384 KB
testcase_19	AC	2 ms 4,380 KB
testcase_20	AC	2 ms 4,376 KB
testcase_21	AC	2 ms 4,380 KB
testcase_22	AC	2 ms 4,376 KB
testcase_23	AC	2 ms 4,380 KB
testcase_24	AC	123 ms 14,884 KB
testcase_25	AC	122 ms 14,832 KB
testcase_26	AC	121 ms 14,896 KB
testcase_27	AC	124 ms 14,876 KB
testcase_28	AC	124 ms 14,888 KB
testcase_29	AC	123 ms 14,984 KB
testcase_30	AC	122 ms 15,052 KB
testcase_31	AC	122 ms 14,780 KB
testcase_32	AC	122 ms 14,988 KB
testcase_33	AC	123 ms 14,884 KB
testcase_34	AC	95 ms 14,872 KB
testcase_35	AC	92 ms 14,744 KB
testcase_36	AC	92 ms 14,812 KB
testcase_37	AC	94 ms 14,824 KB
testcase_38	AC	91 ms 14,816 KB
testcase_39	AC	91 ms 14,780 KB
testcase_40	AC	94 ms 14,604 KB
testcase_41	AC	96 ms 14,620 KB
testcase_42	AC	91 ms 14,740 KB
testcase_43	AC	95 ms 14,672 KB
testcase_44	AC	1 ms 4,376 KB
testcase_45	AC	2 ms 4,380 KB
testcase_46	AC	104 ms 14,632 KB
testcase_47	AC	106 ms 14,876 KB
testcase_48	AC	2 ms 4,376 KB
testcase_49	AC	2 ms 4,380 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
#define FOR(v, a, b) for(int v = (a); v < (b); ++v)
#define FORE(v, a, b) for(int v = (a); v <= (b); ++v)
#define REP(v, n) FOR(v, 0, n)
#define REPE(v, n) FORE(v, 0, n)
#define REV(v, a, b) for(int v = (a); v >= (b); --v)
#define ALL(x) (x).begin(), (x).end()
#define ITR(it, c) for(auto it = (c).begin(); it != (c).end(); ++it)
#define RITR(it, c) for(auto it = (c).rbegin(); it != (c).rend(); ++it)
#define EXIST(c,x) ((c).find(x) != (c).end())
#define LLI long long int
#define fst first
#define snd second
#ifndef M_PI
#define M_PI 3.14159265358979323846
#endif
#ifndef M_E
#define M_E 2.71828182845904523536
#endif

#ifdef DEBUG
#include <misc/C++/Debug.cpp>
#else
#define dump(x)
#endif

#define gcd __gcd

using namespace std;
template <class T> constexpr T lcm(T m, T n){return m/gcd(m,n)*n;}

template <typename I> void join(ostream &ost, I s, I t, string d=" "){for(auto i=s; i!=t; ++i){if(i!=s)ost<<d; ost<<*i;}ost<<endl;}
template <typename T> istream& operator>>(istream &is, vector<T> &v){for(auto &a : v) is >> a; return is;}
template <typename T, typename U> istream& operator>>(istream &is, pair<T,U> &p){is >> p.first >> p.second; return is;}

template <typename T, typename U> T& chmin(T &a, const U &b){return a = (a<=b?a:b);}
template <typename T, typename U> T& chmax(T &a, const U &b){return a = (a>=b?a:b);}
template <typename T, size_t N, typename U> void fill_array(T (&a)[N], const U &v){fill((U*)a, (U*)(a+N), v);}


template <typename T> class LiChaoSegmentTree{
  using line = pair<T,T>;

  struct node{
    line l;
    node *left, *right;
    node(const line &l): l(l), left(nullptr), right(nullptr){}
  };

  vector<T> xs;
  int n;
  node *root;
  T e;
  function<bool(T,T)> cmp;
  function<T(T,T)> chm;

public:
  LiChaoSegmentTree(const vector<T> &xs, const bool min_mode, const T &e): xs(xs), n(xs.size()), root(nullptr), e(e){
    if(min_mode){
      cmp = [](const T &a, const T &b){return a < b;};
      chm = [](const T &a, const T &b){return min(a,b);};
    }else{
      cmp = [](const T &a, const T &b){return a > b;};
      chm = [](const T &a, const T &b){return max(a,b);};
    }
  }

  T apply(const line &l, const T &x) const{
    return l.fst*x + l.snd;
  }

protected:
  node* add(node *t, const int left, const int right, line l){ // [l,r]
    if(!t) return new node(l);

    if(cmp(apply(t->l, xs[left]), apply(l, xs[left])) and cmp(apply(t->l, xs[right]), apply(l, xs[right]))){
      return t;
    }
    if(!cmp(apply(t->l, xs[left]), apply(l, xs[left])) and !cmp(apply(t->l, xs[right]), apply(l, xs[right]))){
      t->l = l;
      return t;
    }

    int mid = (left + right) / 2;

    if(!cmp(apply(t->l, xs[mid]), apply(l, xs[mid]))) swap(t->l, l);
    if(!cmp(apply(t->l, xs[left]), apply(l, xs[left]))){
      t->left = add(t->left, left, mid, l);
    }else{
      t->right = add(t->right, mid+1, right, l);
    }

    return t;
  }

public:
  void add_line(T a, T b){
    root = add(root, 0, n-1, make_pair(a,b));
  }

  

protected:
  T query(node *t, const int left, const int right, const int i) const{
    if(!t) return e;
    if(right == left) return apply(t->l, xs[i]);
    int mid = (left + right) / 2;
    if(i <= mid) return chm(apply(t->l, xs[i]), query(t->left, left, mid, i));
    else return chm(apply(t->l, xs[i]), query(t->right, mid+1, right, i));
  }
  
public:
  T query(const T &x) const{
    return query(root, 0, n-1, lower_bound(ALL(xs), x) - xs.begin());
  }
};


int main(){
  cin.tie(0);
  ios::sync_with_stdio(false);

  int n;
  while(cin >> n){
    vector<LLI> a(n), x(n), y(n); cin >> a >> x >> y;

    LiChaoSegmentTree<LLI> cht(a,true,LLONG_MAX);

    vector<LLI> dp(n);
    
    dp[0] = y[0]*y[0] + (x[0]-a[0])*(x[0]-a[0]);
    cht.add_line(-2*x[1], dp[0]+y[1]*y[1]+x[1]*x[1]);

    FOR(i,1,n){
      dp[i] = y[0]*y[0] + (a[i]-x[0])*(a[i]-x[0]);

      chmin(dp[i], cht.query(a[i]) + a[i]*a[i]);

      if(i<n-1) cht.add_line(-2*x[i+1], dp[i]+y[i+1]*y[i+1]+x[i+1]*x[i+1]);
    }
    cout << dp[n-1] << endl;
  }
  
  return 0;
}

yukicoder

結果

テストケース

ソースコード