题目描述
给定一个序列a1,a2,…,an,如果存在i<j并且ai>aj,那么我们称之为逆序对,求逆序对的数目
数据范围:N<=10^5, Ai<=10^5。时间限制为1s。
输入描述
第一行为n,表示序列长度,接下来的n行,第i+1行表示序列中的第i个数。
输出描述
所有逆序对总数.
样例输入
4
3
2
3
2
样例输出
3
分析
求逆序数主要有两种方法,归并排序和树状数组。
树状数组求逆序数讲解链接 https://www.cnblogs.com/xiongmao-cpp/p/5043340.html
方式一:非递归形式的归并排序
#include<iostream>
using namespace std;
#define N 100000 + 10
#define ll long long
ll ans;
void merge(int a[],int b[],int s,int m,int t)
{
int i = s,j = m + 1,k = s;
while(i <= m && j <= t) {
if(a[i] <= a[j]) b[k++] = a[i++];
else {
b[k++] = a[j++];
ans += m - i + 1;//可以求逆序数
}
}
while(i <= m) b[k++] = a[i++];//收尾处理
while(j <= t) b[k++] = a[j++];
}
void mergePass(int a[],int b[],int n,int h)//一次归并排序
{
int i = 0;//下标从0开始
while(i <= n - 2 * h + 1) {//待排序记录至少有两个长度为h的子序列
merge(a,b,i,i + h - 1,i + 2 * h - 1);
i += 2 * h;
}
if(i < n - h + 1)//待归并序列中有一个长度小于h的子序列
merge(a,b,i,i + h - 1,n);
else//只剩下一个子序列
for(int k = i; k <= n; k++)
b[k] = a[k];
}
void mergeSort(int a[],int b[],int n)
{
int h = 1;//子序列的长度
while(h < n) {
mergePass(a,b,n,h);
h *= 2;
mergePass(b,a,n,h);
h *= 2;
}
}
int main()
{
int n,a[N],b[N];
cin >> n;
for(int i = 0; i < n; i++)
cin >> a[i];
mergeSort(a,b,n - 1);
cout << ans << endl;
return 0;
}
方式二:递归形式的归并排序
#include<iostream>
using namespace std;
#define N 100000 + 10
#define ll long long
ll ans;
void merge(int a[],int b[],int s,int m,int t)
{
int i = s,j = m + 1,k = s;
while(i <= m && j <= t) {
if(a[i] <= a[j]) b[k++] = a[i++];
else {
b[k++] = a[j++];
ans += m - i + 1;
}
}
while(i <= m) b[k++] = a[i++];
while(j <= t) b[k++] = a[j++];
}
void mergeSort(int a[],int b[],int s,int t)
{
if(s == t) b[s] = a[s];
else {
int m = (s + t) / 2;
mergeSort(a,b,s,m);
mergeSort(a,b,m + 1,t);
merge(a,b,s,m,t);
for(int i = s; i <= t; i++)
a[i] = b[i];
}
}
int main()
{
int n,a[N],b[N];
cin >> n;
for(int i = 0; i < n; i++)
cin >> a[i];
mergeSort(a,b,0,n - 1);
cout << ans << endl;
return 0;
}
方式三:树状数组
#include<iostream>
using namespace std;
#define N 100000 + 10
#define ll long long
int c[N];
int n;
int lowbit(int x)
{
return x & (-x);
}
void toTree(int i,int a)
{
while(i < N) {
c[i] += a;
i += lowbit(i);
}
}
int sum(int i)
{
int ans = 0;
while(i) {
ans += c[i];
i -= lowbit(i);
}
return ans;
}
int main()
{
cin >> n;
int a;
ll ans = 0;
for(int i = 1; i <= n; i++) {
cin >> a;
toTree(a,1);
ans += i - sum(a);
}
cout << ans << endl;
return 0;
}
方式四:树状数组+离散化
https://blog.csdn.net/qq_43498798/article/details/90767666