洛谷刷题之p1631

序列合并

题目入口

题目描述

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

输入格式

第一行一个正整数 N N N;

第二行 N N N 个整数 A 1 ... N A_{1\dots N} A1...N。

第三行 N N N 个整数 B 1 ... N B_{1\dots N} B1...N。

输出格式

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

样例 #1

样例输入 #1

复制代码
3
2 6 6
1 4 8

样例输出 #1

复制代码
3 6 7

提示

对于 50 % 50\% 50% 的数据, N ≤ 1 0 3 N \le 10^3 N≤103。

对于 100 % 100\% 100% 的数据, 1 ≤ N ≤ 1 0 5 1 \le N \le 10^5 1≤N≤105, 1 ≤ a i , b i ≤ 1 0 9 1 \le a_i,b_i \le 10^9 1≤ai,bi≤109。

题解

设行为 A i A_i Ai 列为 B j B_j Bj

由题知,很显然排完序的A数组与B数组的和呈此关系,那也知道 A 1 + B 1 A_1+B_1 A1+B1的值是最小的,其余关系如图。

证明:
a i < a i + 1 , a_i<a_{i+1}, ai<ai+1,在 b j b_j bj一定时, a i + b j < a i + 1 + b j a_i+b_j<a_{i+1}+b_j ai+bj<ai+1+bj
b i < b i + 1 , b_i<b_{i+1}, bi<bi+1,在 a j a_j aj一定时, b i + a j < b i + 1 + a j b_i+a_j<b_{i+1}+a_j bi+aj<bi+1+aj

所以左上角最小,右下角最大

那我们可以先把 a i + b 1 a_i+b_1 ai+b1加入到优先队列中,然后弹出最小的,假设这个最小值是由 a x + b y a_x+b_y ax+by构成,那么再把 a x + b y + 1 a_x+b_{y+1} ax+by+1放入优先队列中

最后记得重载运算符

Code

cpp 复制代码
#include <bits/stdc++.h>

using namespace std;

const int Maxn = 1e5 + 10;
int pos_b[Maxn];
int a[Maxn], b[Maxn];
int id[Maxn];
struct node
{
    int pos;
    int num;
    bool operator<(const node &cur) const
    {
        return num > cur.num;
    }
};
priority_queue<node> c;
int n;
void read()
{
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
    }
    for (int i = 1; i <= n; i++)
    {
        cin >> b[i];
    }
}
void solve()
{
    sort(a + 1, a + n + 1);
    sort(b + 1, b + n + 1);
    for (int i = 1; i <= n; i++)
    {
        c.push({i, a[i] + b[1]});
        id[i] = 1;
    }
    for (int i = 1; i <= n; i++)
    {
        node x = c.top();
        c.pop();
        cout << x.num << " ";
        int id2 = x.pos;
        c.push({id2, a[id2] + b[++id[id2]]});
    }
}
int main()
{
    read();
    solve();
    return 0;
}
相关推荐
格林威4 分钟前
常规线扫描镜头有哪些类型?能做什么?
人工智能·深度学习·数码相机·算法·计算机视觉·视觉检测·工业镜头
程序员莫小特2 小时前
老题新解|大整数加法
数据结构·c++·算法
小刘max3 小时前
深入理解队列(Queue):从原理到实践的完整指南
数据结构
过往入尘土3 小时前
服务端与客户端的简单链接
人工智能·python·算法·pycharm·大模型
zycoder.3 小时前
力扣面试经典150题day1第一题(lc88),第二题(lc27)
算法·leetcode·面试
蒙奇D索大3 小时前
【数据结构】考研数据结构核心考点:二叉排序树(BST)全方位详解与代码实现
数据结构·笔记·学习·考研·算法·改行学it
洲覆4 小时前
C++ 模板、泛型与 auto 关键字
开发语言·数据结构·c++
MoRanzhi12034 小时前
15. Pandas 综合实战案例(零售数据分析)
数据结构·python·数据挖掘·数据分析·pandas·matplotlib·零售
智驱力人工智能4 小时前
工厂抽烟检测系统 智能化安全管控新方案 加油站吸烟检测技术 吸烟行为智能监测
人工智能·算法·安全·边缘计算·抽烟检测算法·工厂抽烟检测系统·吸烟监测
程序员爱钓鱼4 小时前
Go语言实战案例——进阶与部署篇:编写Makefile自动构建Go项目
后端·算法·go