php源码str_repeat有趣的实现

今天在翻看php源码时, 研究了一下str_repeat的实现. 算法很有意思, 一般人的思路都是字符串拼接, 重复n次，就拼接n次, 时间复杂度为O(n). 但作者的思路很有意思, 把时间复杂度降到了log2^n+1.

上代码:

PHP_FUNCTION(str_repeat)
{
    zend_string     *input_str;     /* Input string */
    zend_long       mult;           /* Multiplier */
    zend_string *result;        /* Resulting string */
    size_t      result_len;     /* Length of the resulting string */

    if (zend_parse_parameters(ZEND_NUM_ARGS(), "Sl", &input_str, &mult) == FAILURE) {
        return;
    }

    if (mult < 0) {
        php_error_docref(NULL, E_WARNING, "Second argument has to be greater than or equal to 0");
        return;
    }

    /* Don't waste our time if it's empty */
    /* ... or if the multiplier is zero */
    if (ZSTR_LEN(input_str) == 0 || mult == 0)
        RETURN_EMPTY_STRING();

    /* Initialize the result string */
    result = zend_string_safe_alloc(ZSTR_LEN(input_str), mult, 0, 0);
    result_len = ZSTR_LEN(input_str) * mult;

    /* Heavy optimization for situations where input string is 1 byte long */
    if (ZSTR_LEN(input_str) == 1) {
        memset(ZSTR_VAL(result), *ZSTR_VAL(input_str), mult);
    } else {

        char *s, *e, *ee;
        ptrdiff_t l=0;
        memcpy(ZSTR_VAL(result), ZSTR_VAL(input_str), ZSTR_LEN(input_str));
        s = ZSTR_VAL(result);    //猜测这里的变量名为start
        e = ZSTR_VAL(result) + ZSTR_LEN(input_str);  //这里应该是end
        ee = ZSTR_VAL(result) + result_len;   //这里应该是end ending, 最后的结尾 

        //重点看这里, 整个算法的核心
        while (e<ee) {
            //判断剩余内存的大小是否大于已经复制的内存大小, 取小者
            l = (e-s) < (ee-e) ? (e-s) : (ee-e);
            memmove(e, s, l);
            e += l;
        }
    }

    ZSTR_VAL(result)[result_len] = '\0';

    RETURN_NEW_STR(result);
}

内存图:

第一次循环:
因为e-s < ee-e, 所以l取e-s, 也就是1

第二次循环:
因为e-s < ee-e, 所以l取e-s, 也就是2

第三次循环:
因为e-s > ee-e, 所以l取ee-e, 也就是2

第四次循环:
l=0