Question: 
Write a python function to check whether the count of inversion of two types are same or not.


Model's Code:
def solve(arr, n):
    count_0_1 = sum(1 for i in range(n) for j in range(i + 1, n) if arr[i] == 1 and arr[j] == 0)
    count_1_2 = sum(1 for i in range(n) for j in range(i + 1, n) if arr[i] == 1 and arr[j] == 2)
    return count_0_1 == count_1_2



Sample Correct Code:
import sys 
def solve(a,n):   
    mx = -sys.maxsize - 1
    for j in range(1,n):  
        if (mx > a[j]):  
            return False  
        mx = max(mx,a[j - 1])    
    return True


Asserts:
assert solve([1,0,2],3) == True
assert solve([1,2,0],3) == False
assert solve([1,2,1],3) == True


Tests:
Test 1 of 3: PASS
Test 2 of 3: FAIL: AssertionError()

Test 303 of 974: 220/303 (72.60726072607261%): failed