結果

問題 No.1496 不思議な数え上げ
ユーザー 👑 Nachia
提出日時 2021-04-30 23:00:23
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,377 bytes
コンパイル時間 1,240 ms
コンパイル使用メモリ 96,968 KB
最終ジャッジ日時 2025-01-21 04:23:21
ジャッジサーバーID
（参考情報） 		judge5 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other AC * 5 WA * 34
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <iostream>
#include <vector>
#include <algorithm>
#include <atcoder/segtree>
using ll = long long;
using ull = unsigned long long;
using namespace std;
int RMQop(int l,int r){ return min(l,r); }
int RMQe(){ return 1001001001; }
using RMQ = atcoder::segtree<int,RMQop,RMQe>;
int N;
vector<ll> A;
vector<int> P;
vector<int> I;
vector<ll> SA;
vector<ll> ans;
RMQ G;
void solve(int l,int r){
  if(l == r) return;
  int v = G.prod(l,r);
  int vp = I[v];
  ll ansv = 0;
  if(vp-l > r-vp){
    int pl = l;
    for(int pr=vp+1; pr<=r; pr++){
      int pl = lower_bound(SA.begin()+l, SA.begin()+vp+1, SA[pr]-A[v]) - SA.begin();
      ansv += vp - pl + 1;
    }
  }
  else{
    for(int pl=vp; pl>=l; pl--){
      int pr = upper_bound(SA.begin()+vp, SA.begin()+r, SA[pl]+A[v]) - SA.begin();
      ansv += pr - vp;
    }
  }
  ans[v] = ansv;
  solve(l,vp);
  solve(vp+1,r);
}
int main(){
  cin >> N;
  P.resize(N); for(int i=0; i<N; i++){ cin>>P[i]; P[i]--; }
  A.resize(N); for(int i=0; i<N; i++) cin>>A[i];
  I.resize(N); for(int i=0; i<N; i++) I[P[i]] = i;
  SA.assign(N+1,0); for(int i=0; i<N; i++) SA[i+1] = SA[i] + P[i] + 1;
  ans.assign(N,0);
  G = RMQ(P);
  solve(0,N);
  for(int i=0; i<N; i++) cout << ans[i] << "\n";
  return 0;
}
struct ios_do_not_sync{
  ios_do_not_sync(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
  }
} ios_do_not_sync_instance;
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0