校招真题练习029 圈地运动(360)

圈地运动

圈地运动，就是用很多木棍摆在地上组成一个面积大于0的多边形～

小明喜欢圈地运动，于是他需要去小红店里面买一些木棍，期望圈出一块地来。小红想挑战一下小明，所以给小明设置了一些障碍。障碍分别是：
1.如果小明要买第i块木棍的话，他就必须把前i-1块木棍都买下来。
2.买了的木棍都必须用在圈地运动中。
那么请问小明最少买多少根木棍，才能使得木棍围成的图形是个面积大于0多边形呢？

输入描述:
第一行一个数n，表示木棍个数。
第二行n个数，第i个数表示第i个木棍的长度ai
1<=n<=10000
1<=ai<=10000

输出描述:
输出一个数，表示最少需要的木棍个数，如果无解输出-1

def main():
    N = int(input())
    if N <= 2:
        print(-1)
    else:
        ary = list(map(int,input().split()))
        leftpart = ary[0] + ary[1]
        curmax = max(ary[0],ary[1])
        for i in range(2,N):
            cur = ary[i]
            if cur >= curmax:#新元素是最大的
                curmax = cur#更新最大元素
                if leftpart > cur:
                    print(i+1)
                    return
                else:
                    leftpart += cur
            else:#新元素不是最大的
                #不更新最大元素
                leftpart -= curmax
                leftpart += cur
                if leftpart > curmax:
                    print(i+1)
                    return
                else:
                    leftpart += curmax
        print(-1)

if __name__ == '__main__':
    main()

算法思路：几何 + 贪心法。

判断是否能组成三角形：最小的两边之和大于第三边。

判断是否能组成四边形：最小的三边之和大于第四边。

……

判断是否能组成N边形，最小的N-1边之和大于第N边。