#117186 (C++14) No.409 ダイエット

提出ソース

結果

問題	No.409 ダイエット
ユーザー	koyumeishi
提出日時	2016-09-13 23:09:41
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	47 ms / 2,000 ms
コード長	8,088 bytes
コンパイル時間	1,192 ms
コンパイル使用メモリ	111,240 KB
実行使用メモリ	10,340 KB
最終ジャッジ日時	2024-06-26 10:00:39
合計ジャッジ時間	4,942 ms
ジャッジサーバーID （参考情報）	judge4 / judge5

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	2 ms 6,944 KB
testcase_02	AC	2 ms 6,944 KB
testcase_03	AC	2 ms 6,944 KB
testcase_04	AC	2 ms 6,944 KB
testcase_05	AC	2 ms 6,940 KB
testcase_06	AC	2 ms 6,940 KB
testcase_07	AC	2 ms 6,944 KB
testcase_08	AC	2 ms 6,940 KB
testcase_09	AC	2 ms 6,940 KB
testcase_10	AC	2 ms 6,940 KB
testcase_11	AC	2 ms 6,944 KB
testcase_12	AC	2 ms 6,944 KB
testcase_13	AC	2 ms 6,944 KB
testcase_14	AC	2 ms 6,940 KB
testcase_15	AC	2 ms 6,944 KB
testcase_16	AC	2 ms 6,940 KB
testcase_17	AC	2 ms 6,940 KB
testcase_18	AC	2 ms 6,944 KB
testcase_19	AC	2 ms 6,940 KB
testcase_20	AC	2 ms 6,944 KB
testcase_21	AC	2 ms 6,940 KB
testcase_22	AC	2 ms 6,940 KB
testcase_23	AC	2 ms 6,940 KB
testcase_24	AC	2 ms 6,940 KB
testcase_25	AC	3 ms 6,940 KB
testcase_26	AC	2 ms 6,940 KB
testcase_27	AC	2 ms 6,944 KB
testcase_28	AC	2 ms 6,944 KB
testcase_29	AC	2 ms 6,944 KB
testcase_30	AC	2 ms 6,940 KB
testcase_31	AC	2 ms 6,940 KB
testcase_32	AC	2 ms 6,944 KB
testcase_33	AC	2 ms 6,944 KB
testcase_34	AC	2 ms 6,940 KB
testcase_35	AC	2 ms 6,944 KB
testcase_36	AC	2 ms 6,944 KB
testcase_37	AC	2 ms 6,940 KB
testcase_38	AC	3 ms 6,940 KB
testcase_39	AC	2 ms 6,944 KB
testcase_40	AC	3 ms 6,940 KB
testcase_41	AC	3 ms 6,944 KB
testcase_42	AC	2 ms 6,944 KB
testcase_43	AC	3 ms 6,944 KB
testcase_44	AC	2 ms 6,940 KB
testcase_45	AC	2 ms 6,944 KB
testcase_46	AC	2 ms 6,940 KB
testcase_47	AC	2 ms 6,940 KB
testcase_48	AC	3 ms 6,944 KB
testcase_49	AC	3 ms 6,940 KB
testcase_50	AC	2 ms 6,940 KB
testcase_51	AC	2 ms 6,940 KB
testcase_52	AC	3 ms 6,940 KB
testcase_53	AC	2 ms 6,944 KB
testcase_54	AC	2 ms 6,944 KB
testcase_55	AC	21 ms 6,940 KB
testcase_56	AC	38 ms 9,984 KB
testcase_57	AC	41 ms 10,340 KB
testcase_58	AC	19 ms 6,940 KB
testcase_59	AC	27 ms 7,680 KB
testcase_60	AC	18 ms 6,940 KB
testcase_61	AC	37 ms 9,344 KB
testcase_62	AC	38 ms 9,856 KB
testcase_63	AC	36 ms 9,344 KB
testcase_64	AC	20 ms 6,944 KB
testcase_65	AC	39 ms 9,600 KB
testcase_66	AC	41 ms 10,112 KB
testcase_67	AC	31 ms 8,576 KB
testcase_68	AC	26 ms 7,552 KB
testcase_69	AC	36 ms 9,216 KB
testcase_70	AC	39 ms 8,960 KB
testcase_71	AC	23 ms 6,948 KB
testcase_72	AC	47 ms 10,112 KB
testcase_73	AC	45 ms 9,600 KB
testcase_74	AC	28 ms 7,168 KB
testcase_75	AC	44 ms 9,600 KB
testcase_76	AC	32 ms 7,936 KB
testcase_77	AC	23 ms 6,940 KB
testcase_78	AC	24 ms 6,940 KB
testcase_79	AC	21 ms 6,940 KB
testcase_80	AC	44 ms 9,740 KB
testcase_81	AC	44 ms 9,600 KB
testcase_82	AC	32 ms 8,064 KB
testcase_83	AC	36 ms 8,192 KB
testcase_84	AC	30 ms 7,424 KB
testcase_85	AC	5 ms 6,944 KB
testcase_86	AC	29 ms 7,296 KB
testcase_87	AC	39 ms 8,832 KB
testcase_88	AC	17 ms 6,940 KB
testcase_89	AC	35 ms 8,064 KB
testcase_90	AC	17 ms 6,940 KB
testcase_91	AC	37 ms 8,448 KB

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

