Sort simultaneously
(Mylib/Utils/sort_simultaneously.cpp)

• View this file on GitHub
• Last update: 2021-04-23 23:44:44+09:00
• Link: View error logs on GitHub Actions

Operations

Requirements

Notes

Problems

References

Verified with

• test/aoj/2426/main.test.cpp
• test/yosupo-judge/rectangle_sum/main.persistent_segment_tree.test.cpp
• test/yukicoder/235/main.test.cpp

Code

#pragma once
#include <algorithm>
#include <cassert>
#include <initializer_list>
#include <numeric>
#include <utility>
#include <vector>

namespace haar_lib {
  namespace sort_simultaneously_impl {
    template <typename T>
    void helper(int N, const std::vector<int> &ord, std::vector<T> &a) {
      std::vector<T> temp(N);
      for (int i = 0; i < N; ++i) temp[i] = a[ord[i]];
      std::swap(temp, a);
    }
  }  // namespace sort_simultaneously_impl

  template <typename Compare, typename... Args>
  void sort_simultaneously(const Compare &compare, std::vector<Args> &... args) {
    const int N = std::max({args.size()...});
    assert((int) std::min({args.size()...}) == N);
    std::vector<int> ord(N);
    std::iota(ord.begin(), ord.end(), 0);
    std::sort(ord.begin(), ord.end(), compare);

    (void) std::initializer_list<int>{
        (void(sort_simultaneously_impl::helper(N, ord, args)), 0)...};
  }
}  // namespace haar_lib

#line 2 "Mylib/Utils/sort_simultaneously.cpp"
#include <algorithm>
#include <cassert>
#include <initializer_list>
#include <numeric>
#include <utility>
#include <vector>

namespace haar_lib {
  namespace sort_simultaneously_impl {
    template <typename T>
    void helper(int N, const std::vector<int> &ord, std::vector<T> &a) {
      std::vector<T> temp(N);
      for (int i = 0; i < N; ++i) temp[i] = a[ord[i]];
      std::swap(temp, a);
    }
  }  // namespace sort_simultaneously_impl

  template <typename Compare, typename... Args>
  void sort_simultaneously(const Compare &compare, std::vector<Args> &... args) {
    const int N = std::max({args.size()...});
    assert((int) std::min({args.size()...}) == N);
    std::vector<int> ord(N);
    std::iota(ord.begin(), ord.end(), 0);
    std::sort(ord.begin(), ord.end(), compare);

    (void) std::initializer_list<int>{
        (void(sort_simultaneously_impl::helper(N, ord, args)), 0)...};
  }
}  // namespace haar_lib

Back to top page