002 任务总执行时长

002 任务总执行时长

题目描述

任务编排服务负责对任务精选组合调度。参与编排的任务有两种类型,其中一种任务执行时长为taskA,另一种执行时长为taskB。任务一旦开始执行不能被打断,且任务可连续执行。服务每次可以编排num个任务。

请编写一个方法,生成每次编排后的任务所有可能的总执行时长。

输入描述

一行输入三个数字,第一个数字表示第一种任务taskA的执行时长,第二个数字表示第二种任务taskB的执行时长,第三个数字表示要编排的任务个数num,并以逗号分隔。

输出描述

数组形式返回所有总执行时长,需要按从小到大排列。

示例描述

示例一

输入

1,2,3

输出:

3, 4, 5, 6

说明: 一共有4次任务编排的总时长,分别是3、4、5、6,其中:

可以执行3次taskA,得到结果3

可以执行2次taskA和1次taskB,得到结果4

可以执行1次taskA和2次taskB,得到结果5

可以执行3次taskB,得到结果6

解题思路

分别从输入字符串中读取taskA和taskB的执行时长、要编排的任务个数num。

观察输出可知,可用循环依次递减计算taskA的执行次数的时长,并加上依次递增的taskB的执行次数的时长,可得到单次计算时长的关系式:

total_time = i ⋅ taskA + ( num − i ) ⋅ taskB \text{total\_time} = i \cdot \text{taskA} + (\text{num} - i) \cdot \text{taskB} total_time=i⋅taskA+(num−i)⋅taskB

python去写就是

python 复制代码
total_time = i * taskA + (num - i) * taskB

将单次时长存放到数组中,由于需要去重和排序,可将数组用set进行去重,并使用sorted进行排序,得到结果。

解题代码

bash 复制代码
def solve_method(string):
    str_list = string.split(",")
    taskA = int(str_list[0])
    taskB = int(str_list[1])
    num = int(str_list[2])

    result = []
    for i in range(num + 1):
        total_time = i * taskA + (num - i) * taskB
        result.append(total_time)

    return sorted(list(set(result)))


if __name__ == '__main__':
    assert solve_method("1,2,3") == [3, 4, 5, 6]
相关推荐
丁浩66628 分钟前
Python机器学习---2.算法:逻辑回归
python·算法·机器学习
best_virtuoso42 分钟前
PostgreSQL 常见数组操作函数语法、功能
java·数据结构·postgresql
B站_计算机毕业设计之家1 小时前
计算机毕业设计:Python农业数据可视化分析系统 气象数据 农业生产 粮食数据 播种数据 爬虫 Django框架 天气数据 降水量(源码+文档)✅
大数据·爬虫·python·机器学习·信息可视化·课程设计·农业
Q_Q5110082851 小时前
python+uniapp基于微信小程序的旅游信息系统
spring boot·python·微信小程序·django·flask·uni-app·node.js
伏小白白白1 小时前
【论文精度-2】求解车辆路径问题的神经组合优化算法:综合展望(Yubin Xiao,2025)
人工智能·算法·机器学习
鄃鳕1 小时前
python迭代器解包【python】
开发语言·python
无敌最俊朗@2 小时前
数组-力扣hot56-合并区间
数据结构·算法·leetcode
懷淰メ2 小时前
python3GUI--模仿百度网盘的本地文件管理器 By:PyQt5(详细分享)
开发语言·python·pyqt·文件管理·百度云·百度网盘·ui设计
Q_Q5110082852 小时前
python基于web的汽车班车车票管理系统/火车票预订系统/高铁预定系统 可在线选座
spring boot·python·django·flask·node.js·汽车·php
新子y2 小时前
【小白笔记】普通二叉树(General Binary Tree)和二叉搜索树的最近公共祖先(LCA)
开发语言·笔记·python