可见的山峰对数量｜Python解法

题目描述：

        一个不含有负数的数组可以代表一圈环形山，每个位置的值代表山的高度。比如：{3,1,2,4,5}、{4,5,3,1,2}或{1,2,4,5,3}都代表同样结构的环形山。3->1>2>4>5方向叫作next方向（逆时针），3->5->4->2->1->3方向叫作last方向（顺时针），如下图1-1所示：

图1-1

         山峰A和山峰B能够相互看见的条件为：

        1.如果A和B是同一座山，认为不能相互看见。

        2.如果A和B是不同的山，并且在环中相邻，认为可以相互看见。比如图1-8中，相邻的山峰对有(1,2)(2,4)(4,5)(1,3)。

        3.如果A和B是不同的山，并且在环中不相邻，假设两座山高度的最小值为min。如果A通过next方向到B的途中没有高度比min大的山峰，或者A通过last方向到B的途中没有高度比min大的山峰，认为A和B可以互相看见。比如上图1-1中，高度为3的山和高度为4的山，两座山的高度最小值为3.3从last方向走向4，中途会遇见5，所以last方向走不通；3从next方向走向4，中途会遇见1和2，但是都不大于两座山的最小值3，所以next方向可以走通。有一个能走通就认为可以互相看见。再如，高度为2的山和高度为5的山，两个方向上都走不通，所以不能相互看见。上图1-1中所有在环中不相邻，并且能看见的山峰对有(2,3)(3,4)。

        给定一个不含有负数且没有重复值的数组arr，请返回有多少对山峰能够互相看见。

        进阶问题：给定一个不含有负数但可能含有重复值的数组arr，返回多少对山峰能够相互看见。

原始问题：

如果环中含有i个节点，那总共可以返回可见山峰对数量为：2*i-3对。

进阶问题：

如果环中含有i个节点，以(X,K)形式进行记录（X代表当前节点，K代表次数）。当某个(X,K）从stack中弹出了，产生可见山峰对的数量为；

1.如果K==1，产生2对。

2.如果K>1，产生2*K+C(2,K)对。

代码：

class gvn:
    def __init__(self,value):
        self.value = value
        self.times = 1
        
    def getVisiblenum(arr):
        size = len(arr)
        if arr == None or size < 2:
            return 0

        maxIndex = 0
        stack = []
        res = 0
        # 找出圆环最大值
        for i in range(0,size):
            maxIndex = i if arr[maxIndex] < arr[i] else maxIndex
            
        # 将最大值push进去栈中
        stack.append(gvn(arr[maxIndex]))

        # 从最大值位置的下一个位置开始沿next方向遍历
        index = gvn.nextIndex(maxIndex,size)
        while index != maxIndex:
            if stack[-1].value < arr[index]:
                k = stack.pop().times
                res += gvn.getInternalSum(k)+2*k
            if stack[-1].value == arr[index]:
                stack.pop().times+=1
            else:
                stack.append(gvn(arr[maxIndex]))
            index = gvn.nextIndex(index,size)

        # 清算阶段
        # 清算第一阶段
        while len(stack) > 2:
            times = stack.pop().times
            res += gvn.getInternalSum(times) + 2 * times
        # 清算第二阶段
        if len(stack) == 2:
            times = stack.pop().times
            res += gvn.getInternalSum(times) + stack[-1].times == times if stack[-1].times == 1 else 2 * times
        
        # 清算第三阶段
        res += gvn.getInternalSum(stack.pop().times)
        return res

    def getInternalSum(k):
        return 0 if k == 1 else (k*(k-1)/2)

    def nextIndex(i,size):
        return i+1 if i< (size-1) else 0
                    

if __name__ == '__main__':
    print(gvn.getVisiblenum(arr=[1,2,3,4,5]))

代码中存在较多待优化的地方，望轻喷～

有问题望各位大佬指出，谢谢～

Respect