查找数组中给定字符的每个位置

时间:2012-10-15 15:19:20

标签: c char substring performance

我创建了一个小函数,它填充一个已分配的内存块,该内存块包含给定字符串中给定char的每个位置,并返回指向内存块的指针。

此功能的唯一问题是无法检查内存块的大小;所以我还创建了一个函数来计算字符串中给定char的出现次数。

以下是使用示例:

/*count occurences of char within a given string*/
size_t strchroc(const char *str, const char ch)
{ 
    int c = 0;
    while(*str) if(*(str++) == ch) c++;
    return c;
}

/*build array of positions of given char occurences within a given string*/
int *chrpos(const char *str, const char ch)
{
    int *array, *tmp, c = 0, i = 0;

    if(!(array = malloc(strlen(str) * sizeof(int)))) return 0x00;
    while(str[c])
    {
        if(str[c] == ch) array[i++] = c;
        c++;
    }
    if(!(tmp = realloc(array, i * sizeof(int)))) return 0x00;
    array = tmp;
    return array;
}

int main(void)
{
    char *str = "foobar foobar";                //'o' occurs at str[1], str[2], str[8], and str[9]
    int *array, b = 0, d;

    if(!(array = chrpos(str, 'o'))) exit(1);    //array[0] = 1, array[1] = 2, array[2] = 8, array[3] = 9

    /*
     * This is okay since I know that 'o'
     * only occures 4 times in str. There
     * may however be cases where I do not
     * know how many times a given char 
     * occurs so I figure that out before
     * utilizing the contents of array.
     * I do this with my function strchroc.
     * Below is a sample of how I would 
     * utilize the data contained within
     * array. This simply prints out str
     * and on a new line prints the given
     * char's location within the str 
     * array
     */

    puts(str);
    while(b < (int) strchroc(str, 'o'))         //loop once for each 'o' 
    {
        for(d = 0; d < (b == 0 ? array[b] : array[b] - array[b - 1] - 1); d++) putc((int) ' ', stdout);
        printf("%d", array[b]);
        b++;
    }
}

输出:

foobar foobar
 12     89

我唯一担心的是,如果这两个函数中的一个失败,则无法正确使用数据。我正在考虑将字符串中char的出现次数作为chrpos的参数,但即使这样,我仍然需要调用这两个函数。

我想知道是否有人有任何方法可以做到这一点,所以我只需要一个函数来构建数组。

我能想到的唯一方法是将char出现次数存储到array[0]并让array[1] through array[char_occurences]保持char的位置。

如果有人有更好的想法,我会非常感激。

2 个答案:

答案 0 :(得分:1)

您可以更改您的功能,以便它“返回”找到的出现次数。虽然我们实际上无法从C中的函数返回多个值,但我们可以将指针作为参数传递,并让函数使用该指针写下一个值:

int *chrpos(const char *str, char ch, int *found) {
    /* 
    ...
    */
    *found = i;
    return array;
}

请注意,您不需要const的{​​{1}}修饰符。

答案 1 :(得分:1)

正如我在评论中所述,第一件事就是保存数据,以防万一你不能缩小分配的内存:

if (!(tmp = realloc(array, i * sizeof(int))))
  return array;
return (tmp); //array = tmp; is useless

如果你想更多地保护你的strchroc功能,可以在开头添加if (!str) return 0;