题目请戳这里
题目大意:给许多操作,4类:
1,insert x,插入一个数x,x范围2^31内(题目描述有误)
2,query_1 s t k,查询范围[s,t]内第k小的数,s<=t<=len(当前数字长度)
3,query_2 x 查询x的rank,从小到大。x保证出现过
4,query_3 x,查询[1,len]第x小的数。
输出3个数,为3个询问累加和。
题目分析:由于没有删除操作,所以加入的数字序列是不变的,所以离线处理。
q1和q3其实是一个操作,查询区间第k大值问题,划分树搞。
q2的话,其实也可以通过二分转化为划分树操作。如果愿意,写棵线段树单独维护一下也是可以的,不过要离散化。为了偷懒,选择二分+划分树啦
关于划分树,其实也是基于线段树的一个数据结构,模拟归并排序的过程,不断分治。详细讲解戳这里
详情请见代码:
#include <iostream> #include<cstdio> #include<cstring> #include<algorithm> using namespace std; const int N = 100005; const int M = 110005; const int DEEP = 20; typedef __int64 ll; ll ans[3]; struct node { int optype; int a,b,c; }op[M]; int n,m; char str[20]; int num[DEEP][N],lefta[DEEP][N],lcm[N]; void build(int s,int e,int nm,int dp) { if(s == e) return; int mid = (s + e)>>1; int lsame = mid - s + 1; int i,same = 0; for(i = s;i <= e;i ++) if(num[dp][i] < lcm[mid]) lsame --; int lpos = s,rpos = mid + 1; for(i = s;i <= e;i ++) { if(i == s) lefta[dp][i] = 0; else lefta[dp][i] = lefta[dp][i - 1]; if(num[dp][i] < lcm[mid]) { lefta[dp][i] ++; num[dp + 1][lpos ++] = num[dp][i]; } else { if(num[dp][i] > lcm[mid]) num[dp + 1][rpos ++] = num[dp][i]; else { if(same < lsame) { same ++; lefta[dp][i] ++; num[dp + 1][lpos ++] = num[dp][i]; } else num[dp + 1][rpos ++] = num[dp][i]; } } } build(s,mid,nm<<1,dp + 1); build(mid + 1,e,nm<<1|1,dp + 1); } int query(int s,int e,int nm,int dp,int l,int r,int k) { if(s == e) return num[dp][s]; int mid = (s + e)>>1; int sa,sb; if(s == l) sb = 0; else sb = lefta[dp][l - 1]; sa = lefta[dp][r] - sb; if(sa >= k) { int newl = s + sb; int newr = s + sb + sa - 1; return query(s,mid,nm<<1,dp + 1,newl,newr,k); } else { int bb = l - s - sb; int b = r - l + 1 - sa; int newl = mid + bb + 1; int newr = mid + bb + b; return query(mid + 1,e,nm<<1|1,dp + 1,newl,newr,k - sa); } } ll getans1(int a,int b,int c) { int mid,la,ra; la = a;ra = b; while(la <= ra) { mid = (la + ra)>>1; int tmp = query(1,n,1,0,a,b,mid); if(tmp == c) return mid - a + 1; else if(tmp > c) ra = mid - 1; else la = mid + 1; } } int main() { int i; int t,q,a,b,c,cas = 0; while(~scanf("%d",&q)) { n = m = 1; while(q --) { scanf("%s",str); if(str[0] == 'I') { scanf("%d",&lcm[n]); num[0][n] = lcm[n ++]; } else { if(str[strlen(str) - 1] == '1') { scanf("%d%d%d",&a,&b,&c); op[m].optype = 0; op[m].a = a;op[m].b = b;op[m ++].c = c; } else if(str[strlen(str) - 1] == '2') { scanf("%d",&a); op[m].optype = 1; op[m].a = 1;op[m].b = n - 1; op[m ++].c = a; } else { scanf("%d",&a); op[m].optype = 2; op[m].a = 1;op[m].b = n - 1; op[m ++].c = a; } } } sort(lcm + 1,lcm + n); n --; build(1,n,1,0); ans[1] = ans[0] = ans[2] = 0; for(i = 1;i < m;i ++) { if(op[i].optype == 1) { int tmp = getans1(op[i].a,op[i].b,op[i].c); ans[op[i].optype] += tmp; } else { int tmp = query(1,n,1,0,op[i].a,op[i].b,op[i].c); ans[op[i].optype] += tmp; } } printf("Case %d:\n%I64d\n%I64d\n%I64d\n",++cas,ans[0],ans[1],ans[2]); } return 0; } //265MS 15908K