others/compress.cpp

Back to top page

• category: others
• View this file on GitHub
  • Last commit date: 2020-09-08 14:10:34+09:00

Code

template <typename T>
class Compress {
    vector<T> xs;

public:
    Compress() = default;
    explicit Compress(const vector<T> &vs) { add(vs); }
    explicit Compress(const initializer_list<vector<T>> &vs) {
        for (const auto &p : vs)
            add(p);
    }

    void add(const vector<T> &vs) {
        copy(begin(vs), end(vs), back_inserter(xs));
    }

    void add(const T &x) { xs.emplace_back(x); }

    void build() {
        sort(begin(xs), end(xs));
        xs.erase(unique(begin(xs), end(xs)), end(xs));
    }

    vector<int> get(const vector<T> &vs) const {
        vector<int> ret;
        transform(begin(vs), end(vs), back_inserter(ret), [&](const T &x) {
            return lower_bound(begin(xs), end(xs), x) - begin(xs);
        });
        return ret;
    }

    int get(const T &x) const {
        return lower_bound(begin(xs), end(xs), x) - begin(xs);
    }

    const T &operator[](const int k) const { return xs[k]; }
};

#line 1 "others/compress.cpp"
template <typename T>
class Compress {
    vector<T> xs;

public:
    Compress() = default;
    explicit Compress(const vector<T> &vs) { add(vs); }
    explicit Compress(const initializer_list<vector<T>> &vs) {
        for (const auto &p : vs)
            add(p);
    }

    void add(const vector<T> &vs) {
        copy(begin(vs), end(vs), back_inserter(xs));
    }

    void add(const T &x) { xs.emplace_back(x); }

    void build() {
        sort(begin(xs), end(xs));
        xs.erase(unique(begin(xs), end(xs)), end(xs));
    }

    vector<int> get(const vector<T> &vs) const {
        vector<int> ret;
        transform(begin(vs), end(vs), back_inserter(ret), [&](const T &x) {
            return lower_bound(begin(xs), end(xs), x) - begin(xs);
        });
        return ret;
    }

    int get(const T &x) const {
        return lower_bound(begin(xs), end(xs), x) - begin(xs);
    }

    const T &operator[](const int k) const { return xs[k]; }
};

Back to top page