Подправьте задачу. (Определить количество слов в строке, которая содержит введенный символ)

#include "stdafx.h"
#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
 
int _tmain(int argc, _TCHAR* argv[])
{
    setlocale (LC_ALL, "RUS");
    char str[100];
    char m;
    int i, k=0;
    printf ("Введите строку: ");
    gets(str);
    printf ("Введите символ: ");
    scanf ("%c", &m);
    for (i=0; str[i]!='\0'; i++) {
        while (str[i]!=' ' && str[i]!='\0')
            i++;
 
            if (str[i-1]==m) 
                k++;
    }
    printf ("Количество слов, заканчивающихся на символ %c равно %i\n", m, k);
    system ("pause");
    return 0;

#include <iostream>
#include <string>
 
using namespace std;
 
int main()
{
    string str; getline(cin, str);
    char c = cin.get();
    int count = 0;
    bool contains = false;
 
    for (int i = 0; i != str.size(); ++i)
        if ( ::isspace( str[i] ) || ::ispunct( str[i] ) ) {
            count += contains ? 1 : 0;
            contains = false;
        } else
            contains |= str[i] == c;
 
    count += contains ? 1 : 0;
 
    cout << count;
 
    return 0;
}

#include <iostream>
#include <string>
#include <algorithm>
#include <iterator>
 
using namespace std;
 
int main()
{
    using iis = istream_iterator<string>;
    cout << "Filter char: ";
    char c = cin.get();
    cout << count_if( iis(cin), ( iis() ), [&c](const string &word){
                return any_of( begin(word), end(word), bind2nd(equal_to<char>(), c) ); } );
    return 0;
}