题解 | #牛群的第 m大的和# 模拟

牛群的第 m大的和

https://www.nowcoder.com/practice/04257790a4314942b26df19df9f68581

首先题目有误,要求的是第 m 大的和,而不是第 m 小的和

那么就保存一个大小为 m 的数组 sums ,对于数组 nums 中的每一个数,依次与 sums 中的数相加,即可得到结果

class Solution {
  private:
    void getSum(vector<int>& nums, int start, vector<int>& sums, int m) {
        int size = sums.size();
        for (int i = 0; i < size; i++) {
            sums.push_back(start + sums[i]);
        }
        if (sums.size() > m) {
            sort(sums.begin(), sums.end(), greater<>());
            sums.resize(m);
        }
    }
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param nums int整型vector
     * @param m int整型
     * @return int整型
     */
    int mthSmallestSum(vector<int>& nums, int m) {
        vector<int> sums;
        sums.emplace_back(0);
        for (int i = 0; i < nums.size(); i++) {
            getSum(nums, nums[i], sums, m);
        }
        return sums.back();
    }
};

时间复杂度:O(m*n),其中m,n分别为 sums 和 nums 的大小,在最坏情况下,nums 的元素要进行 m 次循环

空间复杂度:O(m),保存 sums 数组

全部评论

相关推荐

不愿透露姓名的神秘牛友
06-29 17:30
点赞 评论 收藏
分享
废物一个0offer:认真的吗二本本科找人工智能岗位
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务