牛客小白月赛9 - E - 换个角度思考（主席树）

本文主要是介绍牛客小白月赛9 - E - 换个角度思考（主席树），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

题目链接：https://ac.nowcoder.com/acm/contest/275/E

思路：求区间内小于  的个数（分块和树状数组也可写），主席树保存的是值域，当我们查询到时，说明此时左子树的值全部满足小于  直接加即可，当查到叶子节点直接返回两颗线段树的差值即可。

#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll ;
const int maxn = 2e5+7;
const int mod = 1e9+7;
int n, m, cnt, root[maxn], a[maxn], vis[maxn], l, r, k;
vector<int> v;
struct node{int l, r, sum;} T[maxn*25];
int getid(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}void update(int l,int r,int &now, int pre, int pos)
{T[++cnt] = T[pre], T[cnt].sum++, now = cnt;if(l == r) return;int mid = (l + r) >> 1;if(mid>=pos) update(l, mid, T[now].l, T[pre].l, pos);else update(mid+1, r, T[now].r, T[pre].r, pos);
}
int query(int l, int r, int L, int R, int k)
{if(l == r) return T[R].sum - T[L].sum;int mid = (l + r) >> 1;int res = 0;if(mid >= k) res =  query(l, mid, T[L].l, T[R].l, k);else res =  query(mid+1, r, T[L].r, T[R].r, k) + T[T[R].l].sum - T[T[L].l].sum;return res;
}
int main()
{scanf("%d%d",&n,&m);for(int i = 1; i <= n; i++)scanf("%d",&a[i]),v.push_back(a[i]), vis[a[i]] = 1;sort(v.begin(),v.end()); int mi = v[0];v.erase(unique(v.begin(),v.end()),v.end());for(int i = 1; i <= n; i++) update(1,n,root[i],root[i-1],getid(a[i]));for(int i = 1; i <= m; i++){scanf("%d%d%d", &l, &r, &k);int pos = getid(k);if(!vis[k]) pos--;if(k >= mi) printf("%d\n",query(1,n,root[l-1],root[r],pos));else puts("0");}return 0;
}

 

这篇关于牛客小白月赛9 - E - 换个角度思考（主席树）的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---