leetcode/1525.cpp at master · Yashbhatt7/leetcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
#include<iostream>
#include<vector>
#include<unordered_map>

int numSplits(std::string s) {
    std::unordered_map<char, int> leftF;
    std::unordered_map<char, int> rightF;
    int count = 0;

    for (const auto& ch : s) {
        ++leftF[ch];
    }
    if (leftF.size() == 1) {
        return s.size() - 1;
    }

    for (int i = 0; i < s.size(); ++i) {
        ++rightF[s[i]];
        --leftF[s[i]];

        if (leftF[s[i]] == 0) {
            leftF.erase(s[i]);
        }

        if (leftF.size() == rightF.size()) {
            std::cout << i << "\n";
            ++count;
        }
    }

    return count;
}

int main() {
    std::string s = "acbadbaada";

    std::cout << numSplits(s);

    std::cin.get();
}