183.小E君自助餐厅客流量分析|Marscode AI刷题

1.题目

问题描述

小E君需要根据每天的客流量数据进行分析，以制定自助餐厅的备货策略。她需要计算前ii天的客流量平均值，并将其四舍五入为整数。

给定餐厅营业的总天数NN，以及每天的客流量数据RiRi，你需要输出一个长度为NN的序列，其中第ii个值表示前ii天的平均客流量。要求所有输出结果四舍五入到最接近的整数。

---

测试样例

样例1：

输入：N = 5 ,R = [1, 2, 3, 4, 10]

输出：[1, 2, 2, 3, 4]

样例2：

输入：N = 3 ,R = [5, 10, 15]

输出：[5, 8, 10]

2.思路

遍历R数组，每次累加R[i]，从而求出前i天的平均客流量

注意：用round()实现四舍五入需要被除数是double类型，如果是int类型，只能取整而不能四舍五入

3.代码

#include <iostream>
#include <vector>
#include <string>
#include <cmath>

using namespace std;

vector<int> solution(int N, vector<int> R) {
    // PLEASE DO NOT MODIFY THE FUNCTION SIGNATURE
    // write code here
    vector<int> result;
    double sum_trafficFlow = 0;
    for (int i = 0; i < N; i++) {
        sum_trafficFlow += R[i];
        result.push_back(round(sum_trafficFlow / (i + 1)));
    }
    return result;
}

int main() {
    // vector<int> result = solution(5, {1, 2, 3, 4, 10});
    // for (int i = 0; i < result.size(); i++) {
    //     cout << result[i] << endl;
    // }
    cout << (solution(5, {1, 2, 3, 4, 10}) == vector<int>{1, 2, 2, 3, 4}) << endl;
    cout << (solution(3, {5, 10, 15}) == vector<int>{5, 8, 10}) << endl;
    return 0;
}