在C++中查找通过重新排列第二个数组的元素可以形成的字典序最小的序列
假设我们有两个包含n个数字的数组A和B,我们必须在B自身中重新排列B的元素,以便在重新排列后由(A[i] + B[i]) % n形成的序列在字典序上最小。最后,我们将返回字典序上最小的可能序列。
因此,如果输入类似于A = {1, 2, 3, 2},B = {4, 3, 2, 2},则输出将为[0, 0, 1, 2]
为了解决这个问题,我们将遵循以下步骤:
n := a的大小
定义一个映射my_map
定义一个集合my_set
对于初始化i := 0,当i < n时,更新(i增加1),执行:
(my_map[b[i]]增加1)
将b[i]插入my_set
定义一个数组sequence
对于初始化i := 0,当i < n时,更新(i增加1),执行:
如果a[i]等于0,则:
it := my_set中不小于0的元素,最初指向第一个元素
value := it
在sequence的末尾插入value mod n
(my_map[value]减少1)
如果my_map[value]为零,则:
从my_set中删除value
否则
x := n - a[i]
it := my_set中不小于x的元素,最初指向第一个元素
如果it不在my_set中,则:
it := my_set中不小于0的元素,最初指向第一个元素
value := it
在sequence的末尾插入(a[i] + value) mod n
(my_map[value]减少1)
如果my_map[value]为零,则:
从my_set中删除value
返回sequence
示例
让我们看看下面的实现,以便更好地理解:
#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<auto> v) {
cout << "[";
for (int i = 0; i < v.size(); i++) {
cout << v[i] << ", ";
}
cout << "]" <; endl;
}
vector<int> solve(vector<int>&a, vector<int> &b) {
int n = a.size();
unordered_map<int, int> my_map;
set<int> my_set;
for (int i = 0; i < n; i++) {
my_map[b[i]]++;
my_set.insert(b[i]);
}
vector<int> sequence;
for (int i = 0; i < n; i++) {
if (a[i] == 0) {
auto it = my_set.lower_bound(0);
int value = *it;
sequence.push_back(value % n);
my_map[value]--;
if (!my_map[value])
my_set.erase(value);
}
else {
int x = n - a[i];
auto it = my_set.lower_bound(x);
if (it == my_set.end())
it = my_set.lower_bound(0);
int value = *it;
sequence.push_back((a[i] + value) % n);
my_map[value]--;
if (!my_map[value])
my_set.erase(value);
}
}
return sequence;
}
int main() {
vector<int> a = {1, 2, 3, 2};
vector<int> b = {4, 3, 2, 2};
vector<int> res = solve(a, b);
print_vector(res);
}输入
{1, 2, 3, 2}, {4, 3, 2, 2}输出
[0, 0, 1, 2, ]
广告
数据结构
网络
关系数据库管理系统 (RDBMS)
操作系统
Java
iOS
HTML
CSS
Android
Python
C语言编程
C++
C#
MongoDB
MySQL
Javascript
PHP