在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, ]

更新于:2020年8月20日

380 次浏览

开启你的职业生涯

通过完成课程获得认证

开始学习
广告
© . All rights reserved.