序列合并

学废c++

于 2023-10-21 08:30:00 发布

阅读量200

点赞数 1

分类专栏： C++ NOIP普及组文章标签： c++ 算法数据结构

本文链接：https://blog.csdn.net/panpanpan17452/article/details/133953425

版权

C++ 同时被 2 个专栏收录

86 篇文章 3 订阅

订阅专栏

NOIP普及组

32 篇文章 1 订阅

订阅专栏

题目描述

有两个长度为 N 的单调不降序列 A,B，在 A,B 中各取一个数相加可以得到 $N^{2}$ 个和，求这 $N^{2}$ 个和中最小的 $N$ 个。

输入格式

第一行一个正整数 N；

第二行 N 个整数 1…N。

第三行 N 个整数 1…N。

输出格式

一行 N 个整数，从小到大表示这 N 个最小的和。

样例输入

3
2 6 6
1 4 8

样例输出

3 6 7

参考代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;

priority_queue<ll> q;
stack <ll> st;

const int N = 1e5 + 5;
int a[N], b[N];

int main()
{
	int n;
	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 + n + 1);
	sort(b + 1, b + n + 1);
		
	for(int i = 1; i <= n; i++)
	{
		for(int j = 1; j <= n; j++)
		{
			ll h = a[i] + b[j];
			if(q.size() < n)
				q.push(h);
			else
			{
				if (q.top() < h)
					break;
				q.push(h);
				q.pop();
			}
		}
	}
	
	for(int i = 1; i <= n; i++)
	{
		st.push(q.top());
		q.pop();
	}	
	
	while(!st.empty())
	{
		cout<<st.top()<<" ";
		st.pop();
	}
	
	return 0;
}