数组的子数组的所有mex的总和

Question

我想找到一个数组的子数组的总和，但是我在优化它时遇到了问题。我想比我的代码更快地工作。请帮我。 Mex：集合中不存在的最小自然数。 这是我的代码

from itertools import count, filterfalse 
n = int(input())
x = list(map(int,input().split()))
mex = 0
x = sorted(x)
r = []
for i in range(n):
    for j in range(i,n):
        q = x[i:j+1]
        mex += int(next(filterfalse(set(q).__contains__, count(1))))
print(mex)