C++ 中的归并排序树


给定一个整数数组、一组区段起始和结束指针以及一个键值,问题陈述是找到给定范围内小于或等于给定键值的所有值。

让我们用例子来理解

输入− arr[] = {7, 8 , 1, 4 , 6 , 8 , 10 }

区段 1:起始 = 2,结束 = 4,k = 2

区段 2:起始 = 1,结束 = 6,k = 3

输出− 给定范围内小于或等于键值的数字个数为 2 6

说明− [8, 1, 4] 表示从 2 到 4 的范围,2 是该范围内第 2 小的数 [7, 8 , 1, 4 , 6 , 8 ] 表示从 1 到 6 的范围,6 是该范围内第 3 小的数

输入− arr[] = {2, 7 , 9, 4 , 6 , 5 , 1 }

区段 1:起始 = 3,结束 = 6,k = 4

区段 2:起始 = 2,结束 = 5,k = 3

输出− 给定范围内小于或等于键值的数字个数为:9 7

说明− [9, 4 , 6 , 5] 表示从 3 到 6 的范围,9 是该范围内第 4 小的数 [7 , 9, 4 , 6 ] 表示从 2 到 4 的范围,7 是该区段范围内第 3 小的数

下面程序中使用的方法如下:

  • 声明一个整数类型数组。计算数组的大小。声明一个向量类型变量,形成整数类型的对。启动 FOR 循环以将数据从数组推送到向量。

  • 对给定的向量进行排序。创建一个具有最大大小的整数类型向量数组。

  • 调用函数 generateTree(1, 0, size - 1, vec, tree) 并将 getSmallestIndex 设置为 queryWrapper(2, 5, 2, size, vec, tree)。

  • 打印 input[getSmallestIndex]。

  • 将 getSmallestIndex 设置为调用函数 queryWrapper(1, 6, 4, size, vec, tree)。

  • 在函数 void generateTree(int treeIndex, int leftIndex, int rightIndex, vector<pair<int, int> > &a, vector<int> tree[]) 内部

    • 检查 IF leftIndex 到 rightIndex 然后设置 tree[treeIndex].push_back(a[leftIndex].second) 并返回

    • 将 midValue 设置为 (leftIndex + rightIndex) / 2 并调用 generateTree(2 * treeIndex, leftIndex, midValue, a, tree), generateTree(2 * treeIndex + 1, midValue + 1, rightIndex, a, tree) 和 merge(tree[2 * treeIndex].begin(), tree[2 * treeIndex].end(), tree[2 * treeIndex + 1].begin(), tree[2 * treeIndex + 1].end(),back_inserter(tree[treeIndex]))

  • 在函数 int calculateKSmallest(int startIndex, int endIndex, int queryStart, int queryEnd, int treeIndex, int key, vector tree[]) 内部

    • 检查 IF startIndex 到 endIndex 然后返回 tree[treeIndex][0]

    • 将 mid 设置为 (startIndex + endIndex) / 2,last_in_query_range 设置为 (upper_bound(tree[2 * treeIndex].begin(),tree[2 * treeIndex].end(), queryEnd) - tree[2 * treeIndex].begin())

    • 将 first_in_query_range 设置为 (lower_bound(tree[2 * treeIndex].begin(),tree[2 * treeIndex].end(), queryStart) - tree[2 * treeIndex].begin()) 并将 M 设置为 last_in_query_range - first_in_query_range

    • 检查 IF M 大于等于 key 然后返回 calculateKSmallest(startIndex, mid, queryStart,queryEnd, 2 * treeIndex, key, tree)

    • 否则,返回 calculateKSmallest(mid + 1, endIndex, queryStart, queryEnd, 2 * treeIndex + 1, key - M, tree)。

  • 在函数 int queryWrapper(int queryStart, int queryEnd, int key, int n, vector<pair<int, int> > &a, vector<int>tree[]) 内部

    • 返回对函数 calculateKSmallest(0, n - 1, queryStart - 1, queryEnd - 1, 1, key, tree) 的调用

示例

#include <bits/stdc++.h>
using namespace std;
const int MAX = 1000;
void generateTree(int treeIndex, int leftIndex, int rightIndex, vector<pair<int, int> > &a, vector<int> tree[]){
   if (leftIndex == rightIndex){
      tree[treeIndex].push_back(a[leftIndex].second);
      return;
   }
   int midValue = (leftIndex + rightIndex) / 2;
   generateTree(2 * treeIndex, leftIndex, midValue, a, tree);
   generateTree(2 * treeIndex + 1, midValue + 1, rightIndex, a, tree);
   merge(tree[2 * treeIndex].begin(), tree[2 * treeIndex].end(), tree[2 * treeIndex + 1].begin(),
   tree[2 * treeIndex + 1].end(), back_inserter(tree[treeIndex]));
}
int calculateKSmallest(int startIndex, int endIndex, int queryStart, int queryEnd, int treeIndex, int key, vector<int> tree[]){
      if (startIndex == endIndex){
         return tree[treeIndex][0];
      }
      int mid = (startIndex + endIndex) / 2;
      int last_in_query_range = (upper_bound(tree[2 * treeIndex].begin(), tree[2 * treeIndex].end(), queryEnd) - tree[2 * treeIndex].begin());
      int first_in_query_range = (lower_bound(tree[2 * treeIndex].begin(), tree[2 * treeIndex].end(),queryStart) - tree[2 * treeIndex].begin());
      int M = last_in_query_range - first_in_query_range;
      if (M >= key){
         return calculateKSmallest(startIndex, mid, queryStart, queryEnd, 2 * treeIndex, key, tree);
      }
      else {
         return calculateKSmallest(mid + 1, endIndex, queryStart,queryEnd, 2 * treeIndex + 1, key - M, tree);
      }
}
int queryWrapper(int queryStart, int queryEnd, int key, int n,
   vector<pair<int, int> > &a, vector<int> tree[]){
      return calculateKSmallest(0, n - 1, queryStart - 1, queryEnd - 1, 1, key, tree);
}
int main(){
   int input[] = { 7, 8 , 1, 4 , 6 , 8 , 10 };
   int size = sizeof(input)/sizeof(input[0]);
   vector<pair<int, int> > vec;
   for (int i = 0; i < size; i++) {
      vec.push_back(make_pair(input[i], i));
   }
   sort(vec.begin(), vec.end());
   vector<int> tree[MAX];
   generateTree(1, 0, size - 1, vec, tree);

   cout<<"Count of number which are smaller than or equal to key value in the given range are:"<<endl;

   int getSmallestIndex = queryWrapper(2, 4, 2, size, vec, tree);
   cout << input[getSmallestIndex] << endl;
   getSmallestIndex = queryWrapper(1, 6, 3, size, vec, tree);
   cout << input[getSmallestIndex] << endl;
   return 0;
}

输出

如果我们运行上述代码,它将生成以下输出

Count of number which are smaller than or equal to key value in the given range are:
4
6

更新于:2021年11月5日

543 次浏览

启动您的 职业生涯

通过完成课程获得认证

开始
广告