Counting Inversions using Set in C++ STL - GeeksforGeeks

28 Jun, 2021
Inversion Count for an array indicates – how far (or close) the array is from being sorted. If array is already sorted then inversion count is 0. If array is sorted in reverse order that inversion count is the maximum.
    Two elements a[i] and a[j] form an inversion if 
     a[i] > a[j] and i < j. For simplicity, we may 
     assume that all elements are unique.

     Example:
     Input:  arr[] = {8, 4, 2, 1}
     Output: 6
     Given array has six inversions (8,4), (4,2),
     (8,2), (8,1), (4,1), (2,1).     

We have already discussed below approaches.1) Naive and Merge Sort based approaches.2) AVL Tree based approach.
In this post an easy implementation of approach 2 using Set in C++ STL is discussed.
1) Create an empty Set in C++ STL (Note that a Set in C++ STL is 
   implemented using Self-Balancing Binary Search Tree). And insert
   first element of array into the set.

2) Initialize inversion count as 0.

3) Iterate from 1 to n-1 and do following for every element in arr[i]
     a) Insert arr[i] into the set.
     b) Find the first element greater than arr[i] in set
        using upper_bound() defined Set STL.
     c) Find distance of above found element from last element in set
        and add this distance to inversion count.

4) Return inversion count.

// A STL Set based approach for inversion count #include<bits/stdc++.h>using namespace std;  // Returns inversion count in arr[0..n-1]int getInvCount(int arr[],int n){    // Create an empty set and insert first element in it    multiset<int> set1;    set1.insert(arr[0]);      int invcount = 0; // Initialize result      multiset<int>::iterator itset1; // Iterator for the set      // Traverse all elements starting from second    for (int i=1; i<n; i++)    {        // Insert arr[i] in set (Note that set maintains        // sorted order)        set1.insert(arr[i]);          // Set the iterator to first greater element than arr[i]        // in set (Note that set stores arr[0],.., arr[i-1]        itset1 = set1.upper_bound(arr[i]);          // Get distance of first greater element from end        // and this distance is count of greater elements        // on left side of arr[i]        invcount += distance(itset1, set1.end());    }      return invcount;}  // Driver program to test aboveint main(){    int arr[] = {8, 4, 2, 1};    int n = sizeof(arr)/sizeof(int);    cout << "Number of inversions count are : "         << getInvCount(arr,n);    return 0;}
Output:
Number of inversions count are : 6
Note that the worst case time complexity of above implementation is O(n2) as distance function in STL takes O(n) time worst case, but this implementation is much simpler than other implementations and would take much less time than Naive method on average.
This article is contributed by Abhiraj Smit. Please write comments if you find anything incorrect, or you want to share more information about the topic discussed above
ManasChhabra2
RajSheth
cpp-set
inversion
STL
C++
STL
CPP
Writing code in comment?
Please use ide.geeksforgeeks.org,
generate link and share the link here.
Inheritance in C++
Map in C++ Standard Template Library (STL)
C++ Classes and Objects
Operator Overloading in C++
Bitwise Operators in C/C++
Socket Programming in C/C++
Constructors in C++
Virtual Function in C++
Multidimensional Arrays in C / C++
Templates in C++ with Examples