Задачка о поиске подстроки

[8Observer8]

Моё решение может не столь элегантное, как у Ромахи, но я свою тоже сдал сейчас

Я сформировал множество из неповторяющихся перестановок (сдвигов) с помощью Set, в который нельзя вставить одинаковые элементы. Точнее он игнорирует вставку одинаковых элементов.

Есть цикл по количеству уникальных подстрок. В этом цикле нахожу в главной строке количество этих подстрок и суммирую их.

/* 
 * File:   main.cpp
 * Author: Ivan
 *
 * Created on February 5, 2014, 9:37 AM
 */

#include <iostream>
#include <fstream>
#include <string>
#include <set>
using namespace std;

string shiftStringToLeft(const string &inStr) {
    string str = inStr;
    if (!str.empty()) {
        char c = str[0];
        str = str.substr(1, str.size() - 1);
        str.push_back(c);
    }
    return str;
}

void numberOfDifferentPermutations(const string &inStr, set<string> &outArray) {
    int n = inStr.size();
    string str = inStr;
    for (int i = 0; i < n; ++i) {
        outArray.insert(str);
        str = shiftStringToLeft(str);
    }
}

int amountOfSubstrings(string inStr, string substr) {
    int nfound = 0;
    size_t found = inStr.find(substr);
    while (found != string::npos) {
        nfound++;
        found = inStr.find(substr, found+1);
    }
    return nfound;
}

int nCyclicalSubstrings(string inStr, string substr) {
    set<string> outArray;
    numberOfDifferentPermutations(substr, outArray);
    set<string>::const_iterator it = outArray.begin();

    int nfound = 0;
    int size = outArray.size();
    for (int i = 0; i < size; ++i) {
        nfound += amountOfSubstrings(inStr, *it);
        it++;
    }

    return nfound;
}

int main(int argc, char** argv) {
    string inFileName = "input.txt";
    ifstream in;
    in.open(inFileName.c_str());
    if (!in.is_open()) {
        cerr << "Error: could not open the file " << inFileName.c_str() << endl;
        return 1;
    }

    string outFileName = "output.txt";
    ofstream out;
    out.open(outFileName.c_str());
    if (!out.is_open()) {
        cerr << "Error: could not open the file " << outFileName.c_str() << endl;
        in.close();
        return 1;
    }

    string str;
    string substr;
    if (in >> str >> substr) {
        int result = nCyclicalSubstrings(str, substr);
        out << result << endl;
    } else {
        cerr << "Error: incorrect data in the file " << inFileName.c_str() << endl;
        in.close();
        out.close();
        return 1;
    }

    in.close();
    out.close();
    return 0;
}