-
E - Super Mario
- HDU - 4417
- lower_bound( 1,n,num):从数组的1位置到n位置二分查找第一个大于或等于num的数字.
- 找到返回该数字的地址,不存在则返回end。通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。
- upper_bound( 1,n,num):从数组的1位置到n位置二分查找第一个大于num的数字,
- 找到返回该数字的地址,不存在则返回end。通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。
- 本题需要用upper_bound来查询,返回下标后再-1才是小于等于K的数目。
- 题意:长度为n的数组a,m次查询,每次给定一个区间[l,r]和h。求在区间[l,r]内小于等于h的数的个数。
- 思路:主席树对每一个前缀建一颗线段树,用主席树去维护。线段树中每一个结点代表一个区间。
- 我们维护结点信息sum,sum表示这个a数组在该区间上的值的个数。这样,求[l,r]中小于等于h的数的个数,
- 可以通过查询第r个版本的线段树和第l-1个版本的线段树关于“在区间[l,r]内小于等于h的数的个数”的差值来解决。
- query需要更改,不是找第K大了,是找小于等于K的数目,所以直接用线段树维护的区间跟K比较来确定
- 这段区间里面的sum有哪些是小于等于K值的sum。
-
#include<bits/stdc++.h> using namespace std; #define maxn 123456 int root[maxn],tot,a[maxn],h; int n,m,num[maxn],snum,t,l,r; struct node { int l,r,sum; } tree[maxn*20]; int getlow(int x) { return lower_bound(num+1,num+1+snum,x)-num; } int getup(int x) { return upper_bound(num+1,num+1+snum,x)-num-1; } void build(int &u,int l,int r) { u=++tot; tree[u].sum=0; if(l==r)return ; int mid=(l+r)/2; build(tree[u].l,l,mid); build(tree[u].r,mid+1,r); } void updata(int p,int &u,int l,int r,int k) { u=++tot; tree[u]=tree[p]; tree[u].sum++; if(l==r)return; int mid=(l+r)/2; if(k<=mid)updata(tree[p].l,tree[u].l,l,mid,k); else updata(tree[p].r,tree[u].r,mid+1,r,k); } int query(int l,int r,int t1,int t2,int k) { if(l==r)return tree[t2].sum-tree[t1].sum; int mid=(l+r)/2; if(k<=mid)return query(l,mid,tree[t1].l,tree[t2].l,k); else { int ret=tree[tree[t2].l].sum-tree[tree[t1].l].sum; ret+=query(mid+1,r,tree[t1].r,tree[t2].r,k); return ret; } } int main() { scanf("%d",&t); for(int cas=1; cas<=t; cas++) { tot=0; scanf("%d%d",&n,&m); for(int i=1; i<=n; i++) { scanf("%d",&a[i]); num[i]=a[i]; } sort(num+1,num+1+n); snum=unique(num+1,num+1+n)-1-num; build(root[0],1,snum); for(int i=1; i<=n; i++) updata(root[i-1],root[i],1,snum,getlow(a[i])); printf("Case %d:\n",cas); while(m--) { scanf("%d%d%d",&l,&r,&h); l++,r++; int k=getup(h); if(k==0) printf("0\n"); else printf("%d\n",query(1,snum,root[l-1],root[r],k)); } } return 0; }
E - Super Mario HDU - 4417 -主席树第二弹-区间小于K的总数
最新推荐文章于 2022-05-10 06:03:05 发布