首页 > 代码库 > HDU 4417 Super Mario
HDU 4417 Super Mario
题解:函数式线段树求区间小于等于k的数有几个,离线做法,首先将所有询问和序列一起离散,然后用函数式线段树处理。
#include <map> #include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int N=200005;const int MAXN=3000005;int n,m,tot,data[MAXN],son[MAXN][2],head[MAXN];map<int,int>M;int b[N],a[N],l[N],r[N],k[N];int make(int l,int r){ int t=++tot,mid=(l+r)>>1; if(l==r)return t; son[t][0]=make(l,mid);son[t][1]=make(mid+1,r); return t;}int insert(int x,int pos,int l,int r){ int t=++tot,mid=(l+r)>>1; data[t]=data[pos]+1; if(l==r)return t; if(x<=mid)son[t][1]=son[pos][1],son[t][0]=insert(x,son[pos][0],l,mid); else{son[t][0]=son[pos][0];son[t][1]=insert(x,son[pos][1],mid+1,r);} return t;}int query(int x,int y,int k,int l,int r){ int mid=(l+r)>>1; if(l==r) return 0; if(k>=mid)return(query(son[x][1],son[y][1],k,mid+1,r)+data[son[y][0]]-data[son[x][0]]); else return query(son[x][0],son[y][0],k,l,mid);}int main(){ int T,cnt=1; scanf("%d",&T); while(T--){ int s=0,t=0; M.clear(); tot=0; printf("Case %d:\n",cnt++); scanf("%d%d",&n,&m); for(int i=1;i<=n;i++){scanf("%d",&a[i]);b[++s]=a[i];} for(int i=1;i<=m;i++){scanf("%d%d%d",&l[i],&r[i],&k[i]);l[i]++,r[i]++;b[++s]=k[i];} sort(b+1,b+s+1); for(int i=1;i<=s;i++)if(!M[b[i]])M[b[i]]=++t; for(int i=1;i<=n;i++)a[i]=M[a[i]]; for(int i=1;i<=m;i++)k[i]=M[k[i]]; head[0]=make(1,t); for(int i=1;i<=n;i++)head[i]=insert(a[i],head[i-1],1,t); for(int i=1;i<=m;i++)printf("%d\n",query(head[l[i]-1],head[r[i]],k[i],1,t)); }return 0;}
HDU 4417 Super Mario
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。