任务总执行时长

题目描述

任务编排服务负责对任务进行组合调度。

参与编排的任务有两种类型，其中一种执行时长为taskA，另一种执行时长为taskB。

任务一旦开始执行不能被打断，且任务可连续执行。

服务每次可以编排num个任务。

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

输入描述

第1行输入分别为

• 第1种任务执行时长taskA
• 第2种任务执行时长taskB
• 这次要编排的任务个数num

以逗号分隔。

输出描述

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

备注

注：每种任务的数量都大于本次可以编排的任务数量

• 0 < taskA
• 0 < taskB
• 0 <= num <= 100000

用例1

输入

1,2,3

Copy

输出

[3, 4, 5, 6]

Copy

说明

可以执行 3 次 taskA，得到结果 3；执行 2 次 taskA和 1 次 taskB，得到结果 4 。以此类推，得到最终结果。

#include <iostream>
#include <sstream>
#include <vector>
#include <set>
#include <string>
#include <iterator>
using namespace std;

string get_result(int taskA, int taskB, int num) {
    //任务数为0时直接返回空数组字符串
    if (num == 0) return "[]";

    //两个任务用时相等时 直接拿出一个乘以总次数再拼接字符串即可
    if (taskA == taskB) {
        return "[" + to_string(taskA * num) + "]";
    }

    //存储唯一值 键值对形式 自动排序
    set<int> ans;

    //向set容器中插入每种可能的计算值
    for (int i = 0; i <= num; i++) {
        //经历过这句之后 set容器中是最后结果的所有整数并有序
        ans.insert(taskA * i + taskB * (num - i));
    }

    //创建一个字符串result初始化为[。
    string result = "[";

    //使用一个布尔变量first来标记是否是第一个元素。
    bool first = true;

    //遍历集合ans中的每个元素，对于不是第一个元素的情况，在result中添加逗号和空格。
    for (const auto& value : ans) {
        if (!first) {
            result += ", ";
        }
        first = false;
        //然后将当前元素转换为字符串并添加到result中。
        result += to_string(value);
        //cout<<result<<endl;
    }
    //最后在result末尾添加]并返回这个字符串，作为最终的结果。
    result += "]";
    return result;
}

int main() {

    //从标准输入读取一个字符串并存储到str中。
    string str;
    cin >> str;

    vector<int> arr;

    //使用istringstream对象iss将输入的字符串str进行处理。
    istringstream iss(str);
    string token;

    //通过循环调用getline函数，以逗号,为分隔符，将字符串分割成一个个的子字符串token。
    while (getline(iss, token, ','))
    {
        //使用stoi函数将每个子字符串转换为整数，并将其添加到arr向量中。
        arr.push_back(stoi(token));
    }

    //保证输入合法性
    if (arr.size()!= 3) {
        cout << "[]" << endl;
        return 0;
    }

    //将taska、taskb、num传入get_result函数并返回结果输出
    cout << get_result(arr[0], arr[1], arr[2]) << endl;
    return 0;
}