結果

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

ソースコード

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;
    }
  }
  for(int i=0;i<N;i++){
    for(int j=0;j<=N;j++) cout << cnt_rp[i][j] << " ";
    cout << endl;
  }
  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_itr == 0 ? 0:cnt_rp[mid][cnt_itr-1])-(r-cnt_itr+1)*(l-1) >= L ? ans_rig:ans_lef) = mid;
    }
    cout << ans_rig+1 << "\n";
  }
}
0