結果

問題 No.703 ゴミ拾い Easy
ユーザー tubuanntubuann
提出日時 2019-06-24 11:45:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 707 ms / 1,500 ms
コード長 3,016 bytes
コンパイル時間 1,429 ms
コンパイル使用メモリ 140,032 KB
最終ジャッジ日時 2025-01-07 05:07:28
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iomanip>
#include<limits>
#include<thread>
#include<utility>
#include<iostream>
#include<string>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<numeric>
#include<cassert>
#include<random>
#include<chrono>
#include<unordered_set>
#include<unordered_map>
#include<fstream>
#include<list>
#include<functional>
#include<bitset>
#include<complex>
#include<tuple>
using namespace std;
typedef unsigned long long int ull;
typedef long long int ll;
typedef pair<ll,ll> pll;
typedef long double D;
typedef complex<D> P;
#define F first
#define S second
const ll E=1e18+7;
const ll MOD=1000000007;


template<typename T,typename U>istream & operator >> (istream &i,pair<T,U> &A){i>>A.F>>A.S; return i;}
template<typename T>istream & operator >> (istream &i,vector<T> &A){for(auto &I:A){i>>I;} return i;}
template<typename T,typename U>ostream & operator << (ostream &o,pair<T,U> &A){o<<A.F<<" "<<A.S; return o;}
template<typename T>ostream & operator << (ostream &o,vector<T> &A){ll i=A.size(); for(auto &I:A){o<<I<<(--i?" ":"");} return o;}
template<typename T>vector<T> & cset(vector<T> &A,T e=T()){for(auto &I:A){I=e;} return A;}



namespace Monotone_Minima{
    //argmin(cul(i,*))<=argmin(cul(j,*)) for(i<j)
    //を満たすn行m列の行列で、各行の最小値を求め、そのindexを返す。
    template<typename T>
    vector<int> argmin(function<T(int,int)> cul,int n,int m,function<bool(T,T)> comp=less<T>()){
        vector<int> ret(n,-1);
        int i=2;
        while(i<n){i<<=1;}
        while(i>>=1){
            for(int j=0;j<n;j+=i){
                if(ret[j]!=-1){continue;}
                int l=(j-i>=0 && ret[j-i]!=-1?ret[j-i]:0),r=(j+i<n && ret[j+i]!=-1?ret[j+i]+1:m);
                T a=cul(j,l);
                ret[j]=l;
                for(int k=l+1;k<r;k++){
                    if(comp(cul(j,k),a)){
                        a=cul(j,k);
                        ret[j]=k;
                    }
                }
            }
        }
        return ret;
    }
    
    template<typename T>
    vector<T> divide_and_conquer(function<T(int,int)> cul,int n,T e=T(),function<bool(T,T)> comp=less<T>()){
        vector<T> ret(n,e);
        for(int i=1;i<n;i++){ret[i]=ret[0]+cul(i,0);}
        for(int i=1;i<n;i++){
            int j=i&(-i);
            int k=i-j;
            auto fnc=[&](int a,int b){return ret[k+b]+cul(i+a,k+b);};
            vector<int> am=argmin<T>(fnc,min(j,n-i),j,comp);
            for(int s=0;s<j && i+s<n;s++){
                T a=fnc(s,am[s]);
                if(comp(a,ret[i+s])){ret[i+s]=a;}
            }
        }
        return ret;
    }
}

using namespace Monotone_Minima;


int main(){
    ll n;
    cin>>n;
    vector<ll> A(n),X(n),Y(n);
    cin>>A>>X>>Y;
    function<ll(int,int)> cul=[&](int i,int j){i--; return i<j?1e15:(A[i]-X[j])*(A[i]-X[j])+Y[j]*Y[j];};
    vector<ll> ans=divide_and_conquer<ll>(cul,n+1,0);
    cout<<ans.back()<<endl;
    
    
    return 0;
}
0