python买卖股票的最佳时机(基于贪心/蛮力算法)


Posted in Python onJuly 05, 2019

开始刷leetcode算法题 今天做的是“买卖股票的最佳时机”

题目要求

给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。

注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

看到这个题目 最初的想法是蛮力法

通过两层循环 不断计算不同天之间的利润及利润和

下面上代码

class Solution(object):
  def maxProfit(self, prices):
    """
    :type prices: List[int]
    :rtype: int
    """

    self.allbuy1 = []  #单次买卖的差值数组 (可能为负)
    self.allbuy2 = []  #所有可能买卖的利润数组 (可能为负)
               # allbuy1和allbuy2的区别为一个是单次买卖 一个是多次买卖和
    self.curbuy(prices,0,0) #prices 为价格表 0:初始 0:
    #print(self.allbuy1)
    #print(self.allbuy2)
    return self.picBigest(self.allbuy2)
  def buyticket(self,prilist,a,b):    #list:放入的价格数组 a:上一次买入的价格 b:今天卖出的价格
    return prilist[b] -prilist[a]  #返回 赚取得价格

  def curbuy(self,plist,x,result): #plist:价格数组 x:当天的数组坐标 result: 利润
    obj=result           #固定上一次的价格 保存为上一个递归
    lens=len(plist)        #天数
    for i in range(x,lens-1):
      for j in range(i+1,lens):
        temp=self.buyticket(plist,i, j)
        self.allbuy1.append(temp)
        self.allbuy2.append(temp)   #单次利润放入数组
        result = obj + temp   #将之前的利润加上今天的利润
        if(x>=2):       #如果买入是第2+1天以后 则可以加上之前的利润
          self.allbuy2.append(result) #多次买卖利润放入数组
        self.curbuy(plist,j+1,result)  #递归 j+1:卖出的后一天 result:利润

  def picBigest(self,reslist): 
    big=0
    for i in reslist:
      if (i>big):
        big=i
    print(big)
    return big

if __name__ == '__main__':
    test=Solution()
    prices = [5,7,3,8] # 输入的每日股票数组
    test.maxProfit(prices)

分析:

这个代码理解起来简单 就是将所有可能都放入数组中 找出最大一个可能

将这个代码提交时 显示 超出时间限制 确实 如果输入的数组长度非常大时 计算量巨大 出现错误

——————————————————————————————————————————————————————————————————————————————

更换思路:利用贪心算法解决此事

首先介绍 一下贪心算法: 对问题只对当前情况进行最优解处理,之后发生什么对之前的决定都不改变。简单的说就是一个局部最优解的过程

介绍个例子就明白了: 找零钱问题

假设有面值为5元、2元、1元、5角、2角、1角的货币,需要找给顾客4元6角现金,为使付出的货币的数量最少

  • 首先找出小于4元6角的最大面值(2元)

  • 其次找出小于2元6角的最大面值(2元)

  • 接着找出小于6角的最大面值(5角)

  • 最后找出小于1角的最大面值(1角) ---付出4张纸币

介绍完了贪心算法简单思想 就利用该方法解决对应问题

在已知股票价格走势情况下 只需要对下一天进行判断 如果涨了 则买 如果跌了则卖 这样收益会保持固定增长

当然了 有人会提出 我可以选择不卖等几天再卖 或不买等几天再买 的方式 一样可以保持增长 但是如图

python买卖股票的最佳时机(基于贪心/蛮力算法)

如果在第2天买入 3天卖出 4天买入 5天卖出 收益为A+B

如果在第2天买入 5天卖出 收益为 C

明显得出A+B大于C 所以贪心法在这种情况非常适用并且肯定得到最优解

直接上代码

class Solution(object):
  def maxProfit(self, prices):
    profit = 0
    for day in range(len(prices)-1):
      differ = prices[day+1] - prices[day]
      if differ > 0:
        profit += differ
    return profit
