Super Mario
题目链接: Super Mario HDU - 4417
题意
给n个数的数组,m次询问,每次询问区间【L,R】中小于等于k的数有几个
思路
就是裸的主席树吧,因为要求小于等于k的值,所以,我们先将其离散化,然后从第一个数据开始建不同版本的主席树,每个线段树了里面所储存的就是总共的子树数目,然后我们就可以通过比较 r 版本和 l-1版本中ID小于等于k的ID的值有多少
注意一些极端情况,比如当k比区间里所有的值都小时或k比区间所有的值都大时的特例。
代码
#include <iomanip>
#include <cstring>
#include <cstdlib>
#include <cctype>
#include <cstdio>
#include <string>
#include <stack>
#include <cmath>
#include <ctime>
#include <list>
#include <set>
#include <map>
#include <queue>
#include <vector>
#include <sstream>
#include <memory>
#include <iostream>
#include <algorithm>
using namespace std;
#define rep(i,j,k) for(int i = (int)j;i <= (int)k;i ++)
#define per(i,j,k) for(int i = (int)j;i >= (int)k;i --)
#define debug(x) cerr<<#x<<" = "<<(x)<<endl
#define mmm(a,b) memset(a,b,sizeof(a))
#define pb push_back
typedef double db;
typedef long long ll;
const int MAXN = (int)1e5+7;
const int INF = (int)0x3f3f3f3f;
int N,M,tot,root[MAXN],A[MAXN],l,r,k;
struct Node {int lson,rson,sum; }T[MAXN*40];
vector<int> vp;
int getID(int x) {return upper_bound(vp.begin(),vp.end(),x)-vp.begin(); };
void Update(int l,int r,int last,int &cur,int pos) {
cur = ++tot;
T[cur] = T[last];T[cur].sum ++;
if (l == r)return ;
int mid = l+r>>1;
if (mid >= pos) Update(l,mid,T[last].lson,T[cur].lson,pos);
else Update(mid+1,r,T[last].rson,T[cur].rson,pos);
}
int Query(int l,int r,int last,int cur,int k){
if (l == r) {
debug(T[cur].sum - T[last].sum);
return T[cur].sum - T[last].sum;
}
int mid = l+r>>1;
int sum = 0;
if (mid >= k) sum += Query(l,mid,T[last].lson,T[cur].lson,k);
else sum += T[T[cur].lson].sum - T[T[last].lson].sum + Query(mid+1,r,T[last].rson,T[cur].rson,k);
return sum;
}
void debugt(int o,int l,int r){
if(l == r) {printf("%d:%d ",l,T[o].sum);return;}
int mid = (l+r) >> 1;
debugt(T[o].lson,l,mid);
debugt(T[o].rson,mid+1,r);
}
void init() {
vp.clear();
T[0].lson = T[0].rson = T[0].sum = 0;
tot = 0;
}
int main()
{
int T;
scanf("%d",&T);
rep(ca,1,T) {
init();
printf("Case %d:\n",ca);
scanf("%d %d",&N,&M);
rep(i,1,N) scanf("%d",&A[i]),vp.pb(A[i]);
sort(vp.begin(),vp.end()),vp.erase(unique(vp.begin(),vp.end()),vp.end());
rep(i,1,N) Update(1,N,root[i-1],root[i],getID(A[i]));
rep(i,1,M) {
scanf("%d %d %d",&l,&r,&k); l++,r++;
int id = getID(k);
if (id == 0) printf("0\n");
printf("%d\n",Query(1,N,root[l-1],root[r],id));
}
}
}
/*
3
3 3
3 3 3
0 1 0
0 1 0
0 1 0
*/