Create 1081_TenderRun.cpp

From https://vijos.org/p/1081/solution

VIJOS/1081_TenderRun.cpp

@@ -0,0 +1,77 @@
+#include <algorithm>
+#include  <iostream>
+#include   <cstring>
+#include    <cstdio>
+
+using namespace std;
+const int maxn=100010;
+
+struct Q{
+    int l,r,a,b,sum;
+}tr[maxn*20];
+
+int fir[maxn],cnt;
+int n,k,Sot[maxn],h[maxn],pos;
+
+void Build(int a,int b,int &rt)
+{
+    int md=(a+b)>>1;
+    rt=++cnt;
+    tr[rt].a=a;
+    tr[rt].b=b;
+    if(a==b)return;
+    Build(a,md,tr[rt].l);
+    Build(md+1,b,tr[rt].r);
+}
+
+void Add(int pre,int &rt,int a,int b)
+{
+    rt=++cnt;
+    tr[rt].l=tr[pre].l;
+    tr[rt].r=tr[pre].r;
+    tr[rt].a=tr[pre].a;
+    tr[rt].b=tr[pre].b;
+
+    tr[rt].sum=tr[pre].sum+1;
+
+    if(a<b){
+        int md=(a+b)>>1;
+
+        if(pos<=md)Add(tr[pre].l,tr[rt].l,a,md);
+        else Add(tr[pre].r,tr[rt].r,md+1,b);
+    }
+}
+#define Cmp(x) tr[tr[x].l].sum
+int Query(int pre,int rt,int a,int b,int K)
+{
+    if(a==b) return Sot[a]; 
+
+    int md=(a+b)>>1;
+
+    int D=Cmp(rt)-Cmp(pre);
+    if(D<K){
+        return Query(tr[pre].r,tr[rt].r,md+1,b,K-D);
+    }
+    else{
+        return Query(tr[pre].l,tr[rt].l,a,md,K);
+    }
+}
+
+int main()
+{
+    scanf("%d%d",&n,&k);
+    for(int i=1;i<=n;Sot[i]=h[i],i++)scanf("%d",&h[i]);
+    sort(Sot+1,Sot+n+1);
+    Build(1,n,fir[0]);
+    for(int i=1;i<=n;i++){
+        pos=lower_bound(Sot+1,Sot+n+1,h[i])-Sot;
+        Add(fir[i-1],fir[i],1,n);
+    }
+    for(int i=1;i<=k;i++)
+    {
+        int l,r,ks;
+        scanf("%d%d%d",&l,&r,&ks);
+        printf("%d\n",Query(fir[l-1],fir[r],1,n,ks));
+    }
+    return 0;
+}