COCI '17 Contest 1 #3 Lozinke Hard

    1#include <bits/stdc++.h>
    2 
    3using namespace std;
    4 
    5int main() {
    6    ios::sync_with_stdio(0);
    7    cin.tie(0);
    8    int q;
    9    cin >> q;
    10    unordered_map<string, int> pass;
    11    while(q--){
    12        int ins;
    13        cin >> ins;
    14        string s;
    15        cin >> s;
    16        if(ins == 1){
    17            unordered_set<string> st;
    18            for(int i = 0; i < s.size(); i++){
    19                for(int j = i; j < s.size(); j++){
    20                    string temp = "";
    21                    for(int k = i; k <= j; k++){
    22                        temp += s[k];
    23                    }
    24                    st.insert(temp);
    25                }
    26            }
    27            for(string cur : st){
    28                pass[cur]++;
    29            }
    30        } else {
    31            cout << pass[s] << '\n';
    32        }
    33    }
    34    return 0;
    35}