Задача на строки

#include <cstdio>
#include <iostream>
#include <cmath>
#include <string>
using namespace std;
 
void toLowerCase(string &s)
{
    for(int i = 0; i < s.length(); ++i)
    {
        if(s[i] >= 'A' && s[i] <= 'Z')
            s[i] = char(s[i] - 'A' + 'a');
    }
}
 
int search(string s, string sub, int k, int b)
{
    int n = s.length();
    int m = sub.length();
 
    if(b)
    {
        toLowerCase(s);
        toLowerCase(sub);
    }
 
    int res = -1;
 
    for(int i = 0; i < k; ++i)
    {
        res = s.find(sub, res + 1);
 
        if(res == -1)
            break;
    }
 
    return res;
}
 
int main()
{
    string s, sub;
    getline(cin, s);
    getline(cin, sub);
 
    int k;
    cin >> k;
 
    int b;
    cin >> b;
 
    int res = search(s, sub, k, b);
 
    cout << res << endl;
 
    return 0;
}

#include <cstdio>
#include <iostream>
#include <cmath>
#include <string>
using namespace std;
 
void toLowerCase(string &s)  // перевод строки в нижний регистр, строка передается по ссылке
{
    for(int i = 0; i < s.length(); ++i)
    {
        if(s[i] >= 'A' && s[i] <= 'Z') // если i-й символ строки - большая буква, то
            s[i] = char(s[i] - 'A' + 'a'); // сделаем его маленькой буквой
    }
}
 
int search(string s, string sub, int k, int b)
{
    int n = s.length(); // длина строки s
    int m = sub.length(); // длина строки sub
 
    if(b) 
    {
        toLowerCase(s); // переводим s, sub в нижний регистр
        toLowerCase(sub);
    }
 
    int res = -1; // текущий ответ 
 
    for(int i = 0; i < k; ++i) // сделаем k итераций (т.к. нужно k-е вхождение)
    {
        res = s.find(sub, res + 1); // ищем первое вхождение строки sub начиная с позиции res + 1
 
        if(res == -1) // если строка sub дальше не встречается, т.е. res == -1
            break; // выходим из цикла
    }
 
    return res;
}
 
int main()
{
    string s, sub;
    getline(cin, s); // читаем строку s
    getline(cin, sub); // читаем строку sub
 
    int k; 
    cin >> k; // читаем число k
 
    int b;
    cin >> b; // читаем число b, если b = 1, то игнорируем регистр букв
 
    int res = search(s, sub, k, b); 
 
    cout << res << endl;
 
    return 0;
}