Submission #50422531

---

Source Code Expand

#include <bits/stdc++.h>

typedef long long valueType;
typedef std::vector<valueType> ValueVector;
typedef std::vector<ValueVector> ValueMatrix;
typedef std::vector<ValueMatrix> ValueCube;
typedef std::string string;
typedef std::vector<string> StringVector;
typedef std::vector<bool> bitset;
typedef std::vector<bitset> BitMatrix;
typedef std::pair<valueType, valueType> ValuePair;
typedef std::vector<ValuePair> PairVector;
typedef std::vector<PairVector> PairMatrix;
typedef std::tuple<valueType, valueType, valueType> ValueTuple;
typedef std::vector<ValueTuple> TupleVector;
typedef std::vector<TupleVector> TupleMatrix;
typedef std::set<valueType> ValueSet;

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    std::cout.tie(nullptr);

    valueType N;

    std::cin >> N;

    string S;

    std::cin >> S;

    ValueVector A;

    for (valueType i = 1; i < N; ++i) {
        if (S[i] == S[0])
            A.push_back(1);
        else
            A.push_back(-1);
    }

    ValueVector B = A;

    std::partial_sum(A.begin(), A.end(), B.begin());

    valueType Ans = 1;

    for (valueType i = 0; i < N - 1; ++i) {
        if (A[i] == 1)
            continue;

        if (B[i] == 0 || B[i] == -1 || B[i] == -2)
            ++Ans;
    }

    std::cout << Ans << std::endl;

    return 0;
}

Submission Info

Judge Result