結果

問題 No.1921 Range LthKth Query
ユーザー souta-1326souta-1326
提出日時 2022-05-06 22:37:07
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 517 ms / 4,000 ms
コード長 1,064 bytes
コンパイル時間 2,061 ms
コンパイル使用メモリ 197,008 KB
最終ジャッジ日時 2025-01-29 03:51:11
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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];
int cnt_rp[MAX_N][MAX_N];
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 = -1;
    for(int j=0;j<N;j++){
      rp[j+1] = rp[j]+(A[j]<=i);
      while(k < N && rp[j+1]-rp[k+1] >= K) k++;
      cnt[i][j] = k+1;
    }
    cnt_rp[i][0] = cnt[i][0];
    for(int j=1;j<N;j++) cnt_rp[i][j] = cnt_rp[i][j-1]+cnt[i][j];
  }
  int Q;cin >> Q;
  while(Q--){
    int l,r;cin >> l >> r;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)-cnt[mid];
      if(cnt_rp[mid][r]-(cnt_itr == 0 ? 0:cnt_rp[mid][cnt_itr-1])-(r-cnt_itr+1)*l >= L) ans_rig = mid;
      else ans_lef = mid;
    }
    cout << ans_rig+1 << "\n";
  }
}
0