[Python]A stupid question / 据说是某前端面试题

对于给定list：

1、若最高位个数大于2 ，返回最高位

2、若最高位个数不大于2，返回最高位和次高位

example：

'''
[5,4,3,2,1] -> [5,4]
[5,5,4,4,4] -> [5,5]
[5,5,5,4,4] -> [5,5,5]
[5,5,5,5,5] -> [5,5,5,5,5]
[5,4,4,3,2] -> [5,4,4]
[5,4,4,4,3] -> [5,4,4,4]
[5,4,4,4,4] -> [5,4,4,4,4]
'''

'''
[5,4,3,2,1] -> [5,4]
[5,5,4,4,4] -> [5,5]
[5,5,5,4,4] -> [5,5,5]
[5,5,5,5,5] -> [5,5,5,5,5]
[5,4,4,3,2] -> [5,4,4]
[5,4,4,4,3] -> [5,4,4,4]
[5,4,4,4,4] -> [5,4,4,4,4]
'''
class Solution:
    def isMatch(self, s: list) -> list:
        lens = len(s)
        # 0
        if lens == 0:
            return []
        #1,2
        if lens <= 2:
            return s
        #normal
        s.sort()
        max_1 = s[-1]
        countmax_1 = s.count(s[-1])
        #[5,5,5,5,5]
        if countmax_1 < lens:
            max_2 = s[-(1 + countmax_1)]
            countmax_2 = s.count(s[-(1 + countmax_1)])
        #return
        if countmax_1 >=2:
            return s[ (lens - countmax_1):]
        else:
            return s [(lens - countmax_1 -countmax_2):][::-1]
def main():
    import sys
    #import io

    s = [
        [5,4,3,2,1],
        [5,5,4,4,4],
        [5,5,5,4,4],
        [5,5,5,5,5],
        [5,4,4,3,2],
        [5,4,4,4,3],
        [5,4,4,4,4]
    ]

    for i in s:
        ret = Solution().isMatch(i)
        print(ret)

if __name__ == '__main__':
    main()