给定一个数组’a []’和查询数量q。每个查询都可以用l,r,x表示。您的任务是在l到r表示的子数组中打印小于或等于x的元素数。
例子:
Input : arr[] = {2, 3, 4, 5}
q = 2
0 3 5
0 2 2
Output : 4
1
Number of elements less than or equal to
5 in arr[0..3] is 4 (all elements)
Number of elements less than or equal to
2 in arr[0..2] is 1 (only 2)
天真的方法每个查询的天真的方法遍历子数组并计算给定范围内的元素数。
高效的方法思想是使用二叉索引树。
请注意,在以下步骤中,x是必须根据其查找元素的数字,并且子数组由l,r表示。
步骤1:按升序对数组进行排序。
步骤2:按照x升序对查询进行排序,将位数组初始化为0。
步骤3:从第一个查询开始,遍历数组,直到数组中的值小于等于x。对于每个这样的元素,将BIT更新为等于1的值
步骤4:查询范围为l到r的BIT数组
// C++ program to answer queries to count number
// of elements smaller tban or equal to x.
#include
using namespace std;
// structure to hold queries
struct Query
{
int l, r, x, idx;
};
// structure to hold array
struct ArrayElement
{
int val, idx;
};
// bool function to sort queries according to k
bool cmp1(Query q1, Query q2)
{
return q1.x < q2.x;
}
// bool function to sort array according to its value
bool cmp2(ArrayElement x, ArrayElement y)
{
return x.val < y.val;
}
// updating the bit array
void update(int bit[], int idx, int val, int n)
{
for (; idx<=n; idx +=idx&-idx)
bit[idx] += val;
}
// querying the bit array
int query(int bit[], int idx, int n)
{
int sum = 0;
for (; idx > 0; idx -= idx&-idx)
sum += bit[idx];
return sum;
}
void answerQueries(int n, Query queries[], int q,
ArrayElement arr[])
{
// initialising bit array
int bit[n+1];
memset(bit, 0, sizeof(bit));
// sorting the array
sort(arr, arr+n, cmp2);
// sorting queries
sort(queries, queries+q, cmp1);
// current index of array
int curr = 0;
// array to hold answer of each Query
int ans[q];
// looping through each Query
for (int i=0; i
输出:
1
4