結果

問題 No.1433 Two color sequence
ユーザー kyon2326
提出日時 2021-03-20 00:06:03
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 157 ms / 2,000 ms
コード長 9,561 bytes
コンパイル時間 3,780 ms
コンパイル使用メモリ 255,264 KB
最終ジャッジ日時 2025-01-19 19:49:27
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
#include <bits/extc++.h>
#if __has_include(<atcoder/internal_math>)
    #include <atcoder/internal_math>
    #include <atcoder/internal_type_traits>
#endif
using namespace std;
/*
#include <atcoder/all>
using namespace atcoder;
*/
/*
#include <boost/multiprecision/cpp_int.hpp>
#include <boost/multiprecision/cpp_dec_float.hpp>
using bll = boost::multiprecision::cpp_int;
using bdouble = boost::multiprecision::number<boost::multiprecision::cpp_dec_float<100>>;
using namespace boost::multiprecision;
*/
#if defined(LOCAL_TEST) || defined(LOCAL_DEV)
    #define BOOST_STACKTRACE_USE_ADDR2LINE
    #define BOOST_STACKTRACE_ADDR2LINE_LOCATION /usr/local/opt/binutils/bin/addr2line
    #define _GNU_SOURCE 1
    #include <boost/stacktrace.hpp>
#endif
#ifdef LOCAL_TEST
    namespace std {
        template<typename T> class dvector : public std::vector<T> {
        public:
            dvector() : std::vector<T>() {}
            explicit dvector(size_t n, const T& value = T()) : std::vector<T>(n, value) {}
            dvector(const std::vector<T>& v) : std::vector<T>(v) {}
            dvector(const std::initializer_list<T> il) : std::vector<T>(il) {}
            dvector(const std::string::iterator first, const std::string::iterator last) : std::vector<T>(first, last) {}
            dvector(const typename std::vector<T>::iterator first, const typename std::vector<T>::iterator last) : std::vector<T>(first, last) {}
            dvector(const typename std::vector<T>::reverse_iterator first, const typename std::vector<T>::reverse_iterator last) : std::vector<T
                >(first, last) {}
            dvector(const typename std::vector<T>::const_iterator first, const typename std::vector<T>::const_iterator last) : std::vector<T>(first, 
                last) {}
            dvector(const typename std::vector<T>::const_reverse_iterator first, const typename std::vector<T>::const_reverse_iterator last) : std
                ::vector<T>(first, last) {}
            template <typename T_ = T, typename std::enable_if_t<std::is_same_v<T_, bool>, std::nullptr_t> = nullptr>
            std::vector<bool>::reference operator[](size_t n) {
                if (this->size() <= n) { std::cerr << boost::stacktrace::stacktrace() << '\n' << "vector::_M_range_check: __n (which is " << n << ") 
                    >= this->size() (which is " << this->size() << ")" << '\n'; } return this->at(n);
            }
            template <typename T_ = T, typename std::enable_if_t<std::is_same_v<T_, bool>, std::nullptr_t> = nullptr>
            const T_ operator[](size_t n) const {
                if (this->size() <= n) { std::cerr << boost::stacktrace::stacktrace() << '\n' << "vector::_M_range_check: __n (which is " << n << ") 
                    >= this->size() (which is " << this->size() << ")" << '\n'; } return this->at(n);
            }
            template <typename T_ = T, typename std::enable_if_t<!std::is_same_v<T_, bool>, std::nullptr_t> = nullptr>
            T_& operator[](size_t n) {
                if (this->size() <= n) { std::cerr << boost::stacktrace::stacktrace() << '\n' << "vector::_M_range_check: __n (which is " << n << ") 
                    >= this->size() (which is " << this->size() << ")" << '\n'; } return this->at(n);
            }
            template <typename T_ = T, typename std::enable_if_t<!std::is_same_v<T_, bool>, std::nullptr_t> = nullptr>
            const T_& operator[](size_t n) const {
                if (this->size() <= n) { std::cerr << boost::stacktrace::stacktrace() << '\n' << "vector::_M_range_check: __n (which is " << n << ") 
                    >= this->size() (which is " << this->size() << ")" << '\n'; } return this->at(n);
            }
        };
        template<typename T, typename Compare = less<T>, class Allocator = allocator<T>> class dmultiset : public std::multiset<T,Compare,Allocator> 
            {
        public:
            dmultiset() : std::multiset<T,Compare,Allocator>() {}
            dmultiset(const std::multiset<T,Compare,Allocator>& m) : std::multiset<T,Compare,Allocator>(m) {}
            dmultiset(const std::initializer_list<T> il) : std::multiset<T,Compare,Allocator>(il) {}
            dmultiset(const Compare& comp) : std::multiset<T,Compare,Allocator>(comp) {}
            const typename std::multiset<T,Compare,Allocator>::iterator erase(const typename std::multiset<T,Compare,Allocator>::iterator it) {
                return std::multiset<T,Compare,Allocator>::erase(it);
            }
            size_t erase([[maybe_unused]] const T& x) {
                std::cerr << boost::stacktrace::stacktrace() << '\n'; assert(false);
            }
            size_t erase_all_elements(const T& x) {
                return std::multiset<T,Compare,Allocator>::erase(x);
            }
        };
    }
    #define vector dvector
    #define multiset dmultiset
    class SIGFPE_exception : std::exception {};
    class SIGSEGV_exception : std::exception {};
    void catch_SIGFPE([[maybe_unused]] int e) { std::cerr << boost::stacktrace::stacktrace() << '\n'; throw SIGFPE_exception(); }
    void catch_SIGSEGV([[maybe_unused]] int e) { std::cerr << boost::stacktrace::stacktrace() << '\n'; throw SIGSEGV_exception(); }
    signed convertedmain();
    signed main() { signal(SIGFPE, catch_SIGFPE); signal(SIGSEGV, catch_SIGSEGV); return convertedmain(); }
    #define main() convertedmain()
#else
    #define erase_all_elements erase
#endif
#ifdef LOCAL_DEV
    template<typename T1, typename T2> std::ostream& operator<<(std::ostream& s, const std::pair<T1, T2>& p) {
        return s << "(" << p.first << ", " << p.second << ")"; }
    template <typename T, size_t N> std::ostream& operator<<(std::ostream& s, const std::array<T, N>& a) {
        s << "{ "; for (size_t i = 0; i < N; ++i){ s << a[i] << "\t"; } s << "}"; return s; }
    template<typename T> std::ostream& operator<<(std::ostream& s, const std::set<T>& se) {
        s << "{ "; for (auto itr = se.begin(); itr != se.end(); ++itr){ s << (*itr) << "\t"; } s << "}"; return s; }
    template<typename T> std::ostream& operator<<(std::ostream& s, const std::multiset<T>& se) {
        s << "{ "; for (auto itr = se.begin(); itr != se.end(); ++itr){ s << (*itr) << "\t"; } s << "}"; return s; }
    template<typename T1, typename T2> std::ostream& operator<<(std::ostream& s, const std::map<T1, T2>& m) {
        s << "{\n"; for (auto itr = m.begin(); itr != m.end(); ++itr){ s << "\t" << (*itr).first << " : " << (*itr).second << "\n"; } s << "}"; 
            return s; }
    template<typename T> std::ostream& operator<<(std::ostream& s, const std::deque<T>& v) {
        for (size_t i = 0; i < v.size(); ++i){ s << v[i]; if (i < v.size() - 1) s << "\t"; } return s; }
    template<typename T> std::ostream& operator<<(std::ostream& s, const std::vector<T>& v) {
        for (size_t i = 0; i < v.size(); ++i){ s << v[i]; if (i < v.size() - 1) s << "\t"; } return s; }
    template<typename T> std::ostream& operator<<(std::ostream& s, const std::vector<std::vector<T>>& vv) {
        s << "\\\n"; for (size_t i = 0; i < vv.size(); ++i){ s << vv[i] << "\n"; } return s; }
    void debug_impl() { std::cerr << '\n'; }
    template<typename Head, typename... Tail> void debug_impl(Head head, Tail... tail) { std::cerr << " " << head << (sizeof...(tail) ? "," : ""); 
        debug_impl(tail...); }
    #define debug(...) do { std::cerr << ":" << __LINE__ << " (" << #__VA_ARGS__ << ") ="; debug_impl(__VA_ARGS__); } while (false)
    constexpr inline long long prodlocal([[maybe_unused]] long long prod, [[maybe_unused]] long long local) { return local; }
#else
    #define debug(...) do {} while (false)
    constexpr inline long long prodlocal([[maybe_unused]] long long prod, [[maybe_unused]] long long local) { return prod; }
#endif
//#define int long long
using ll = long long;
//INT_MAX = (1<<31)-1 = 2147483647, INT64_MAX = (1LL<<63)-1 = 9223372036854775807
constexpr ll INF = numeric_limits<ll>::max() == INT_MAX ? (ll)1e9 + 7 : (ll)1e18;
//constexpr ll MOD = (ll)1e9 + 7; //primitive root = 5
//constexpr ll MOD = 998244353; //primitive root = 3
constexpr double EPS = 1e-9;
constexpr ll dx[4] = {1, 0, -1, 0};
constexpr ll dy[4] = {0, 1, 0, -1};
constexpr ll dx8[8] = {1, 0, -1, 0, 1, 1, -1, -1};
constexpr ll dy8[8] = {0, 1, 0, -1, 1, -1, 1, -1};
#define rep(i, n)   for(ll i=0, i##_length=(n); i< i##_length; ++i)
#define repeq(i, n) for(ll i=1, i##_length=(n); i<=i##_length; ++i)
#define rrep(i, n)   for(ll i=(n)-1; i>=0; --i)
#define rrepeq(i, n) for(ll i=(n)  ; i>=1; --i)
#define all(v) (v).begin(), (v).end()
#define rall(v) (v).rbegin(), (v).rend()
void p() { std::cout << '\n'; }
template<typename Head, typename... Tail> void p(Head head, Tail... tail) { std::cout << head << (sizeof...(tail) ? " " : ""); p(tail...); }
template<typename T> inline void pv(std::vector<T>& v) { for(ll i=0, N=v.size(); i<N; i++) std::cout << v[i] << " \n"[i==N-1]; }
template<typename T> inline bool chmax(T& a, T b) { return a < b && (a = b, true); }
template<typename T> inline bool chmin(T& a, T b) { return a > b && (a = b, true); }
template<typename T> inline void uniq(std::vector<T>& v) { v.erase(std::unique(v.begin(), v.end()), v.end()); }
template<typename T> inline ll sz(T& v) { return v.size(); }
/*-----8<-----template-----8<-----*/
/*-----8<-----library-----8<-----*/
void solve() {
    ll N;
    string S;
    cin >> N >> S;
    vector<ll> a(N);
    rep(i, N) cin >> a[i];
    vector<ll> ruiR(a.size()+1, 0);
    rep(i, a.size()) { ruiR[i+1] = ruiR[i] + (S[i]=='R' ? a[i] : 0); }
    vector<ll> ruiB(a.size()+1, 0);
    rep(i, a.size()) { ruiB[i+1] = ruiB[i] + (S[i]=='B' ? a[i] : 0); }
    set<ll> sRB, sBR;
    sRB.insert(0);
    sBR.insert(0);
    ll ans = -INF;
    repeq(i, N) {
        ll rmax = *prev(sRB.end());
        ll rmin = *sRB.begin();
        ll bmax = *prev(sBR.end());
        ll bmin = *sBR.begin();
        chmax(ans, abs(ruiB[i]-ruiR[i] - bmax));
        chmax(ans, abs(ruiB[i]-ruiR[i] - bmin));
        chmax(ans, abs(ruiR[i]-ruiB[i] - rmax));
        chmax(ans, abs(ruiR[i]-ruiB[i] - rmin));
        sRB.insert(ruiR[i]-ruiB[i]);
        sBR.insert(ruiB[i]-ruiR[i]);
    }
    p(ans);
}
signed main() {
    std::cin.tie(nullptr);
    std::ios::sync_with_stdio(false);
    //ll Q; cin >> Q; while(Q--)solve();
    solve();
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0