#698210 (C++17) No.1677 mæx

提出ソース

結果

問題	No.1677 mæx
ユーザー	ebi_fly
提出日時	2021-09-10 22:11:34
言語	C++17 (gcc 13.2.0 + boost 1.83.0)
結果	AC
実行時間	8 ms / 2,000 ms
コード長	6,260 bytes
コンパイル時間	3,304 ms
コンパイル使用メモリ	194,564 KB
実行使用メモリ	7,440 KB
最終ジャッジ日時	2023-09-02 18:04:27
合計ジャッジ時間	4,515 ms
ジャッジサーバーID （参考情報）	judge12 / judge11

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 4,372 KB
testcase_01	AC	2 ms 4,368 KB
testcase_02	AC	1 ms 4,368 KB
testcase_03	AC	2 ms 4,372 KB
testcase_04	AC	8 ms 6,260 KB
testcase_05	AC	8 ms 6,192 KB
testcase_06	AC	7 ms 6,188 KB
testcase_07	AC	7 ms 6,268 KB
testcase_08	AC	8 ms 6,468 KB
testcase_09	AC	7 ms 6,516 KB
testcase_10	AC	8 ms 6,492 KB
testcase_11	AC	8 ms 6,360 KB
testcase_12	AC	8 ms 6,456 KB
testcase_13	AC	7 ms 6,220 KB
testcase_14	AC	7 ms 6,208 KB
testcase_15	AC	8 ms 6,216 KB
testcase_16	AC	8 ms 6,252 KB
testcase_17	AC	7 ms 6,292 KB
testcase_18	AC	7 ms 6,212 KB
testcase_19	AC	1 ms 4,372 KB
testcase_20	AC	2 ms 4,368 KB
testcase_21	AC	8 ms 7,440 KB

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

ソースコード

raw source code

#include <iostream>
#include <algorithm>
#include <utility>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <queue>
#include <map>
#include <math.h>
#include <string.h>
#include <iomanip>
#include <numeric>
#include <cstdlib>
#include <cstdint>
#include <cmath>
#include <functional>
#include <limits>
#include <cassert>
#include <bitset>
#include <chrono>
#include <random>
#include <memory>
#include <atcoder/all>

/* macro */
 
#define rep(i, a, n) for (int i = (int)(a); i < (int)(n); i++)
#define rrep(i, a, n) for (int i = ((int)(n-1)); i >= (int)(a); i--)
#define Rep(i, a, n) for (i64 i = (i64)(a); i< (i64)(n); i++)
#define RRep(i, a, n) for (i64 i = ((i64)(n-i64(1))); i>=(i64)(a); i--)
#define all(v) (v).begin(),(v).end()
#define rall(v) (v).rbegin(),(v).rend()
#define Bit(n) (1LL<<(n))

/* macro end */

/* template */

