AC

许诺  •  4天前


#include <iostream>
#include <vector>

using namespace std; long long mergeAndCount(vector& arr, vector& temp, int left, int mid, int right) {

int i = left, j = mid + 1, k = left;
long long inv_count = 0;
while (i <= mid && j <= right) {
    if (arr[i] <= arr[j]) {
        temp[k++] = arr[i++];
    } else {
        temp[k++] = arr[j++];
        inv_count += (mid - i + 1);
    }
}
while (i <= mid) {
    temp[k++] = arr[i++];
}
while (j <= right) {
    temp[k++] = arr[j++];
}
for (i = left; i <= right; ++i) {
    arr[i] = temp[i];
}
return inv_count;

} long long mergeSortAndCount(vector& arr, vector& temp, int left, int right) {

long long inv_count = 0;
if (left < right) {
    int mid = left + (right - left) / 2;
    inv_count += mergeSortAndCount(arr, temp, left, mid);
    inv_count += mergeSortAndCount(arr, temp, mid + 1, right);
    inv_count += mergeAndCount(arr, temp, left, mid, right);
}
return inv_count;

} long long countInversions(vector& arr) {

vector<int> temp(arr.size());
return mergeSortAndCount(arr, temp, 0, arr.size() - 1);

} int main() {

ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vector<int> arr(n);
for (int i = 0; i < n; ++i) {
    cin >> arr[i];
}
cout << countInversions(arr) << '\n';
return 0;

}


评论:

请先登录,才能进行评论