#859230 (C++17) No.2268 NGワード回避

提出ソース

結果

問題	No.2268 NGワード回避
ユーザー	cutmdo
提出日時	2023-04-14 21:47:38
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	198 ms / 2,000 ms
コード長	5,585 bytes
コンパイル時間	3,534 ms
コンパイル使用メモリ	259,748 KB
実行使用メモリ	112,896 KB
最終ジャッジ日時	2024-04-27 04:42:40
合計ジャッジ時間	5,411 ms
ジャッジサーバーID （参考情報）	judge5 / judge3

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 6,812 KB
testcase_01	AC	2 ms 6,940 KB
testcase_02	AC	2 ms 6,940 KB
testcase_03	AC	1 ms 6,944 KB
testcase_04	AC	2 ms 6,940 KB
testcase_05	AC	1 ms 6,940 KB
testcase_06	AC	1 ms 6,944 KB
testcase_07	AC	2 ms 6,940 KB
testcase_08	AC	3 ms 6,944 KB
testcase_09	AC	2 ms 6,940 KB
testcase_10	AC	3 ms 6,940 KB
testcase_11	AC	2 ms 6,944 KB
testcase_12	AC	3 ms 6,940 KB
testcase_13	AC	3 ms 6,944 KB
testcase_14	AC	3 ms 6,944 KB
testcase_15	AC	2 ms 6,940 KB
testcase_16	AC	3 ms 6,944 KB
testcase_17	AC	3 ms 6,940 KB
testcase_18	AC	3 ms 6,944 KB
testcase_19	AC	3 ms 6,940 KB
testcase_20	AC	3 ms 6,940 KB
testcase_21	AC	3 ms 6,944 KB
testcase_22	AC	3 ms 6,944 KB
testcase_23	AC	3 ms 6,940 KB
testcase_24	AC	3 ms 6,940 KB
testcase_25	AC	3 ms 6,940 KB
testcase_26	AC	3 ms 6,940 KB
testcase_27	AC	3 ms 6,944 KB
testcase_28	AC	3 ms 6,940 KB
testcase_29	AC	3 ms 6,940 KB
testcase_30	AC	3 ms 6,940 KB
testcase_31	AC	3 ms 6,944 KB
testcase_32	AC	3 ms 6,940 KB
testcase_33	AC	3 ms 6,940 KB
testcase_34	AC	3 ms 6,940 KB
testcase_35	AC	3 ms 6,944 KB
testcase_36	AC	3 ms 6,944 KB
testcase_37	AC	3 ms 6,944 KB
testcase_38	AC	3 ms 6,944 KB
testcase_39	AC	3 ms 6,940 KB
testcase_40	AC	3 ms 6,944 KB
testcase_41	AC	2 ms 6,940 KB
testcase_42	AC	3 ms 6,940 KB
testcase_43	AC	3 ms 6,940 KB
testcase_44	AC	4 ms 6,940 KB
testcase_45	AC	3 ms 6,944 KB
testcase_46	AC	3 ms 6,940 KB
testcase_47	AC	198 ms 112,768 KB
testcase_48	AC	198 ms 112,768 KB
testcase_49	AC	195 ms 112,768 KB
testcase_50	AC	186 ms 112,896 KB
testcase_51	AC	187 ms 112,896 KB

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

ソースコード

raw source code

//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,avx512f")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <iostream>
#include <iomanip>
#include <string>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#include <list>
#include <stack>
#include <queue>
#include <bitset>
#include <numeric>
#include <cassert>
#include <memory>
#include <random>
#include <functional>
#include <complex>
#include <immintrin.h>
#include <stdexcept>
#ifdef DEBUG
#include "./CompetitiveProgrammingCpp/Utils/debug.hpp"
#include "./CompetitiveProgrammingCpp/Utils/Timer.hpp"
#include "./CompetitiveProgrammingCpp/Utils/sample.hpp"
#else
#define dump(...)
template<class T>constexpr inline auto d_val(T a, T b) { return a; }
#endif

/* macro */
#define FOR(i, b, e) for(ll i = (ll)(b); i < (ll)(e); ++i)
#define RFOR(i, b, e) for(ll i = (ll)((e)-1); i >= (ll)(b); --i)
#define REP(i, n) FOR(i, 0, (n))
#define RREP(i, n) RFOR(i, 0, (n))
#define REPC(x,c) for(const auto& x:(c))
#define REPI2(it,b,e) for(auto it = (b); it != (e); ++it)
#define REPI(it,c) REPI2(it, (c).begin(), (c).end())
#define RREPI(it,c) REPI2(it, (c).rbegin(), (c).rend())
#define REPI_ERACE2(it, b, e) for(auto it = (b); it != (e);)
#define REPI_ERACE(it, c) REPI_ERACE2(it, (c).begin(), (c).end())
#define ALL(x) (x).begin(),(x).end()
#define cauto const auto&
/* macro func */
template<class T>
inline auto sort(T& t) { std::sort(ALL(t)); }
template<class T>
inline auto rsort(T& t) { std::sort((t).rbegin(), (t).rend()); }
template<class T>
inline auto unique(T& t) { (t).erase(unique((t).begin(), (t).end()), (t).end()); }
template<class T, class S>
inline auto chmax(T& t, const S& s) { if(s > t) { t = s; return true; } return false; }
template<class T, class S>
inline auto chmin(T& t, const S& s) { if(s < t) { t = s; return true; } return false; }
inline auto BR() { std::cout << "\n"; }

