C函数库中的strpbrk实现

/*
*copyright@nciaebupt 转载请注明出处
*原型：char *strpbrk(const char *s1, const char *s2);
*用法：#include <string.h>
*功能：在字符串s1中寻找字符串s2中任何一个字符相匹配的第一个字符的位置，
*   空字符NULL不包括在内。
*说明：返回指向s1中第一个相匹配的字符的指针，如果没有匹配字符则返回空指针NULL。
*使用C函数库中的strpbrk
*/
#include <cstdio>
#include <cstring>

int main(int args,char ** argv)
{
    char str[] = "This is a sample string";
    char keys[] = "aeiou";
    printf("Vowels in '%s' : ",str);
    char * pch;
    pch = strpbrk(str,keys);
    while(pch != NULL)
    {
        printf("%c ",*pch);
        pch = strpbrk(pch + 1,keys);
    }
    getchar();
    return 0;
}

/*
*copyright@nciaebupt 转载请注明出处
*原型：char *strpbrk(const char *s1, const char *s2);
*用法：#include <string.h>
*功能：在字符串s1中寻找字符串s2中任何一个字符相匹配的第一个字符的位置，
*   空字符NULL不包括在内。
*说明：返回指向s1中第一个相匹配的字符的指针，如果没有匹配字符则返回空指针NULL。
*自己实现strpbrk
*/
#include <cstdio>
#include <cstring>

char * _strpbrk(const char * string,const char * control)
{
    const unsigned char *str = (const unsigned char *)string;
    const unsigned char *ctrl = (const unsigned char *)control;
    unsigned char map[32];
    int count;
    /*clear the map*/
    memset(map,0,32*sizeof(unsigned char));
    /*set bit in the control map*/
    while(*ctrl)
    {
        map[*ctrl >> 3] |= (0x01 << (*ctrl & 7));
        ctrl++;
    }
    /*search control in str*/
    while(*str)
    {
        if((map[*str >> 3] & (1 << (*str & 7))))
            return((char *)str);
        str++;
    }
    return NULL;

}

int main(int args,char ** argv)
{
    char str[] = "This is a sample string";
    char keys[] = "aeiou";
    printf("Vowels in '%s' : ",str);
    char * pch;
    pch = _strpbrk(str,keys);
    while(pch != NULL)
    {
        printf("%c ",*pch);
        pch = _strpbrk(pch + 1,keys);
    }
    getchar();
    return 0;
}