Codeforces C. Min-Max Array Transformation

Problem - C - Codeforces

题意：给出两个非降序整数列a,b，求出每个a[i]+d=b[i] d的最大和最小值，d为非负整数。每个d的             最大值和最小值独立，也就是说可以是不同的对应关系。

思路：对于最小值，因为d必须是非负数，故第一个大于等于a[i]的b[i]与a[i]的差即为最小值，此步            操作可以用lower_bound函数实现。

           对于最大值，若不考虑其他数的对应情况，直接用b的最大值减去a[i]。但考虑a与b必须一一对应的情况，则需要对于a数组中大于a[i]的数都有匹配过的b，a[i]才可以剩下的b中选出最大值。此操作用multiset来实现，从a[n]开始，mx[i]就等于set中最大的数减去a[i]，然后将set中第一个大于等于a[i]的数删去，就达到了目的。

代码如下：

#include <iostream>
#include <cstring>
#include <queue>
#include <algorithm>
#include <cstdio>
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define ll long long
#define ull unsigned long long
#define inf 0x3f3f3f3f3f3f3f3f
#define mem(a, b) memset(a, b, sizeof(a))
#define rep(i, a, b) for (auto i = a; i <= b; ++i)
#define bep(i, a, b) for (auto i = a; i >= b; --i)
#define lowbit(x) x &(-x)
#define PII pair<int, int>
#define PLL pair<ll, ll>
#define PI acos(-1)
#define pb push_back
#define eps 1e-6
#define X1 first
#define Y1 second
#define IOS                      \
    ios::sync_with_stdio(false); \
    cin.tie(0);                  \
    cout.tie(0);
const int mod = 998244353;
const int MO = 1e9 + 10;
const int NN = 1e8 + 10;
const int MM = 1e7 + 10;
const int N = 1e6 + 10;
const int M = 2e5 + 10;
int dx[] = {-1, 0, 1, 0};
int dy[] = {0, 1, 0, -1};
int dxy[][2] = {{0, 1}, {1, 0}, {1, 1}, {-1, 1}};
using namespace std;

int  a[M], b[M];
int mi[M], mx[M];
int main()
{
    int t;
    cin >> t;
    while (t--)
    {
        int n;
        cin >> n;
        for (int i=0;i<n;i++)
            scanf("%lld",&a[i]);
        for (int i=0;i<n;i++)
            scanf("%lld",&b[i]);
        for (int i=0;i<n;i++)
        {
            int pos = lower_bound(b , b + n , a[i]) - b;
            mi[i] = b[pos] - a[i];
        }

        multiset<int> p(b , b + n );
        for (int i = n-1; i>=0; i--)
        {
            mx[i] = *(--p.end()) - a[i];
            p.erase(p.lower_bound(a[i]));
        }

        for (int i=0;i<n;i++)
            cout << mi[i] << " ";
        cout << "\n";
        for (int i=0;i<n;i++)
            cout << mx[i] << " ";
        cout << "\n";
    }
    return 0;
}

(7条消息) multiset用法总结_二喵君的博客-CSDN博客_multiset