/* type define */
using ll = long long;
using VS = std::vector<std::string>;
using VL = std::vector<long long>;
using VVL = std::vector<VL>;
using VVVL = std::vector<VVL>;
using VVVVL = std::vector<VVVL>;
using VVVVVL = std::vector<VVVVL>;
using VD = std::vector<double>;
template<class T>
using V = std::vector<T>;
template<class T = ll, class U = T>
using P = std::pair<T, U>;
using PAIR = P<ll>;

/* using std */
using std::cout;
constexpr char endl = '\n';
using std::cin;
using std::pair;
using std::string;
using std::stack;
using std::queue;
using std::deque;
using std::vector;
using std::list;
using std::map;
using std::unordered_map;
using std::multimap;
using std::unordered_multimap;
using std::set;
using std::unordered_set;
using std::unordered_multiset;
using std::multiset;
using std::bitset;
using std::priority_queue;

/* Initial processing  */
struct Preprocessing { Preprocessing() { std::cin.tie(0); std::ios::sync_with_stdio(0); }; }_Preprocessing;

/* define hash */
namespace std {
template <>	class hash<std::pair<ll, ll>> { public:	size_t operator()(const std::pair<ll, ll>& x) const { return hash<ll>()(1000000000 * x.first + x.second); } };
}

/* input */
template<class T> std::istream& operator >> (std::istream& is, vector<T>& vec) { for(T& x : vec) is >> x; return is; }

/* constant value */
// constexpr ll MOD = 1000000007;
constexpr ll MOD = 998244353;

//=============================================================================================


constexpr auto nullLambda = [](int n) {};
template<class Val = bool, Val ignore = false>
class TrieTree {
public:
    Val m_val;
    std::vector<std::shared_ptr<TrieTree>> m_next;
    //static constexpr auto nullLambda = [](ll n) {}; c++17

    auto emplace(const std::vector<int>& vec, Val val, int it) {
        m_val += val;
        if(it == vec.size()) { return; }
        if(!m_next[vec[it]]) {
            m_next[vec[it]] = std::make_shared<TrieTree>();
        }
        m_next[vec[it]]->emplace(vec, val, it + 1);
    }

    template<class Lambda>
    auto find(const std::vector<int>& vec, int it, const Lambda& lambda)const {
        if(m_val != ignore) { lambda(m_val); }
        if(it == vec.size()) { return m_val; }
        if(!m_next[vec[it]]) { return ignore; }
        return m_next[vec[it]]->find(vec, it + 1, lambda);
    }

    TrieTree() : TrieTree(ignore) {}
    TrieTree(Val val) :m_val(val), m_next(std::vector<std::shared_ptr<TrieTree>>(2)) {}

    auto emplace(const std::vector<int>& vec, Val val) { return emplace(vec, val, 0); }

    template<class Lambda = decltype(nullLambda)>
    auto find(const std::vector<int>& vec, const Lambda& lambda = nullLambda) { return find(vec, 0, lambda); }
};

auto solve(ll n, const VS& s) {
    auto tt = std::make_shared<TrieTree<ll, 0>>();
    REPC(t, s) {
        V<int> v; v.reserve(n);
        REPC(c, t) { v.emplace_back((c == 'a' ? 0 : 1)); }
        tt->emplace(v, 1);
    }
    ll d = n;
    string ans;
    REP(_, n) {
        if(!tt->m_next[0]) { while(ans.size() < n) { ans += 'a'; }break; }
        if(!tt->m_next[1]) { while(ans.size() < n) { ans += 'b'; }break; }
        if(tt->m_next[0]->m_val != n) {
            ans += 'a';
            d -= tt->m_next[1]->m_val;
            tt = tt->m_next[0];
        } else {
            ans += 'b';
            d -= tt->m_next[0]->m_val;
            tt = tt->m_next[1];
        }
    }
    return ans;
}

signed main() {
    ll n;
    cin >> n;
    VS s(n);
    cin >> s;
    auto ans = solve(n, s);
    cout << ans << endl;
}

yukicoder

結果

テストケース

ソースコード