結果

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

ソースコード

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++){
      while(SA[pr] - SA[pl] > A[v]) pl++;
      if(pl <= vp) ansv += vp - pl + 1;
    }
  }
  else{
    int pr = r;
    for(int pl=vp; pl>=l; pl--){
      while(SA[pr] - SA[pl] > A[v]) pr--;
      if(pr >= vp+1) 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;
0