【经典题目】逆序对的数量 离散化技巧

P1908 逆序对

逆序对

题目描述

逆序对就是序列中 \(a_i>a_j\) 且 \(i<j\) 的有序对。

以下解法时间复杂度均为 \(O(log n)\)

两种算法比较:

  • 归并分治的做法属于离线,并且修改了原始数组;他不受数据值域的限制
  • 树状数组的做法属于在线,每次都可以实时的拿到答案,并且不需要修改原始数组,但是受数据值域的限制

归并分治解法

C++ 复制代码
#include <bits/stdc++.h>
using namespace std;

int help[500005];

long long merge(vector<int>& arr, int l, int m, int r) {
	long long ans = 0;
	for (int i = m, j = r; i >= l; i--) {
		while (j > m && arr[i] <= arr[j]) {
			j--;
		}
		ans += j - m;
	}
	int i = l, a = l, b = m+1;
	while (a <= m && b <= r) {
		help[i++] = arr[a] < arr[b] ? arr[a++] : arr[b++];
	}
	while (a <= m) {
		help[i++] = arr[a++];
	}
	while (b <= r) {
		help[i++] = arr[b++];
	}
	for (int i = l; i <= r; i++) {
		arr[i] = help[i];
	}
}

long long f(vector<int>& arr, int l, int r) {
	if (l == r) {
		return 0;
	}
	int mid = (r - l) / 2 + l;
	return f(arr, l, m) + f(arr, m+1, r) + merge(arr, l, m, r);
}

int main() {
	
	return 0;
}

树状数组解法

首先,逆序对就是每遍历到一个数,看看在他后面的数中有几个比他大,因此可以从右往左遍历,统计 \([1, x-1]\) 词频的词频,每次遍历到新的数字需要单点修改,与范围查询前缀和,因此选择树状数组,注意到数据范围过大,如果不做处理将会达到 \(10^9\) 的量级。可以确定的是,我们不需要具体数字大小,只需要相对大小,因此可以离散化处理。

具体处理如下,将原始数据从小到大排序,它的下标就为新值,再将该值刷回数组中,这样将值域从 \(10^9\) 减小到了和数组大小一致的量级,且保证了数据的相对大小

处理原始数据,保证新值不小于 1

C++ 复制代码
void transfer() {
    sort(arr+1, arr+n+1);
    len = 1;
    for (int i = 2; i <= n; i++) {
        if (arr[len] != arr[i]) arr[++len] = arr[i];
    }
    for (int i = 1; i <= n; i++) {
        sorted[i] = lower_bound(arr+1, arr+1+len, sorted[i]) - arr;
    }
}

总代码

c++ 复制代码
#include <bits/stdc++.h>
using namespace std;

const int MAXN = 5e5+7;
int n;
int arr[MAXN];
int sorted[MAXN];
int tree[MAXN];
int len;
void transfer() {
    sort(arr+1, arr+n+1);
    len = 1;
    for (int i = 2; i <= n; i++) {
        if (arr[len] != arr[i]) arr[++len] = arr[i];
    }
    for (int i = 1; i <= n; i++) {
        sorted[i] = lower_bound(arr+1, arr+1+len, sorted[i]) - arr;
    }
}

void add(int i, int v) {
    while (i <= n) {
        tree[i] += v;
        i += i & (-i);
    }
}

long long sum(int i) {
    long long ans = 0;
    while (i > 0) {
        ans += tree[i];
        i -= i & (-i);
    }
    return ans;
}

int main() {
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> arr[i];
        sorted[i] = arr[i];
    }
    transfer();
    long long ans = 0;
    for (int i = n; i > 0; i--) {
        int x = sorted[i];
        ans += sum(x-1);
        add(x, 1);
    }
    cout << ans << "\n";
    
    return 0;
}
相关推荐
LGL6030A4 小时前
算法题实战积累(3)——方块转换(C语言)
c语言·算法
一条星星鱼4 小时前
深度学习是如何收敛的?梯度下降算法原理详解
人工智能·深度学习·算法
长路归期无望7 小时前
C语言小白实现多功能计算器的艰难历程
c语言·开发语言·数据结构·笔记·学习·算法
MobotStone8 小时前
AI训练的悖论:为什么越追求准确率越会产生幻觉?
算法
怀旧,9 小时前
【C++】26. 智能指针
开发语言·c++·算法
Haooog9 小时前
654.最大二叉树(二叉树算法)
java·数据结构·算法·leetcode·二叉树
Swift社区9 小时前
LeetCode 392 判断子序列
算法·leetcode·职场和发展
芒果量化9 小时前
ML4T - 第7章第8节 利用LR预测股票价格走势Predicting stock price moves with Logistic Regression
算法·机器学习·线性回归
东方芷兰10 小时前
JavaWeb 课堂笔记 —— 20 SpringBootWeb案例 配置文件
java·开发语言·笔记·算法·log4j·intellij-idea·lua
Diligence81510 小时前
最优化方法
算法