ソースコード

raw source code

#include <iostream>
#include <vector>
#include <cstdio>
#include <sstream>
#include <map>
#include <string>
#include <algorithm>
#include <queue>
#include <cmath>
#include <functional>
#include <set>
#include <ctime>
#include <random>
#include <chrono>
#include <cassert>
using namespace std;

namespace {
  using Integer = long long; //__int128;
  template<class T> istream& operator >> (istream& is, vector<T>& vec){for(T& val: vec) is >> val; return is;}
  template<class T> istream& operator ,  (istream& is, T& val){ return is >> val;}
  template<class T> ostream& operator << (ostream& os, const vector<T>& vec){for(int i=0; i<vec.size(); i++) os << vec[i] << (i==vec.size()-1?"":" "); return os;}
  template<class T> ostream& operator ,  (ostream& os, const T& val){ return os << " " << val;}

  template<class H> void print(const H& head){ cout << head; }
  template<class H, class ... T> void print(const H& head, const T& ... tail){ cout << head << " "; print(tail...); }
  template<class ... T> void println(const T& ... values){ print(values...); cout << endl; }

  template<class H> void eprint(const H& head){ cerr << head; }
  template<class H, class ... T> void eprint(const H& head, const T& ... tail){ cerr << head << " "; eprint(tail...); }
  template<class ... T> void eprintln(const T& ... values){ eprint(values...); cerr << endl; }

  class range{
    long long start_, end_, step_;
   public:
    struct range_iterator{
      long long val, step_;
      range_iterator(long long v, long long step) : val(v), step_(step) {}
      long long operator * (){return val;}
      void operator ++ (){val += step_;}
      bool operator != (range_iterator& x){return step_ > 0 ? val < x.val : val > x.val;}
    };
    range(long long len) : start_(0), end_(len), step_(1) {}
    range(long long start, long long end) : start_(start), end_(end), step_(1) {}
    range(long long start, long long end, long long step) : start_(start), end_(end), step_(step) {}
    range_iterator begin(){ return range_iterator(start_, step_); }
    range_iterator   end(){ return range_iterator(  end_, step_); }
  };

  string operator "" _s (const char* str, size_t size){ return move(string(str)); }
  constexpr Integer my_pow(Integer x, Integer k, Integer z=1){return k==0 ? z : k==1 ? z*x : (k&1) ? my_pow(x*x,k>>1,z*x) : my_pow(x*x,k>>1,z);}
  constexpr Integer my_pow_mod(Integer x, Integer k, Integer M, Integer z=1){return k==0 ? z%M : k==1 ? z*x%M : (k&1) ? my_pow_mod(x*x%M,k>>1,M,z*x%M) : my_pow_mod(x*x%M,k>>1,M,z);}
  constexpr unsigned long long operator "" _ten (unsigned long long value){ return my_pow(10,value); }

  inline int k_bit(Integer x, int k){return (x>>k)&1;} //0-indexed

  mt19937 mt(chrono::duration_cast<chrono::nanoseconds>(chrono::steady_clock::now().time_since_epoch()).count());

