結果

問題 No.705 ゴミ拾い Hard
コンテスト
ユーザー beet
提出日時 2018-06-21 13:15:43
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 406 ms / 1,500 ms
コード長 1,217 bytes
コンパイル時間 1,990 ms
コンパイル使用メモリ 179,788 KB
実行使用メモリ 12,656 KB
最終ジャッジ日時 2024-06-30 17:35:17
合計ジャッジ時間 10,083 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using Int = long long;
//INSERT ABOVE HERE

template<typename T>
struct MonotoneMinima{
  using F = function<T(int, int)>;
  int n;
  F dist;
  T INF;
  vector<T> dp;
  MonotoneMinima(int n,F dist,T INF):
    n(n),dist(dist),INF(INF),dp(n+1,INF){}

  void dfs(int l,int r,int a,int b){
    if(l==r) return;
    int m=(l+r)>>1;
    T ma=INF;
    int mi=-1;
    for(int i=a;i<b;i++){
      T tmp=dp[i+1]+dist(i,m);
      if(tmp<ma) ma=tmp,mi=i;
    }
    dp[m]=min(dp[m],ma);
    dfs(l,m,a,mi+1);
    dfs(m+1,r,mi,b);
  }
  
  void solve(int l,int r){
    if(l+1==r){
      dp[l]=min(dp[l],dp[r]+dist(l,l));
      return;
    }
    int m=(l+r)>>1;
    solve(m,r);
    dfs(l,m,m,r);
    solve(l,m);
  }

  T solve(){
    dp[n]=0;
    solve(0,n);
    return dp[0];
  }
};

signed main(){
  int n;
  cin>>n;
  vector<Int> a(n),x(n),y(n);
  for(int i=0;i<n;i++) cin>>a[i];
  for(int i=0;i<n;i++) cin>>x[i];
  for(int i=0;i<n;i++) cin>>y[i];

  auto dist=
    [&](int i,int j)->Int{
      Int s=abs(a[i]-x[j]);
      Int t=abs(y[j]);
      return s*s*s+t*t*t;
    };

  const Int INF = 1e18;
  MonotoneMinima<Int> monge(n,dist,INF);
  
  cout<<monge.solve()<<endl;
  return 0;
}
0