1631 序列合并

1631 序列合并

个人感觉，二叉堆类型的题都很类似
这道题和之前做过的一些题非常的类似，如果用暴力很好就能做出来，得64分，详细得看代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;
const int SIZE=1e5+5;
int n;
long long a[SIZE],b[SIZE];
priority_queue<int,vector<int>,greater<int> > q;
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=1;i<=n;i++)
		cin>>b[i];
	sort(a+1,a+1+n);
	sort(b+1,b+1+n);//从小到大排序，方便求和的时候从小开始
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			q.push(a[i]+b[j]);//求和
		}
	}
	for(int i=1;i<=n;i++)
		cout<<q.top()<<" ",q.pop();//从小到大输出
	return 0; 
}

所以，这就是一个纯暴力，只有64分会超时，可以进行小小的优化，从而达到正解的操作
我们只是要输出从小大的和的n个数，所以没必要存储n个数以外的数
所以这就限制了一个长度为n的窗口，队列不足长度的时候，可以不断地压入；如果长度满了，这里还得用一个优化，当前两个数的和不是前n个小的数的和，数列是升序的，所以后面的数也不会满足这个性质，直接退出不做多余的操作，反之，需要更新堆顶，满足堆顶为第n小的数

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;
const int SIZE=1e5+5;
int n;
long long a[SIZE],b[SIZE];
int answer[SIZE];
priority_queue<int> q;
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=1;i<=n;i++)
		cin>>b[i];
	sort(a+1,a+1+n);
	sort(b+1,b+1+n);//从小到大的和 
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			int x=a[i]+b[j];
			if(q.size()<n) //因为只需要n个数，所以限制窗口 
			{
				q.push(x);
			 }
			 else
			 {
			 	if(q.top()>x)//当前的堆顶不满足第n个小的数 
			 	{
			 		q.pop();
			 		q.push(x);
				 }
				 else break;//后序队列有序，当前点进不去，其他的也进不去 
			 }
		}
	}
	for(int i=1;i<=n;i++) 
	{
		answer[i]=q.top();
		q.pop();
	}
	for(int i=n;i>=1;i--)
	{
		cout<<answer[i]<<" ";
	}
	return 0; 
}