首页 > 代码库 > HDU 4417 划分树+二分
HDU 4417 划分树+二分
题意:有n个数,m个询问(l,r,k),问在区间[l,r] 有多少个数小于等于k。
划分树——查找区间第k大的数。。。。
利用划分树的性质,二分查找在区间[l,r]小于等于k的个数。
如果在区间第 i 大的数tmp>k,则往下找,如果tmp<k,往上找。
#include<cstdio> #include<stdlib.h> #include<string.h> #include<string> #include<map> #include<cmath> #include<iostream> #include <queue> #include <stack> #include<algorithm> #include<set> using namespace std; #define INF 1e8 #define eps 1e-8 #define LL long long #define N 100010 #define mod 1000000007 int a[N],s[N],t[20][N],num[20][N],n,m; void build(int c,int l,int r) { if(l==r) return; int mid=(l+r)/2; int lm=mid-l+1,lp=l,rp=mid+1; for(int i=l;i<=r;i++) lm-=s[i]<s[mid]; for(int i=l;i<=r;i++) { if(i==l) num[c][i]=0; else num[c][i]=num[c][i-1]; if(t[c][i]==s[mid]) { if(lm) { lm--; num[c][i]++; t[c+1][lp++]=t[c][i]; } else t[c+1][rp++]=t[c][i]; } else if(t[c][i]<s[mid]) { num[c][i]++; t[c+1][lp++]=t[c][i]; } else t[c+1][rp++]=t[c][i]; } build(c+1,l,mid); build(c+1,mid+1,r); } int query(int c,int l,int r,int ql,int qr,int k) { if(l==r) return t[c][l]; int s,ss,mid=(l+r)/2; if(ql==l) s=0,ss=num[c][qr]; else s=num[c][ql-1],ss=num[c][qr]-num[c][ql-1]; if(k<=ss) return query(c+1,l,mid,l+s,s+l+ss-1,k); else return query(c+1,mid+1,r,mid+1+ql-l-s,mid+1+qr-l-s-ss,k-ss); } int main() { int T,ca=1; scanf("%d",&T); while(T--) { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) { scanf("%d",&a[i]); s[i]=t[0][i]=a[i]; } sort(s+1,s+n+1); build(0,1,n); printf("Case %d:\n",ca++); while(m--) { int l,r,k; scanf("%d%d%d",&l,&r,&k); l++;r++; if(query(0,1,n,l,r,r-l+1)<=k) printf("%d\n",r-l+1); else if(query(0,1,n,l,r,1)>k) puts("0"); else { int ll=0,rr=r-l+1,mid; while(ll<=rr) { mid=(ll+rr)>>1; int tmp=query(0,1,n,l,r,mid);//如果在区间[l,r]第mid大的数大于k,rr=mid-1 if(tmp>k) rr=mid-1; else ll=mid+1;//否则ll=mid+1 } printf("%d\n",ll-1); } } } return 0; } /* 1 10 10 0 5 2 7 5 4 3 8 7 7 2 8 6 3 5 0 1 3 1 1 9 4 0 1 0 3 5 5 5 5 1 4 6 3 1 5 7 5 7 3 */
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。