力扣初级算法2

怎么大年纪了才开始学习Python，是不是有点儿晚了？但还是挺喜欢敲代码的感觉。看了python有几天了，开始刷力扣的题，在这里记录一下自己的学习过程，和代码。希望大伙提提意见，帮我找出代码中不足之处，让我能够进步快点儿，另外也给同样开始刷题的小伙伴一个参考。（代码已通过测试）（本人现在使用Labview开发测试设备的软件，有懂的小伙伴可以给我留言交流啊）

 代码如下：

#coding:utf-8
#力扣初级算法第二题：买卖股票的最佳时机
#测试数据
n = [1,2,3,4,5]
m = [3,1,5,7,2]
q = [5,4,3,2,1]
p = [2]
x = [0,3,8,6,8,6,6,8,2,0,2,7]
#测试代码
def maxProfit(prices):
    buy = []
    sell = []
    rich = 0
    #数据处理
    #在输入的数据中相邻数据如果相等，是没有必要的，需要去除其中一个，本程序定义去除前面的数据
    removelist = []     #存放输入数据中需要去除的元素的索引
    temp = 0            #中间变量
    #找到需要去除的元素的索引，将其放入removelist列表中
    #因为要输出的元素在原列表中不止一个，而我们想要删除的是特定位置的元素，所以不能使用remove方法
    for i in range(len(prices)-1):
        if prices[i] == prices[i+1]:
            removelist.append(i)
    #print(removelist)
    #去除元素
    for i in removelist:
        prices.pop(i-temp)  #pop方法：删除列表中指定位置的元素
        #因为列表中某个元素被删除后，列表长度将变短1
        temp += 1
    #print(prices)

    #找到所有可购买的点
    if len(prices) <= 1:    #如果输入只有一个数据或没有数据，就没有购买的必要
        return 0
    else:
        #从第一个元素开始，到倒数第二个元素为止，因为最后一定不能买入
        #买入一定比卖出价格低，所以先找满足prices[i]0,则需要判断prices[i]= prices[j+1]:
                    sell.append(prices[j])
            else:
                if prices[j] >= prices[j-1]:
                    sell.append(prices[j])

    #print(sell)
    for i in range(len(buy)):
        rich += (sell[i] - buy[i])

    return rich

print(maxProfit(n))
print(maxProfit(m))
print(maxProfit(p))
print(maxProfit(q))
print(maxProfit(x))