Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include <iostream>
- #include <set>
- #include <map>
- using namespace std;
- const int MOD = 1e9 + 7;
- int main() {
- ios_base::sync_with_stdio(false);
- int n;
- cin >> n;
- string s;
- cin >> s;
- set<char> st;
- for(int i = 0; i < n; i++) {
- st.insert(s[i]);
- }
- int K = (int) st.size();
- int res = 0;
- for(int i = 0; i < n; i++) {
- map<char, int> m;
- string p = "";
- for(int j = i; j < n; j++) {
- m[s[j]]++;
- p += s[j];
- if((j - i + 1) % K == 0) {
- int cnt = (j - i + 1) / K;
- bool ok = true;
- for(pair<char, int> mp: m) {
- if(mp.second != cnt) {
- ok = false;
- break;
- }
- }
- if(ok) {
- res++;
- res %= MOD;
- }
- }
- }
- }
- cout << res << endl;
- return 0;
- }
Advertisement
Add Comment
Please, Sign In to add comment