代码随想录训练营第十二天|LeetCode 239滑动窗口最大值、347前k个高频元素

最新推荐文章于 2024-09-15 15:43:59 发布

gin7773

最新推荐文章于 2024-09-15 15:43:59 发布

阅读量34

点赞数

分类专栏： LeetCode刷题文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_43133169/article/details/128978957

版权

LeetCode刷题专栏收录该内容

44 篇文章 0 订阅

订阅专栏

文章介绍了如何利用单调队列、优先队列（小顶堆）解决LeetCode中的两道问题：239.滑动窗口最大值和347.前k个高频元素。对于滑动窗口最大值，通过维护一个单调递减的队列来高效地找到每个窗口的最大值；对于前k个高频元素，使用哈希映射统计频率后，可以借助小顶堆在O(n*logk)的时间复杂度内找出高频元素。

摘要由CSDN通过智能技术生成

LeetCode 239滑动窗口最大值

题目链接：239.滑动窗口最大值

定义单调队列

//写法一
class Solution {
private:
	class MyQue {
	public:
		deque<int> que;
		void pop(int value) {
			if (!que.empty() && que.front() == value) {
				que.pop_front();
			}
		}
		void push(int value) {
			while (!que.empty() && que.back() < value) {
				que.pop_back();
			}
			que.push_back(value);
		}
		int getFront() {
			return que.front();
		}
	};
public:
	vector<int> maxSlidingWindow(vector<int>& nums, int k) {
		MyQue myQue;
		vector<int> ans;
		for (int i = 0; i < k; ++i) {
			myQue.push(nums[i]);
		}
		ans.push_back(myQue.getFront());
		for (int i = k; i < nums.size(); ++i) {
			myQue.pop(nums[i - k]);
			myQue.push(nums[i]);
			ans.push_back(myQue.getFront());
		}
		return ans;
	}
};

//写法二
class Solution {
public:
	vector<int> maxSlidingWindow(vector<int>& nums, int k) {
		vector<int> ans;
		deque<int> que;
		for (int i = 0; i < k; ++i) {
			while (!que.empty() && nums[i] > que.back()) {
				que.pop_back();
			}
			que.push_back(nums[i]);
		}
		ans.push_back(que.front());
		for (int i = k; i < nums.size(); ++i) {
			if (nums[i - k] == que.front()) {
				que.pop_front();
			}
			while (!que.empty() && nums[i] > que.back()) {
				que.pop_back();
			}
			que.push_back(nums[i]);
			ans.push_back(que.front());
		}
		return ans;
	}
};

使用优先队列priority_queue()

class Solution {
public:
	vector<int> maxSlidingWindow(vector<int>& nums, int k) {
		vector<int> ans;
		priority_queue<pair<int, int>> pque;
		for (int i = 0; i < k; ++i) {
			pque.emplace(nums[i], i);
		}
		ans.push_back(pque.top().first);
		for (int i = k; i < nums.size(); ++i) {
			pque.emplace(nums[i], i);
			while (pque.top().second <= i - k) pque.pop();
			ans.push_back(pque.top().first);
		}
		return ans;
	}
};

LeetCode 347前k个高频元素

题目链接：347.前k个高频元素

class Solution {
public:
	bool static cmp(const pair<int, int>& a, const pair<int, int>& b) {
		return a.second > b.second;
	}
	vector<int> topKFrequent(vector<int>& nums, int k) {
		vector<int> ans;
		unordered_map<int, int> umap;
		for (int i = 0; i < nums.size(); ++i) {
			umap[nums[i]]++;
		}
		vector<pair<int, int>> vec(umap.begin(), umap.end());
		sort(vec.begin(), vec.end(), cmp);
		int count = 0;
		for (auto it = vec.begin(); it != vec.end() && count < k; ++it, ++count) {
			ans.push_back((*it).first);
		}
		return ans;
	}
};

用小顶堆，不用对整个统计频率的数组进行排序，优化时间复杂度
O(n*logk)

class Solution {
public:
	//小顶堆
	class cmp {
	public:
		bool operator() (const pair<int, int>& a, const pair<int, int>& b) {
			return a.second > b.second;
		}
	};
	vector<int> topKFrequent(vector<int>& nums, int k) {
		vector<int> ans(k);
		unordered_map<int, int> umap;
		for (int i = 0; i < nums.size(); ++i) {
			umap[nums[i]]++;
		}
		
		//用小顶堆.仅需维护k个元素
		priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> pque;
		for (auto it = umap.begin(); it != umap.end(); ++it) {
			pque.emplace(*it);
			if (pque.size() > k) {
				pque.pop();
			}
		}
		for (int i = k - 1; i >= 0; --i) {
			ans[i] = pque.top().first;
			pque.pop();
		}
		return ans;
	}
};