其他分享
首页 > 其他分享> > 洛谷P1168 中位数

洛谷P1168 中位数

作者:互联网

【解题报告】洛谷P1168 中位数

题目链接

https://www.luogu.com.cn/problem/P1168

思路

这道题目是个数据结构

由于数据结构很长时间没有码了,所以就开了一个这个

发现不怎么有思路

本来想开一个数组,然后比较位置,然后加入数组的,发现炸掉

然后主流的思路有以下两种,我也基本搞懂了

STL大法

stl大法好!

用一个vector,然后直接查询就好了

然后vector我似乎不太会用,所以我就查了一下下百度

  • void push_back(const T& x):向量尾部增加一个元素X

  • iterator insert(iterator it,const T& x):向量中迭代器指向元素前增加一个元素x

  • iterator insert(iterator it,int n,const T& x):向量中迭代器指向元素前增加n个相同的元素x

  • iterator insert(iterator it,const_iterator first,const_iterator last):向量中迭代器指向元素前插入另一个相同类型向量的[first,last)间的数据

——https://www.cnblogs.com/zsq1993/p/5929806.html

这里可以直接使用第二个用法,然后加上一个 upper_bound() 就行了

话说我第一次知道有这种用法

如果数据是单调递减会不会炸掉啊

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#include <vector>
using namespace std;
const int maxn=100005;
int n;
vector <int> a;
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		int x;
		cin>>x;
		a.insert(upper_bound(a.begin(),a.end(),x),x);
		if(i&1)
		cout<<a[(i-1)/2]<<'\n';
	}
	return 0;

对顶堆

还是用到了stl

但是思想完全不一样

我们建一个大根堆,建一个小根堆

然后就完了

左边是一个大根堆,堆顶靠近mid,右边是一个小根堆,堆顶靠近mid

每次查询的时候调整堆的大小,是两个堆的大小相差为1,多的那个第一个就是中位数,速度会比用vector的更快一些

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#include <vector>
#include <queue>
using namespace std;
const int maxn=100005;
int n;
int a[maxn];
int jdz(int x)
{
	return x>0? x:-x;
}
priority_queue <int,vector<int>,less<int> > q1; //大根堆 
priority_queue <int,vector<int>,greater<int> > q2;//小根堆 
int main()
{
	cin>>n;
	cin>>a[1];
	int mid=a[1];
	cout<<mid<<'\n';
	for(int i=2;i<=n;i++)
	{
		cin>>a[i];
		if(a[i]>mid) 
		q2.push(a[i]);
		else
		q1.push(a[i]);
		if(i&1)
		{
			while(q1.size()!=q2.size())
			{
				if(q1.size()>q2.size())
				{
					q2.push(mid);
					mid=q1.top();
					q1.pop();
				}
				else
				{
					q1.push(mid);
					mid=q2.top();
					q2.pop();
				}
			}
			cout<<mid<<'\n';
		}
	}
	return 0;
}

标签:q1,q2,洛谷,iterator,int,中位数,mid,include,P1168
来源: https://www.cnblogs.com/wweiyi2004/p/15421720.html