#591593 (C++17) No.1313 N言っちゃダメゲーム (4)

提出ソース

結果

問題	No.1313 N言っちゃダメゲーム (4)
ユーザー	Pachicobue
提出日時	2020-12-10 13:17:15
言語	C++17 (gcc 13.2.0 + boost 1.83.0)
結果	AC
実行時間	59 ms / 2,000 ms
コード長	6,716 bytes
コンパイル時間	2,124 ms
コンパイル使用メモリ	211,248 KB
実行使用メモリ	4,348 KB
最終ジャッジ日時	2023-10-20 00:42:19
合計ジャッジ時間	4,133 ms
ジャッジサーバーID （参考情報）	judge14 / judge12

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 4,348 KB
testcase_01	AC	2 ms 4,348 KB
testcase_02	AC	2 ms 4,348 KB
testcase_03	AC	2 ms 4,348 KB
testcase_04	AC	2 ms 4,348 KB
testcase_05	AC	1 ms 4,348 KB
testcase_06	AC	2 ms 4,348 KB
testcase_07	AC	2 ms 4,348 KB
testcase_08	AC	2 ms 4,348 KB
testcase_09	AC	2 ms 4,348 KB
testcase_10	AC	2 ms 4,348 KB
testcase_11	AC	2 ms 4,348 KB
testcase_12	AC	2 ms 4,348 KB
testcase_13	AC	40 ms 4,348 KB
testcase_14	AC	40 ms 4,348 KB
testcase_15	AC	59 ms 4,348 KB
testcase_16	AC	59 ms 4,348 KB
testcase_17	AC	41 ms 4,348 KB
testcase_18	AC	37 ms 4,348 KB
testcase_19	AC	34 ms 4,348 KB
testcase_20	AC	7 ms 4,348 KB
testcase_21	AC	25 ms 4,348 KB
testcase_22	AC	17 ms 4,348 KB
testcase_23	AC	25 ms 4,348 KB
testcase_24	AC	5 ms 4,348 KB
testcase_25	AC	32 ms 4,348 KB
testcase_26	AC	2 ms 4,348 KB
testcase_27	AC	20 ms 4,348 KB
testcase_28	AC	48 ms 4,348 KB
testcase_29	AC	50 ms 4,348 KB
testcase_30	AC	50 ms 4,348 KB
testcase_31	AC	48 ms 4,348 KB
testcase_32	AC	50 ms 4,348 KB
testcase_33	AC	39 ms 4,348 KB
testcase_34	AC	39 ms 4,348 KB
testcase_35	AC	39 ms 4,348 KB
testcase_36	AC	39 ms 4,348 KB
testcase_37	AC	39 ms 4,348 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
using ll = long long;
using uint = unsigned int;
using ull = unsigned long long;
using ld = long double;
template<typename T> using max_heap = std::priority_queue<T>;
template<typename T> using min_heap = std::priority_queue<T, std::vector<T>, std::greater<T>>;
constexpr int popcount(const ull v) { return v ? __builtin_popcountll(v) : 0; }
constexpr int log2p1(const ull v) { return v ? 64 - __builtin_clzll(v) : 0; }
constexpr int lsbp1(const ull v) { return __builtin_ffsll(v); }
constexpr int clog(const ull v) { return v ? log2p1(v - 1) : 0; }
constexpr ull ceil2(const ull v) { return 1ULL << clog(v); }
constexpr ull floor2(const ull v) { return v ? (1ULL << (log2p1(v) - 1)) : 0ULL; }
constexpr bool btest(const ull mask, const int ind) { return (mask >> ind) & 1ULL; }
template<typename T> void bset(T& mask, const int ind) { mask |= ((T)1 << ind); }
template<typename T> void breset(T& mask, const int ind) { mask &= ~((T)1 << ind); }
template<typename T> void bflip(T& mask, const int ind) { mask ^= ((T)1 << ind); }
template<typename T> void bset(T& mask, const int ind, const bool b) { (b ? bset(mask, ind) : breset(mask, ind)); }
template<typename T> bool chmin(T& a, const T& b) { return (a > b ? a = b, true : false); }
template<typename T> bool chmax(T& a, const T& b) { return (a < b ? a = b, true : false); }
template<typename T> constexpr T inf_v = std::numeric_limits<T>::max() / 4;
template<typename Real> constexpr Real pi_v = Real{3.141592653589793238462643383279502884};
template<typename T> constexpr T TEN(const int n) { return n == 0 ? T{1} : TEN<T>(n - 1) * T{10}; }
template<typename F> struct fix : F
{
    fix(F&& f) : F{std::forward<F>(f)} {}
    template<typename... Args> auto operator()(Args&&... args) const { return F::operator()(*this, std::forward<Args>(args)...); }
};
template<typename T, int n, int i = 0>
auto nd_array(int const (&szs)[n], const T x = T{})
{
    if constexpr (i == n) {
        return x;
    } else {
        return std::vector(szs[i], nd_array<T, n, i + 1>(szs, x));
    }
}
class printer
{
public:
    printer(std::ostream& os_ = std::cout) : m_os{os_} { m_os << std::fixed << std::setprecision(15); }
    template<typename... Args> int ln(const Args&... args) { return dump(args...), m_os << '\n', 0; }
    template<typename... Args> int el(const Args&... args) { return dump(args...), m_os << std::endl, 0; }
private:
    template<typename T> void dump(const T& v) { m_os << v; }
    template<typename T> void dump(const std::vector<T>& vs)
    {
        for (int i = 0; i < (int)vs.size(); i++) { m_os << (i ? " " : ""), dump(vs[i]); }
    }
    template<typename T> void dump(const std::vector<std::vector<T>>& vss)
    {
        for (int i = 0; i < (int)vss.size(); i++) { m_os << (0 <= i or i + 1 < (int)vss.size() ? "\n" : ""), dump(vss[i]); }
    }
    template<typename T, typename... Args> int dump(const T& v, const Args&... args) { return dump(v), m_os << ' ', dump(args...), 0; }
    std::ostream& m_os;
};
printer out;
class scanner
{
public:
    scanner(std::istream& is_ = std::cin) : m_is{is_} { m_is.tie(nullptr), std::ios::sync_with_stdio(false); }
    template<typename T> T val()
    {
        T v;
        return m_is >> v, v;
    }
    template<typename T> T val(const T offset) { return val<T>() - offset; }
    template<typename T> std::vector<T> vec(const int n)
    {
        return make_v<T>(n, [this]() { return val<T>(); });
    }
    template<typename T> std::vector<T> vec(const int n, const T offset)
    {
        return make_v<T>(n, [this, offset]() { return val<T>(offset); });
    }
    template<typename T> std::vector<std::vector<T>> vvec(const int n0, const int n1)
    {
        return make_v<std::vector<T>>(n0, [this, n1]() { return vec<T>(n1); });
    }
    template<typename T> std::vector<std::vector<T>> vvec(const int n0, const int n1, const T offset)
    {
        return make_v<std::vector<T>>(n0, [this, n1, offset]() { return vec<T>(n1, offset); });
    }
    template<typename... Args> auto tup() { return std::tuple<std::decay_t<Args>...>{val<Args>()...}; }
    template<typename... Args> auto tup(const Args&... offsets) { return std::tuple<std::decay_t<Args>...>{val<Args>(offsets)...}; }
private:
    template<typename T, typename F>
    std::vector<T> make_v(const int n, F f)
    {
        std::vector<T> ans;
        for (int i = 0; i < n; i++) { ans.push_back(f()); }
        return ans;
    }
    std::istream& m_is;
};
scanner in;
template<typename MergeMonoid>
class segtree
{
    using T = typename MergeMonoid::T;
    static constexpr T e() { return MergeMonoid::e(); }
public:
    segtree(const std::vector<T>& vs) : m_size{(int)vs.size()}, m_half{(int)ceil2(m_size)}, m_vals(m_half << 1, MergeMonoid::e()), m_merge{}
    {
        std::copy(vs.begin(), vs.end(), m_vals.begin() + m_half);
        for (int i = m_half - 1; i >= 1; i--) { up(i); }
    }
    T get(const int i) const { return assert(0 <= i and i < m_size), m_vals[i + m_half]; }
    void set(int i, const T& v)
    {
        assert(0 <= i and i < m_size);
        m_vals[i += m_half] = v;
        while (i >>= 1) { up(i); }
    }
    T fold(int l, int r) const
    {
        assert(0 <= l and l <= r and r <= m_size);
        T vl = e(), vr = e();
        for (l += m_half, r += m_half; l < r; l >>= 1, r >>= 1) {
            if (l & 1) { vl = m_merge(vl, m_vals[l++]); }
            if (r & 1) { vr = m_merge(m_vals[--r], vr); }
        }
        return m_merge(vl, vr);
    }
    friend std::ostream& operator<<(std::ostream& os, const segtree& seg)
    {
        os << "[";
        for (int i = seg.m_half; i < seg.m_half + seg.m_size; i++) { os << seg.m_vals[i] << (i + 1 == seg.m_half + seg.m_size ? "" : ","); }
        return (os << "]");
    }
private:
    void up(const int i) { m_vals[i] = m_merge(m_vals[i << 1], m_vals[i << 1 | 1]); }
    int m_size, m_half;
    std::vector<T> m_vals;
    const MergeMonoid m_merge;
};
int main()
{
    const auto [N, K] = in.tup<int, int>();
    const auto S = "o" + in.val<std::string>();
    struct Monoid
    {
        using T = bool;
        T operator()(const T& x1, const T& x2) const { return x1 & x2; }
        static constexpr T e() { return 1; }
    };
    auto dp = segtree<Monoid>(std::vector<bool>(2 * N + 1, true));
    for (int i = N - 1; i >= 0; i--) {
        if (S[i] == 'x') {
            dp.set(i, 1);
        } else {
            dp.set(i, not dp.fold(i + 1, i + K + 1));
        }
    }
    static_cast<void>(0);
    std::vector<int> ans;
    for (int k = 1; k <= K; k++) {
        if (not dp.get(k)) { ans.push_back(k); }
    }
    if (ans.empty()) {
        out.ln(0);
    } else {
        out.ln(ans);
    }
    return 0;
}

yukicoder

結果

テストケース

ソースコード