# Python Program to Count Inversions of size three in a given array

Given an array arr[] of size n. Three elements arr[i], arr[j] and arr[k] form an inversion of size 3 if a[i] > a[j] >a[k] and i < j < k. Find total number of inversions of size 3.**Example :**

Input: {8, 4, 2, 1} Output: 4 The four inversions are (8,4,2), (8,4,1), (4,2,1) and (8,2,1). Input: {9, 6, 4, 5, 8} Output: 2 The two inversions are {9, 6, 4} and {9, 6, 5}

We have already discussed inversion count of size two by merge sort, Self Balancing BST and BIT.**Simple approach :-** Loop for all possible value of i, j and k and check for the condition a[i] > a[j] > a[k] and i < j < k.

## Python

`# A simple python O(n^3) program` `# to count inversions of size 3` ` ` `# Returns counts of inversions` `# of size threee` `def` `getInvCount(arr):` ` ` `n ` `=` `len` `(arr)` ` ` `invcount ` `=` `0` `#Initialize result ` ` ` `for` `i ` `in` `range` `(` `0` `,n` `-` `1` `):` ` ` `for` `j ` `in` `range` `(i` `+` `1` `, n):` ` ` `if` `arr[i] > arr[j]:` ` ` `for` `k ` `in` `range` `(j` `+` `1` `, n):` ` ` `if` `arr[j] > arr[k]:` ` ` `invcount ` `+` `=` `1` ` ` `return` `invcount` ` ` `# Driver program to test above function` `arr ` `=` `[` `8` `, ` `4` `, ` `2` `, ` `1` `]` `print` `"Inversion Count : %d"` `%` `(getInvCount(arr))` ` ` `# This code is contributed by Nikhil Kumar Singh(nickzuck_007)` |

Output:

Inversion Count : 4

**Time complexity** of this approach is : O(n^3)**Better Approach :**

We can reduce the complexity if we consider every element arr[i] as middle element of inversion, find all the numbers greater than a[i] whose index is less than i, find all the numbers which are smaller than a[i] and index is more than i. We multiply the number of elements greater than a[i] to the number of elements smaller than a[i] and add it to the result.

Below is the implementation of the idea.

## Python3

`# A O(n^2) Python3 program to` `# count inversions of size 3` ` ` `# Returns count of inversions` `# of size 3` `def` `getInvCount(arr, n):` ` ` ` ` `# Initialize result` ` ` `invcount ` `=` `0` ` ` ` ` `for` `i ` `in` `range` `(` `1` `,n` `-` `1` `):` ` ` ` ` `# Count all smaller elements` ` ` `# on right of arr[i]` ` ` `small ` `=` `0` ` ` `for` `j ` `in` `range` `(i` `+` `1` `,n):` ` ` `if` `(arr[i] > arr[j]):` ` ` `small` `+` `=` `1` ` ` ` ` `# Count all greater elements` ` ` `# on left of arr[i]` ` ` `great ` `=` `0` `;` ` ` `for` `j ` `in` `range` `(i` `-` `1` `,` `-` `1` `,` `-` `1` `):` ` ` `if` `(arr[i] < arr[j]):` ` ` `great` `+` `=` `1` ` ` ` ` `# Update inversion count by` ` ` `# adding all inversions that` ` ` `# have arr[i] as middle of` ` ` `# three elements` ` ` `invcount ` `+` `=` `great ` `*` `small` ` ` ` ` `return` `invcount` ` ` `# Driver program to test above function` `arr ` `=` `[` `8` `, ` `4` `, ` `2` `, ` `1` `]` `n ` `=` `len` `(arr)` `print` `(` `"Inversion Count :"` `,getInvCount(arr, n))` ` ` `# This code is Contributed by Smitha Dinesh Semwal` |

**Output :**

Inversion Count : 4

**Time Complexity** of this approach : O(n^2)**Binary Indexed Tree Approach :**

Like inversions of size 2, we can use Binary indexed tree to find inversions of size 3. It is strongly recommended to refer below article first.

Count inversions of size two Using BIT

The idea is similar to above method. We count the number of greater elements and smaller elements for all the elements and then multiply greater[] to smaller[] and add it to the result. **Solution :**

- To find out the number of smaller elements for an index we iterate from n-1 to 0. For every element a[i] we calculate the getSum() function for (a[i]-1) which gives the number of elements till a[i]-1.
- To find out the number of greater elements for an index we iterate from 0 to n-1. For every element a[i] we calculate the sum of numbers till a[i] (sum smaller or equal to a[i]) by getSum() and subtract it from i (as i is the total number of element till that point) so that we can get number of elements greater than a[i].

Please refer complete article on Count Inversions of size three in a given array for more details!