主席树做法求助, WA20
查看原帖
主席树做法求助, WA20
118498
xwmwr楼主2020/8/23 15:20
#include<bits/stdc++.h>
using namespace std;
const int N = 200003;
const int M = 10003;

struct sgt{
	int tot, ch[5000003][2], cnt[5000003], root[N];
	//  节点总数  节点的左(0)右(1)儿子, 值域区间的数的个数, 时间轴上从早到晚第 i 颗线段树的根
	void insert(int p, int &q, int l, int r, int x) {
	  //上个版本的节点 p , 新建版本的节点 q, 区间【l, r】, 要在 x 位置插入。
	  q = ++tot;
	  if(l==r) {cnt[q]=1; return;}
	  ch[q][0]=ch[p][0], ch[q][1]=ch[p][1];
	  int mid = (l+r)>>1;
	  if(x<=mid) insert(ch[p][0], ch[q][0], l, mid, x);
	  else insert(ch[p][1], ch[q][1], mid+1, r, x);
	  cnt[q] = cnt[ch[q][0]] + cnt[ch[q][1]];
	}
	
	int ask(int p, int q, int l, int r, int k) {
	  if(l==r) return l;
	  int mid = (l+r)>>1;
	  int lcnt = cnt[ch[q][0]] - cnt[ch[p][0]];
	  if(k<=lcnt) return ask(ch[p][0], ch[q][0], l, mid, k);
	  else return ask(ch[p][1], ch[q][1], mid+1, r, k-lcnt);
	}
	
} T;

int n,m,a[N], b[N], c[N], nn, row[N];

int main() {
	scanf("%d%d", &n,&m);
	for(int i=1;i<=n;++i) scanf("%d", &a[i]), b[i]=a[i];
	sort(b+1,b+1+n);
	b[0] = -2000000000;
	for(int i=1;i<=n;++i) if(b[i]!=b[i-1]) c[++nn] = b[i];
	for(int i=1;i<=n;++i) {
		int to = lower_bound(c+1,c+1+nn,a[i]) - c;
		row[to] = a[i];
		a[i] = to;
    T.insert(T.root[i-1], T.root[i], 1, n, a[i]);
	}
	
	while(m--) {
	  int l,r,k;
	  scanf("%d%d%d", &l,&r,&k);
	  cout << row[T.ask(T.root[l-1], T.root[r], 1, n, k)] << '\n';
	}
	
	return 0;
}
2020/8/23 15:20
加载中...