HDU-4417 - Super Mario （主席树）

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

*/