校招真题练习011 种花(美团)

种花

题目描述
公园里有N个花园，初始时每个花园里都没有种花，园丁将花园从1到N编号并计划在编号为i的花园里恰好种A_i朵花，他每天会选择一个区间[L，R]（1≤L≤R≤N）并在编号为L到R的花园里各种一朵花，那么园丁至少要花多少天才能完成计划？

输入描述:
第一行包含一个整数N，1≤N≤10^5。
第二行包含N个空格隔开的整数A_1到A_N，0≤A_i≤10^4。

输出描述:
输出完成计划所需的最少天数。

思路一：贪心法(AC)

if __name__ == '__main__':
    N = int(input())
    ary = list(map(int,input().strip().split()))
    cnt = 0
    for i in range(1,N):
        if ary[i-1] > ary[i]:
            cnt += ary[i-1] - ary[i]
    print(cnt + ary[N-1])

思路二：分治法(只能通过60%)

def findMinIndex(ary):
    minValue = sys.maxsize
    minIndex = -1
    for i in range(len(ary)):
        if ary[i] < minValue:
            minValue = ary[i]
            minIndex = i
    for i in range(len(ary)):
        ary[i] -= minValue
    return ary,minIndex,minValue

def calFlowers(ary,L,R):
    if L == R:
        return ary[L]
    elif L < R:
        ary2,minI,minV = findMinIndex(ary[L:R+1])
        left = calFlowers(ary2,0,minI-1)
        right = calFlowers(ary2,minI+1,len(ary2)-1)
        return left + right + minV
    else:
        return 0
        
if __name__ == '__main__':
    N = int(input())
    ary = list(map(int,input().strip().split()))
    
    result = calFlowers(ary,0,N-1)
    print(result)