如何使用段树计算数组中的反转次数
How to count number of inversions in an array using segment trees
我知道这个问题可以使用修改后的合并排序来解决,并且我已经编写了相同的代码。现在我想使用段树来解决这个问题。基本上,如果我们从右到左遍历数组,那么我们必须计算"有多少值大于当前值"。分段树如何实现这一点?
我们必须在段树节点上存储什么类型的信息?
如果可能,请提供代码。
让我用一个例子逐步解释:
arr : 4 3 7 1
position : 0 1 2 3
首先,按降序对数组进行排序,作为 {值, 索引} 对。
arr : 7 4 3 1
index : 2 0 1 3
position : 0 1 2 3
从左到右迭代,对于每个元素arr[i]
-
查询每个元素的index
(查询范围[0, arr[i].index]
以获取左侧更大的数字计数),并将查询结果放在输出数组的相应index
上。
每次查询后,递增覆盖该index
的相应段树节点。
这样,我们确保从0
到index - 1
只获得更大的数字计数,因为到目前为止插入的值仅大于arr[i]
。
下面C++实现将更有意义。
class SegmentTree {
vector<int> segmentNode;
public:
void init(int n) {
int N = /* 2 * pow(2, ceil(log((double) n / log(2.0)))) - 1 */ 4 * n;
segmentNode.resize(N, 0);
}
void insert(int node, int left, int right, const int indx) {
if(indx < left or indx > right) {
return;
}
if(left == right and indx == left) {
segmentNode[node]++;
return;
}
int leftNode = node << 1;
int rightNode = leftNode | 1;
int mid = left + (right - left) / 2;
insert(leftNode, left, mid, indx);
insert(rightNode, mid + 1, right, indx);
segmentNode[node] = segmentNode[leftNode] + segmentNode[rightNode];
}
int query(int node, int left, int right, const int L, const int R) {
if(left > R or right < L) {
return 0;
}
if(left >= L and right <= R) {
return segmentNode[node];
}
int leftNode = node << 1;
int rightNode = leftNode | 1;
int mid = left + (right - left) / 2;
return query(leftNode, left, mid, L, R) + query(rightNode, mid + 1, right, L, R);
}
};
vector<int> countGreater(vector<int>& nums) {
vector<int> result;
if(nums.empty()) {
return result;
}
int n = (int)nums.size();
vector<pair<int, int> > data(n);
for(int i = 0; i < n; ++i) {
data[i] = pair<int, int>(nums[i], i);
}
sort(data.begin(), data.end(), greater<pair<int, int> >());
result.resize(n);
SegmentTree segmentTree;
segmentTree.init(n);
for(int i = 0; i < n; ++i) {
result[data[i].second] = segmentTree.query(1, 0, n - 1, 0, data[i].second);
segmentTree.insert(1, 0, n - 1, data[i].second);
}
return result;
}
// Input : 4 3 7 1
// output: 0 1 0 3
这很容易,但不像其他典型的段树问题那样"明显"。使用笔和纸进行任意输入的模拟会有所帮助。
BST、芬威克树和合并排序还有其他O(nlogn)
方法。
它解决得很简单。我们用运算总和构造一个大小为 n
的空段树。现在从左到右浏览排列元素。段树的叶子中的一人意味着已经访问过这样的元素。当移动到 p[i]
的 i-th
元素时,我们将请求计算段树中的[p[i],n]
和:它只会计算左侧大于 p[i]
的元素数。最后,把一个放在p[i]
的位置.总时间为 O(nlogn)
.
相关文章:
- 当字段可以为null时,如何使用C++接口在Avro中写入数据
- 在 c++ 中解决段树以外的范围查询的有效方法是什么?
- 程序以使用C++找到树的最深叶子的总和
- C++:段树中的运行时错误
- RMQ段树中的错误
- 使用xsdcxx树解析器的XSD到C++类.方法从文件中打开xml
- 使用AVL树的动态哈希表的复杂性
- 使用 Avl 树的数据结构项目
- Webkit GTK:使用DOM树遍历器
- 正在段树中查询
- 用于C++中的段树的STL
- 段树2D,矩形之和
- 段树的迭代代码
- 如何使用段树计算数组中的反转次数
- 使用AVL树的缺点是什么?
- 段树实现中的问题
- 如何使用后缀树找到所有匹配子字符串的索引
- 寻找最大使用段树给出错误的结果
- 内存分配错误-段树
- 如何在matlab中使用kd树文件交换和mex