結果

問題 No.457 (^^*)
ユーザー Ricky_pon
提出日時 2021-09-08 11:17:00
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 2,421 bytes
コンパイル時間 1,942 ms
コンパイル使用メモリ 197,796 KB
最終ジャッジ日時 2025-01-24 08:51:04
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

#include <atcoder/segtree>

#include <algorithm>
#include <cassert>
#include <vector>

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

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

template <class T>
T div_floor(T a, T b) {
    if (b < 0) a *= -1, b *= -1;
    return a >= 0 ? a / b : (a + 1) / b - 1;
}

template <class T>
T div_ceil(T a, T b) {
    if (b < 0) a *= -1, b *= -1;
    return a > 0 ? (a - 1) / b + 1 : a / b;
}

template <typename T>
struct CoordComp {
    std::vector<T> v;
    bool sorted;

    CoordComp() : sorted(false) {}

    int size() { return v.size(); }

    void add(T x) { v.push_back(x); }

    void build() {
        std::sort(v.begin(), v.end());

        v.erase(std::unique(v.begin(), v.end()), v.end());
        sorted = true;
    }

    int get_idx(T x) {
        assert(sorted);
        return lower_bound(v.begin(), v.end(), x) - v.begin();
    }

    T &operator[](int i) { return v[i]; }
};


#define For(i, a, b) for (int i = (int)(a); (i) < (int)(b); ++(i))
#define rFor(i, a, b) for (int i = (int)(a)-1; (i) >= (int)(b); --(i))
#define rep(i, n) For(i, 0, n)
#define rrep(i, n) rFor(i, n, 0)
#define fi first
#define se second

using namespace std;

using lint = long long;
using pii = pair<int, int>;
using pll = pair<lint, lint>;

using namespace atcoder;

int op(int a, int b) { return a + b; }

int e() { return 0; }

int target;

bool f(int x) { return x < target; }

lint solve(string &s) {
    int n = s.size();
    segtree<int, op, e> st1(n), st2(n), st3(n);  // ^, *, )
    rep(i, n) {
        if (s[i] == '^') st1.set(i, 1);
        if (s[i] == '*') st2.set(i, 1);
        if (s[i] == ')') st3.set(i, 1);
    }

    lint ans = 0;
    rep(i, n) if (s[i] == '(') {
        target = 2;
        int a = st1.max_right<f>(i + 1);
        if (a == n) continue;
        target = 1;
        int b = st2.max_right<f>(a + 1);
        if (b == n) continue;
        ans += st3.prod(b + 1, n);
    }

    return ans;
}

int main() {
    string s;
    cin >> s;
    printf("%lld ", solve(s));
    reverse(s.begin(), s.end());
    rep(i, s.size()) {
        if (s[i] == '(')
            s[i] = ')';
        else if (s[i] == ')')
            s[i] = '(';
    }
    printf("%lld\n", solve(s));
}
0