[LeetCode]628. Maximum Product of Three Numbers

[LeetCode]628. Maximum Product of Three Numbers

题目描述

这里写图片描述

思路

简单的做法,扫描一次即可,维护两个最小值和三个最大值

看到这个题目,想起一道面试题,一个无法载入内存的数组,从中选n个最大的数
解法,建立一个小根堆,限定元素个数为n,每次插入一个新元素后弹出堆顶元素,扫描一次,堆中的n个元素即为n个最大的数。

这道题思路可以借鉴这种方式,顺便回顾一下调整堆的方法
维护n = 3的小根堆 和 n = 2 的大根堆
最后将小根堆使用大根堆调整,这样最大的元素就是堆顶元素

详见代码

代码

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

class Solution {
public:
    void print(vector<int> nums) {
        for (int num : nums)
            cout << num << ' ';
        cout << endl;
    }

    int maximumProduct(vector<int>& nums) {
        vector<int> maxHeap, minHeap;
        for (int i = 0; i < nums.size(); ++i) {
            if (i < 3)
                minHeap.push_back(nums[i]);
            else {
                minHeap.push_back(nums[i]);
                adjustMinHeap(minHeap, 0);
                swap(minHeap[0], minHeap[3]);
                minHeap.pop_back();
            }
            if (i < 2)
                maxHeap.push_back(nums[i]);
            else {
                maxHeap.push_back(nums[i]);
                adjustMaxHeap(maxHeap, 0);
                swap(maxHeap[0], maxHeap[2]);
                maxHeap.pop_back();
            }
        }
        adjustMaxHeap(minHeap, 0);
        print(minHeap);
        print(maxHeap);
        return max(minHeap[0] * minHeap[1] * minHeap[2], maxHeap[0] * maxHeap[1] * minHeap[0]);
    }

    void adjustMaxHeap(vector<int>& nums, int start) {
        int leftChild = 2 * start + 1, rightChild = 2 * start + 2;
        if (leftChild < nums.size()) {
            adjustMaxHeap(nums, leftChild);
            if (nums[start] < nums[leftChild]) {
                swap(nums[start], nums[leftChild]);
                adjustMaxHeap(nums, leftChild);
            }
        }
        if (rightChild < nums.size()) {
            adjustMaxHeap(nums, rightChild);
            if (nums[start] < nums[rightChild]) {
                swap(nums[start], nums[rightChild]);
                adjustMaxHeap(nums, rightChild);
            }
        }
    }

    void adjustMinHeap(vector<int>& nums, int start) {
        int leftChild = 2 * start + 1, rightChild = 2 * start + 2;
        if (leftChild < nums.size()) {
            adjustMinHeap(nums, leftChild);
            if (nums[start] > nums[leftChild]) {
                swap(nums[start], nums[leftChild]);
                adjustMinHeap(nums, leftChild);
            }
        }
        if (rightChild < nums.size()) {
            adjustMinHeap(nums, rightChild);
            if (nums[start] > nums[rightChild]) {
                swap(nums[start], nums[rightChild]);
                adjustMinHeap(nums, rightChild);
            }
        }
    }
};

int main() {
    vector<int> nums = {9,1,5,6,7,2};
    Solution s;

    cout << s.maximumProduct(nums) << endl;

    system("pause");
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值