C++ 统计字符串中某字符出现的次数

题目描述

输入一个字符串s和一个字符ch，统计并输出字符ch在字符串s中出现的总次数。改写要求：编写一个函数，求字符指针p所指向的字符串中，字符ch出现的次数，并将该结果作为函数值返回。函数原型为
int CountChar ( char* p, char ch ) ;

输入描述

输入一个字符串后按回车键，再输入一个字符后回车。（提示：输入字符串用gets(s)，s为存放字符串的字符数组）

输出描述

输出为一个正整数，表示字符ch出现的总次数。

输入样例

bb
a

输出样例

0

#include <iostream>
#include <cstdlib>

using namespace std;

int countchar(char *str,char a){
    int n = 0;
    int i = 0;
    
    while(*(str+i) != '\0'){
        if(*(str+i) == a){
        	n++;
		}
        i++;
    }
    return n;
}

int main() {
    char str[20], a;
    int n;
    
    gets(str);
    a = getchar();
    n = countchar(str,a);
    cout << n << endl;
}