if __name__ == '__main__':
    test=Solution()
    prices = [5,7,3,9] # 输入的每日股票数组
    print(test.maxProfit(prices))

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中使用第三方库xlrd来写入Excel文件示例
Apr 05 Python
简单的python后台管理程序
Apr 13 Python
Python request设置HTTPS代理代码解析
Feb 12 Python
详解分布式任务队列Celery使用说明
Nov 29 Python
使用PyQtGraph绘制精美的股票行情K线图的示例代码
Mar 14 Python
python 中Arduino串口传输数据到电脑并保存至excel表格
Oct 14 Python
Python 类的魔法属性用法实例分析
Nov 21 Python
将keras的h5模型转换为tensorflow的pb模型操作
May 25 Python
在python中list作函数形参,防止被实参修改的实现方法
Jun 05 Python
opencv 图像滤波(均值,方框,高斯,中值)
Jul 08 Python
Python如何把字典写入到CSV文件的方法示例
Aug 23 Python
python接口自动化框架实战
Dec 23 Python
Python 计算任意两向量之间的夹角方法
Jul 05 #Python
python实现两个经纬度点之间的距离和方位角的方法
Jul 05 #Python
Python3+Appium实现多台移动设备操作的方法
Jul 05 #Python
Python PIL读取的图像发生自动旋转的实现方法
Jul 05 #Python
python读出当前时间精度到秒的代码
Jul 05 #Python
python读写csv文件方法详细总结
Jul 05 #Python
Python考拉兹猜想输出序列代码实践
Jul 05 #Python
You might like
一篇有意思的技术文章php介绍篇
2010/10/26 PHP
探讨方法的重写(覆载)详解
2013/06/08 PHP
PHP mail()函数使用及配置方法
2014/01/14 PHP
PHP中nowdoc和heredoc使用需要注意的一点
2014/03/21 PHP
php获取URL中带#号等特殊符号参数的解决方法
2014/09/02 PHP
php查找字符串出现次数的方法
2014/12/01 PHP
yii 2.0中表单小部件的使用方法示例
2017/05/23 PHP
Javascript 检测键盘按键信息及键码值对应介绍
2013/01/03 Javascript
js style动态设置table高度
2014/10/21 Javascript
js读取并解析JSON类型数据的方法
2015/11/14 Javascript
JS取数字小数点后两位或n位的简单方法
2016/10/24 Javascript
原生js验证简洁注册登录页面
2016/12/17 Javascript
JS前端开发判断是否是手机端并跳转操作(小结)
2017/02/05 Javascript
在vue中使用jointjs的方法
2018/03/24 Javascript
vue watch关于对象内的属性监听
2019/04/22 Javascript
原生JS实现烟花效果
2020/03/10 Javascript
VUE中V-IF条件判断改变元素的样式操作
2020/08/09 Javascript
[05:53]完美世界携手游戏风云打造 卡尔工作室观战系统篇
2013/04/22 DOTA
Python def函数的定义、使用及参数传递实现代码
2014/08/10 Python
Redis使用watch完成秒杀抢购功能的代码
2018/05/07 Python
Python中实例化class的执行顺序示例详解
2018/10/14 Python
Django集成搜索引擎Elasticserach的方法示例
2019/06/04 Python
python关于调用函数外的变量实例
2019/12/26 Python
MySQL面试题
2014/01/12 面试题
长安大学毕业生自我鉴定
2014/01/17 职场文书
物流管理毕业生自荐信范文
2014/03/15 职场文书
应届生求职自荐信范文
2014/04/07 职场文书
《水上飞机》教学反思
2014/04/10 职场文书
小学教师评语大全
2014/04/23 职场文书
争当四好少年演讲稿
2014/09/13 职场文书
民间个人借款协议书
2014/09/30 职场文书
个人公司授权委托书范本
2014/10/12 职场文书
答谢词范文
2015/01/05 职场文书
2015年仓管员工作总结
2015/04/21 职场文书
联谊活动总结范文
2015/05/09 职场文书
Zabbix对Kafka topic积压数据监控的解决方案
2022/07/07 Servers