  template<class T> string join(const vector<T>& v, const string& sep){
    stringstream ss; for(int i=0; i<v.size(); i++){ if(i>0) ss << sep; ss << v[i]; } return ss.str();
  }

  string operator * (string s, int k){ string ret; while(k){ if(k&1) ret += s; s += s; k >>= 1; } return ret; }

}

constexpr long long mod = 9_ten + 7;

int main(){
  long long n,a,b,w;
  cin >> n,a,b,w;

  vector<long long> d(n);
  //cin >> d;
  for(auto i : range(n) ){
    scanf("%lld", &d[i]);
  }

  vector<long long> acc(n+1, 0);
  for(auto i : range(n+1) ){
    acc[i] = -(i)*a + (i)*(i+1)/2 * b; //acc[i] -a + (i+1)*b;
  }

  vector<long long> dp(n+1, 1ll<<50);
  dp[0] = 0;

  deque<pair<long long,long long>> L;
  auto get_v = [&](long long x) -> long long{
    if(L.size()==0) return 0;
    while(L.size() >= 2){
      long long v1 = L[0].first * x + L[0].second;
      long long v2 = L[1].first * x + L[1].second;
      if(v1<v2) return v1;
      L.pop_front();
    }
    return L[0].first * x + L[0].second;
    
    long long lb = 0;
    long long ub = L.size();
    while(ub-lb>1){
      long long mid = (lb+ub)/2;
      bool ok = [&](){
        long long v1 = L[mid-1].first * x + L[mid-1].second;
        long long v2 = L[mid  ].first * x + L[mid  ].second;
        return v1>v2;
      }();
      (ok? lb : ub) = mid;
    }
    while(lb){
      L.pop_front();
      lb--;
    }
    return L[0].first * x + L[0].second;
  };

  auto check = [](pair<long long,long long> a, pair<long long,long long> b, pair<long long,long long> c){
    long long a21 = b.first - a.first;
    long long a23 = b.first - c.first;
    long long b32 = c.second - b.second;
    long long b12 = a.second - b.second;
    return a21*b32 >= a23*b12;
  };

  auto add_l = [&](pair<long long, long long> l){
    while(L.size() >= 2 && check(L[L.size()-2], L[L.size()-1], l)){
      L.pop_back();
    }
    L.push_back(l);
  };

  add_l({ -(2*0+1)*b -2*a, 2*dp[0] + 2*a*(0+1) + 0*(0+1)*b });
  for(auto i : range(1, n+1) ){
    dp[i] = 2*d[i-1] + i*i*b + get_v(i);
    pair<long long,long long> nx = { -(2*i+1)*b -2*a, dp[i] + 2*a*(i+1) + i*(i+1)*b };
    add_l(nx);
  }

  long long ans = 1ll<<50;
  for(auto i : range(n+1) ){
    ans = min(ans, dp[i]/2 + acc[n-i] + w);
  }
  println(ans);

  return 0;
}

/*
                                                                                
                                   ### #                                        
                              ############# #                                   
                          #######################                               
                     ################################                           
                 #########################################                      
              ##############################################                    
             #     #####################################     #                  
            ######     ############################     ######                  
            ###########     ###################   # ##########                  
            ################    #########      ###############                  
            ####################        # ####################                  
            #######   ##############  ########################                  
            ########       #########  ########################                  
            ###########      #######  ########################                  
            ############## # #######  ########################                  
            ########################  ########################                  
            ########################  ########################                  
            ########################  ########################                  
              ######################  #######################                   
                 ###################  ###################                       
                      ##############  ###############                           
                          ##########  ##########                                
                              ######  ######                                    
                                  ##  ##                                        
                                                                                
                                                                                
                                                                                
                        #                             #                         
                        #      #                      #                         
        #    #  #    #  #   #      ####   ####     ####   ####   ####           
         #   #  #    #  # ##   #  #      #    #  #    #  #____#  ##             
          # #/  #    #  ## #   #  #      #    #  #    #  #^^^^   #              
           ##    ###/#  #   #  #   ####   ####    ###/#   ####   #              
           #                                                                    
         ##                                                                     
                                                                                
*/

yukicoder

結果

テストケース

ソースコード