forked from mirror/toddcox-faster
128 lines
2.7 KiB
C++
128 lines
2.7 KiB
C++
#pragma once
|
|
|
|
#include <array>
|
|
#include <functional>
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
namespace tc {
|
|
struct Action {
|
|
int from_idx = -1;
|
|
int gen = -1;
|
|
|
|
Action() = default;
|
|
|
|
Action(const Action &) = default;
|
|
|
|
Action(int from_idx, int gen);
|
|
};
|
|
|
|
struct Path {
|
|
std::vector<Action> path;
|
|
|
|
Path() = default;
|
|
|
|
Path(const Path &) = default;
|
|
|
|
void add_row();
|
|
|
|
[[nodiscard]] Action get(int to_idx) const;
|
|
|
|
void put(int from_idx, int gen, int to_idx);
|
|
|
|
template<class T>
|
|
[[nodiscard]] std::vector<T> walk(
|
|
T start,
|
|
std::vector<T> gen_values,
|
|
std::function<T(const T&, const T&)> op
|
|
) const {
|
|
std::vector<T> res(size());
|
|
res[0] = start;
|
|
|
|
for (int i = 1; i < res.size(); ++i) {
|
|
auto action = path[i];
|
|
const auto &from = res[action.from_idx];
|
|
const auto &val = gen_values[action.gen];
|
|
res[i] = op(from, val);
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
|
|
[[nodiscard]] size_t size() const;
|
|
};
|
|
|
|
struct Cosets {
|
|
int ngens;
|
|
std::vector<int> data;
|
|
Path path;
|
|
|
|
Cosets(const Cosets &) = default;
|
|
|
|
explicit Cosets(int ngens);
|
|
|
|
void add_row();
|
|
|
|
void put(int coset, int gen, int target);
|
|
|
|
void put(int idx, int target);
|
|
|
|
[[nodiscard]] int get(int coset, int gen) const;
|
|
|
|
[[nodiscard]] int get(int idx) const;
|
|
|
|
[[nodiscard]] size_t size() const;
|
|
};
|
|
|
|
struct Rel {
|
|
std::array<int, 2> gens;
|
|
int mult;
|
|
|
|
Rel() = default;
|
|
|
|
Rel(const Rel &) = default;
|
|
|
|
Rel(int a, int b, int m);
|
|
|
|
[[nodiscard]] Rel shift(int off) const;
|
|
};
|
|
|
|
struct SubGroup;
|
|
|
|
struct Group {
|
|
const int ngens;
|
|
std::vector<std::vector<int>> _mults;
|
|
std::string name;
|
|
|
|
Group(const Group &) = default;
|
|
|
|
explicit Group(int ngens, const std::vector<Rel> &rels = {}, std::string name = "G");
|
|
|
|
void set(const Rel &r);
|
|
|
|
[[nodiscard]] int get(int a, int b) const;
|
|
|
|
[[nodiscard]] std::vector<Rel> rels() const;
|
|
|
|
[[nodiscard]] SubGroup subgroup(const std::vector<int> &gens) const;
|
|
|
|
[[nodiscard]] Group product(const Group &other) const;
|
|
|
|
[[nodiscard]] Group power(int p) const;
|
|
|
|
[[nodiscard]] Cosets solve(const std::vector<int> &sub_gens = {}) const;
|
|
};
|
|
|
|
struct SubGroup : public Group {
|
|
const std::vector<int> gen_map;
|
|
const Group &parent;
|
|
|
|
SubGroup(const Group &parent, const std::vector<int> &gen_map);
|
|
};
|
|
|
|
Group operator*(const Group &g, const Group &h);
|
|
|
|
Group operator^(const Group &g, int p);
|
|
}
|