namespace ebi {

#ifdef LOCAL
#define debug(...)                                                    \
    std::cerr << "LINE: " << __LINE__ << "  [" << #__VA_ARGS__ << "]:", \
        debug_out(__VA_ARGS__)
#else
#define debug(...)
#endif
 
void debug_out() { std::cerr << std::endl; }
 
template <typename Head, typename... Tail>
void debug_out(Head h, Tail... t) {
    std::cerr << " " << h;
    if (sizeof...(t) > 0) std::cout << " :";
    debug_out(t...);
}
 
template <typename T1, typename T2>
std::ostream& operator<<(std::ostream& os, std::pair<T1, T2> pa) {
    return os << pa.first << " " << pa.second;
}
 
template <typename T>
std::ostream& operator<<(std::ostream& os, std::vector<T> vec) {
    for (std::size_t i = 0; i < vec.size(); i++) os << vec[i] << (i + 1 == vec.size() ? "" : " ");
    return os;
}

std::ostream& operator<<(std::ostream& os, atcoder::modint1000000007 a){
    return os << a.val();
}

std::ostream& operator<<(std::ostream& os, atcoder::modint998244353 a){
    return os << a.val();
}

using size_t = std::size_t;
using i32 = std::int32_t;
using u32 = std::uint32_t;
using i64 = long long;
using u64 = unsigned long long;

template<class T> void fill(std::vector<T> &v) { for(T &e: v) std::cin >> e; }

template<class T>
struct Edge {
    int to;
    T cost;
    Edge(int to, T cost) : to(to), cost(cost) { }
};

template<class T>
std::ostream& operator<<(std::ostream& os, Edge<T> e) {
    return os << "[" << e.to << ", " << e.cost << "] ";
}

template<class T>
struct Graph : std::vector<std::vector<Edge<T>>> {
    using std::vector<std::vector<Edge<T>>>::vector;
    void add_edge(int u, int v, T w, bool directed = false) {
        (*this)[u].emplace_back(v, w);
        if(directed) return; 
        (*this)[v].emplace_back(u, w);
    }
};

struct graph : std::vector<std::vector<int>> {
    using std::vector<std::vector<int>>::vector;
    void add_edge(int u, int v, bool directed = false) {
        (*this)[u].emplace_back(v);
        if(directed) return;
        (*this)[v].emplace_back(u);
    }
};

template <class T>
inline bool chmin(T &a, T b) {
    if (a > b){
        a = b;
        return true;
    }
    return false;
}
 
template <class T>
inline bool chmax(T &a, T b) {
    if (a < b){
        a = b;
        return true;
    }
    return false;
}

template <class T>
T gcd(T a, T b){
    if( b==0 ) return a;
    else return gcd(b, a%b);
}

template <class T>
T lcm(T a, T b) {
    return (a*b)/gcd(a,b);
}

template<class T>
T pow(T x, i64 n) {
    T res = 1;
    while(n>0){
        if(n&1) res = res *x;
        x = x*x;
        n>>=1;
    }
    return res;
}

template<class T>
T mod_pow(T x, i64 n, i64 mod) {
    T res = 1;
    while(n > 0) {
        if(n & 1) res = (res * x)%mod;
        x = (x*x)%mod;
        n >>= 1;
    }
    return res;
}

template<class T>
T scan() {
    T val;
    std::cin >> val;
    return val;
}
 
constexpr i64 LNF = std::numeric_limits<i64>::max()/2;

constexpr int INF = std::numeric_limits<int>::max()/2;

const long double PI = acos(-1);

const std::vector<int> dy = {1,0,-1,0,1,1,-1,-1};
const std::vector<int> dx = {0,1,0,-1,1,-1,1,-1};

}

/* template end */

namespace ebi {

using pii = std::pair<int,int>;

using ld = long double;

using mint = atcoder::modint998244353;

void main_() {
    std::string s;
    int k;
    std::cin >> s >> k;
    int idx = 0;
    auto mex = [](int a, int b) -> int {
        if(a > b) std::swap(a,b);
        if(0 < a) {
            return 0;
        }  
        else if(a == 0 && b == 1) {
            return 2;
        }
        else {
            return 1;
        }
    };
    auto nxt = [&](char c) -> void {
        assert(s[idx] == c);
        idx++;
        return;
    };
    auto f = [&](auto &&self) -> std::vector<mint> {
        std::vector<mint> ret(3,0);
        if(s[idx] != 'm') {
            if(s[idx] == '?') {
                rep(i,0,3) {
                    ret[i]++;
                }
                nxt('?');
                return ret;
            }
            ret[s[idx]-'0']++;
            nxt(s[idx]);
            return ret;
        }
        int flag = 0;
        nxt('m');
        if(s[idx] == 'e') {
            nxt('e');
            flag = 0;
        }
        else if(s[idx] == 'a') {
            nxt('a');
            flag = 1;
        }
        else {
            nxt('?');
            flag = 2;
        }
        nxt('x');
        nxt('(');
        auto lhs = self(self);
        nxt(',');
        auto rhs = self(self);
        nxt(')');
        if(flag == 0) {
            rep(i,0,3) {
                rep(j,0,3) {
                    ret[mex(i,j)] += lhs[i] * rhs[j];
                }
            }
        }
        else if(flag == 1) {
            rep(i,0,3) {
                rep(j,0,3) {
                    ret[std::max(i, j)] += lhs[i] * rhs[j];
                }
            }
        }
        else {
            rep(i,0,3) {
                rep(j,0,3) {
                    ret[mex(i,j)] += lhs[i] * rhs[j];
                    ret[std::max(i,j)] += lhs[i] * rhs[j];
                }
            }
        }
        return ret;
    };
    auto ans = f(f);
    std::cout << ans[k] << '\n'; 

}

}

int main() {
    std::cout << std::fixed << std::setprecision(15);
    std::cin.tie(nullptr);
    std::ios::sync_with_stdio(false);
    int t = 1;
    while(t--) {
        ebi::main_();
    }
    return 0;
}

yukicoder

結果

テストケース

ソースコード