#if __INCLUDE_LEVEL__ == 0 #include __BASE_FILE__ namespace { void solve() { int n, k; scan(n, k); std::vector w(n); std::vector v(n); { i64 seed, a, b, m; scan(seed, a, b, m); std::vector f(n * 2); f[0] = seed; for (const int i : rep(1, n * 2)) { f[i] = f[i - 1] * a + b; f[i] %= m; } for (const int i : rep(n)) { w[i] = f[i] % 3 + 1; v[i] = f[i + n] * w[i]; } } std::vector> vs(4); for (const int i : rep(n)) { vs[w[i]].push_back(v[i]); } for (auto& e : vs) { ranges::sort(e, std::greater{}); } i64 ans = 0; for (const int r : rep(3)) { std::array s{}; auto keys = vs[3]; keys.push_back(0); for (int i = r; i < len(vs[1]); i += 3) { i64 tmp = vs[1][i]; if (i + 1 < len(vs[1])) { tmp += vs[1][i + 1]; } if (i + 2 < len(vs[1])) { tmp += vs[1][i + 2]; } keys.push_back(tmp); } ranges::sort(keys); std::apply(LIFT(keys.erase), ranges::unique(keys)); atcoder::fenwick_tree f0(len(keys)); atcoder::fenwick_tree f1(len(keys)); const auto add = [&](i64 x) { const int i = static_cast(ranges::lower_bound(keys, x) - keys.begin()); f0.add(i, 1); f1.add(i, x); }; const auto rm = [&](i64 x) { const int i = static_cast(ranges::lower_bound(keys, x) - keys.begin()); f0.add(i, -1); f1.add(i, -x); }; const auto go = [&](int k) { int ng = -1; int ok = len(keys); while (ng + 1 < ok) { const int mid = std::midpoint(ng, ok); (f0.sum(mid, len(keys)) <= k ? ok : ng) = mid; } return f1.sum(ok, len(keys)); }; for (const i64 e : vs[3]) { add(e); } for (int i = r; i < len(vs[1]); i += 3) { i64 tmp = vs[1][i]; if (i + 1 < len(vs[1])) { tmp += vs[1][i + 1]; } if (i + 2 < len(vs[1])) { tmp += vs[1][i + 2]; } add(tmp); } for (int c2 = r; c2 <= std::min(len(vs[2]), k); c2 += 3) { for (const int c : rep(std::max(c2 - 3, 0), c2)) { if (c < len(vs[2])) { s[2] += vs[2][c]; } if (c < len(vs[1])) { s[1] += vs[1][c]; } } i64 cur = s[1] + s[2]; cur += go(k - c2); chmax(ans, cur); if (const int i = c2; i < len(vs[1])) { i64 tmp = vs[1][i]; if (i + 1 < len(vs[1])) { tmp += vs[1][i + 1]; } if (i + 2 < len(vs[1])) { tmp += vs[1][i + 2]; } rm(tmp); } } } print(ans); } // c3+c2+ceil(max(c1-c2,0)/3) <= k } // namespace int main() { std::ios::sync_with_stdio(false); std::cin.tie(nullptr); solve(); } #else // __INCLUDE_LEVEL__ #include #include template bool chmin(T& x, U&& y) { return y < x && (x = std::forward(y), true); } template bool chmax(T& x, U&& y) { return x < y && (x = std::forward(y), true); } template T inf() { T ret; std::memset(&ret, 0x3f, sizeof(ret)); return ret; } template T inf() { return std::numeric_limits::infinity(); } template concept Range = std::ranges::range && !std::convertible_to; template concept Tuple = std::__is_tuple_like::value && !Range; namespace std { istream& operator>>(istream& is, Range auto&& r) { for (auto&& e : r) { is >> e; } return is; } istream& operator>>(istream& is, Tuple auto&& t) { return apply([&](auto&... xs) -> istream& { return (is >> ... >> xs); }, t); } ostream& operator<<(ostream& os, Range auto&& r) { for (string_view sep = ""; auto&& e : r) { os << exchange(sep, " ") << e; } return os; } ostream& operator<<(ostream& os, Tuple auto&& t) { const auto f = [&](auto&... xs) -> ostream& { [[maybe_unused]] string_view sep = ""; ((os << exchange(sep, " ") << xs), ...); return os; }; return apply(f, t); } } // namespace std #define DEF_INC_OR_DEC(op) \ auto& operator op(Range auto&& r) { \ for (auto&& e : r) { \ op e; \ } \ return r; \ } \ auto& operator op(Tuple auto&& t) { \ std::apply([](auto&... xs) { (op xs, ...); }, t); \ return t; \ } DEF_INC_OR_DEC(++) DEF_INC_OR_DEC(--) #undef DEF_INC_OR_DEC void scan(auto&&... xs) { std::cin >> std::tie(xs...); } void print(auto&&... xs) { std::cout << std::tie(xs...) << '\n'; } #define FWD(...) static_cast(__VA_ARGS__) #define LIFT(...) [&](auto&&... xs) -> decltype(auto) { return (__VA_ARGS__)(FWD(xs)...); } template class fix { public: explicit fix(F f) : f_(std::move(f)) {} decltype(auto) operator()(auto&&... xs) const { return f_(std::ref(*this), FWD(xs)...); } private: F f_; }; template concept LambdaExpr = std::is_placeholder_v> != 0 || std::is_bind_expression_v>; auto operator++(LambdaExpr auto&& x, int) { return std::bind([](auto&& x) -> decltype(auto) { return FWD(x)++; }, FWD(x)); } auto operator--(LambdaExpr auto&& x, int) { return std::bind([](auto&& x) -> decltype(auto) { return FWD(x)--; }, FWD(x)); } #define DEF_UNARY_OP(op) \ auto operator op(LambdaExpr auto&& x) { \ return std::bind([](auto&& x) -> decltype(auto) { return op FWD(x); }, FWD(x)); \ } DEF_UNARY_OP(++) DEF_UNARY_OP(--) DEF_UNARY_OP(+) DEF_UNARY_OP(-) DEF_UNARY_OP(~) DEF_UNARY_OP(!) DEF_UNARY_OP(*) DEF_UNARY_OP(&) #undef DEF_UNARY_OP #define DEF_BINARY_OP(op) \ template \ requires LambdaExpr || LambdaExpr \ auto operator op(T1&& x, T2&& y) { \ return std::bind([](auto&& x, auto&& y) -> decltype(auto) { return FWD(x) op FWD(y); }, \ FWD(x), FWD(y)); \ } DEF_BINARY_OP(+=) DEF_BINARY_OP(-=) DEF_BINARY_OP(*=) DEF_BINARY_OP(/=) DEF_BINARY_OP(%=) DEF_BINARY_OP(^=) DEF_BINARY_OP(&=) DEF_BINARY_OP(|=) DEF_BINARY_OP(<<=) DEF_BINARY_OP(>>=) DEF_BINARY_OP(+) DEF_BINARY_OP(-) DEF_BINARY_OP(*) DEF_BINARY_OP(/) DEF_BINARY_OP(%) DEF_BINARY_OP(^) DEF_BINARY_OP(&) DEF_BINARY_OP(|) DEF_BINARY_OP(<<) DEF_BINARY_OP(>>) DEF_BINARY_OP(==) DEF_BINARY_OP(!=) DEF_BINARY_OP(<) DEF_BINARY_OP(>) DEF_BINARY_OP(<=) DEF_BINARY_OP(>=) DEF_BINARY_OP(&&) DEF_BINARY_OP(||) #undef DEF_BINARY_OP template requires LambdaExpr || LambdaExpr auto at(T1&& x, T2&& y) { return std::bind([](auto&& x, auto&& y) -> decltype(auto) { return FWD(x)[FWD(y)]; }, FWD(x), FWD(y)); } template auto get(LambdaExpr auto&& x) { return std::bind([](auto&& x) -> decltype(auto) { return std::get(FWD(x)); }, FWD(x)); } inline auto rep(int l, int r) { return std::views::iota(std::min(l, r), r); } inline auto rep(int n) { return rep(0, n); } inline auto rep1(int l, int r) { return rep(l, r + 1); } inline auto rep1(int n) { return rep(1, n + 1); } using namespace std::literals; using namespace std::placeholders; namespace ranges = std::ranges; namespace views = std::views; using i64 = std::int64_t; #define len(...) static_cast(ranges::size(__VA_ARGS__)) #endif // __INCLUDE_LEVEL__