Количество разных подстрок

#include <iostream>
#include <string>
#include <vector>
#include <unordered_set>


// Задача № 2. Количество разных подстрок
// Задача найти в строке число всех подстрок. Пустая строка также учитывается.

// Тест
// Входные данные:
// abc
// Выходные данные:
// 7

// Входные данные:
// ababab
// Выходные данные:
// 12

int mod = 1000000007;

void forward (std::string s, int k, std::unordered_set<int> &sh, int j) {
    int h = 0;
    int m = 1;
    for (int i = j; i < s.length(); i++) {
        char c = s[i];
        int x = (int) (c - 'a' + 1);
        h = (h + m * x) % mod;
        sh.insert(h);
        m = (m * k) % mod;
    }
}

int main() {
    std::string s;
    std::unordered_set<int> sh;
    int f = 0;
    int b = 0;
    int k = 31;
    std::cin >> s;
    for (int i = 0; i < s.length(); i++) {
        forward(s, k, sh, i);
    }
    std::cout << sh.size() + 1;
}