結果

問題 No.1921 Range LthKth Query
ユーザー souta-1326souta-1326
提出日時 2022-05-06 22:36:28
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 666 ms / 4,000 ms
コード長 970 bytes
コンパイル時間 2,966 ms
コンパイル使用メモリ 197,852 KB
最終ジャッジ日時 2025-01-29 03:50:39
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
using namespace std;
constexpr int MAX_N = 5000;
int A[MAX_N];
int cnt[MAX_N][MAX_N+1];
int cnt_rp[MAX_N][MAX_N+1];
int rp[MAX_N+1];
int main(){
  cin.tie(0);ios::sync_with_stdio(false);
  int N,K,L;cin >> N >> K >> L;
  for(int i=0;i<N;i++){cin >> A[i];A[i]--;}
  for(int i=0;i<N;i++){
    int k = 0;
    for(int j=0;j<N;j++){
      rp[j+1] = rp[j]+(A[j]<=i);
      while(k < N && rp[j+1]-rp[k] >= K) k++;
      cnt[i][j+1] = k;
      cnt_rp[i][j+1] = cnt_rp[i][j]+k;
    }
  }
  int Q;cin >> Q;
  while(Q--){
    int l,r;cin >> l >> r;
    int ans_lef = -1,ans_rig = N-1;
    while(ans_rig - ans_lef > 1){
      int mid = (ans_lef+ans_rig)/2;
      int cnt_itr = upper_bound(cnt[mid]+l,cnt[mid]+r+1,l-1)-cnt[mid];
      (cnt_rp[mid][r]-cnt_rp[mid][cnt_itr-1]-(r-cnt_itr+1)*(l-1) >= L ? ans_rig:ans_lef) = mid;
    }
    cout << ans_rig+1 << "\n";
  }
}
0