当前位置:首页 > PHP教程 > php数组 > 列表

PHP中array_keys和array_unique函数源码的分析

发布:smiling 来源: PHP粉丝网  添加日期:2019-12-04 10:16:31 浏览: 评论:0 

性能分析:从运行性能上分析,看看下面的测试代码:

  1. $test=array(); 
  2.  
  3. for($run=0; $run<10000; $run++) 
  4.  
  5. $test[]=rand(0,100); 
  6.  
  7. $time=microtime(true); 
  8.  
  9. $out = array_unique($test); 
  10.  
  11. $time=microtime(true)-$time
  12.  
  13. echo 'Array Unique: '.$time."\n"
  14.  
  15. $time=microtime(true); 
  16.  
  17. $out=array_keys(array_flip($test)); 
  18.  
  19. $time=microtime(true)-$time
  20.  
  21. echo 'Keys Flip: '.$time."\n"
  22.  
  23. $time=microtime(true); 
  24.  
  25. $out=array_flip(array_flip($test)); 
  26.  
  27. $time=microtime(true)-$time
  28.  
  29. echo 'Flip Flip: '.$time."\n"

运行结果如下:

从上图可以看到,使用array_unique函数需要0.069s;使用array_flip后再使用array_keys函数需要0.00152s;使用两次array_flip函数需要0.00146s。

测试结果表明,使用array_flip后再调用array_keys函数比array_unique函数快。那么,具体原因是什么呢?让我们看看在PHP底层,这两个函数是怎么实现的。

源码分析:

  1. /* {{{ proto array array_keys(array input [, mixed search_value[, bool strict]]) 
  2.  
  3.   Return just the keys from the input array, optionally only for the specified       search_value */ 
  4.  
  5. PHP_FUNCTION(array_keys
  6.  
  7.  
  8.   //变量定义 
  9.  
  10.   zval *input,        /* Input array */ 
  11.  
  12.      *search_value = NULL,  /* Value to search for */ 
  13.  
  14.      **entry,        /* An entry in the input array */ 
  15.  
  16.       res,          /* Result of comparison */ 
  17.  
  18.      *new_val;        /* New value */ 
  19.  
  20.   int  add_key;        /* Flag to indicate whether a key should be added */ 
  21.  
  22.   char *string_key;      /* String key */ 
  23.  
  24.   uint  string_key_len; 
  25.  
  26.   ulong num_key;        /* Numeric key */ 
  27.  
  28.   zend_bool strict = 0;    /* do strict comparison */ 
  29.  
  30.   HashPosition pos; 
  31.  
  32.   int (*is_equal_func)(zval *, zval *, zval * TSRMLS_DC) = is_equal_function; 
  33.  
  34.   
  35.  
  36.   //程序解析参数 
  37.  
  38.   if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "a|zb", &input, &search_value, &strict) == FAILURE) { 
  39.  
  40.     return
  41.  
  42.   } 
  43.  
  44.   
  45.  
  46.   // 如果strict是true,则设置is_equal_func为is_identical_function,即全等比较 
  47.  
  48.   if (strict) { 
  49.  
  50.     is_equal_func = is_identical_function; 
  51.  
  52.   } 
  53.  
  54.   
  55.  
  56.   /* 根据search_vale初始化返回的数组大小 */ 
  57.  
  58.   if (search_value != NULL) { 
  59.  
  60.     array_init(return_value); 
  61.  
  62.   } else { 
  63.  
  64.     array_init_size(return_value, zend_hash_num_elements(Z_ARRVAL_P(input))); 
  65.  
  66.   } 
  67.  
  68.   add_key = 1; 
  69.  
  70.   
  71.  
  72.   /* 遍历输入的数组参数,然后添加键值到返回的数组 */ 
  73.  
  74.   zend_hash_internal_pointer_reset_ex(Z_ARRVAL_P(input), &pos);//重置指针 
  75.  
  76.   //循环遍历数组 
  77.  
  78.   while (zend_hash_get_current_data_ex(Z_ARRVAL_P(input), (void **)&entry, &pos) == SUCCESS) { 
  79.  
  80.     // 如果search_value不为空 
  81.  
  82.     if (search_value != NULL) { 
  83.  
  84.       // 判断search_value与当前的值是否相同,并将比较结果保存到add_key变量 
  85.  
  86.       is_equal_func(&res, search_value, *entry TSRMLS_CC); 
  87.  
  88.       add_key = zval_is_true(&res); 
  89.  
  90.     } 
  91.  
  92.   
  93.  
  94.     if (add_key) { 
  95.  
  96.       // 创建一个zval结构体 
  97.  
  98.       MAKE_STD_ZVAL(new_val); 
  99.  
  100.   
  101.  
  102.       // 根据键值是字符串还是整型数字将值插入到return_value中 
  103.  
  104.       switch (zend_hash_get_current_key_ex(Z_ARRVAL_P(input), &string_key, &string_key_len, &num_key, 1, &pos)) { 
  105.  
  106.         case HASH_KEY_IS_STRING: 
  107.  
  108.           ZVAL_STRINGL(new_val, string_key, string_key_len - 1, 0); 
  109.  
  110.           // 此函数负责将值插入到return_value中,如果键值已存在,则使用新值更新对应的值,否则直接插入 
  111.  
  112.           zend_hash_next_index_insert(Z_ARRVAL_P(return_value), &new_val, sizeof(zval *), NULL); 
  113.  
  114.           break
  115.  
  116.   
  117.  
  118.         case HASH_KEY_IS_LONG: 
  119.  
  120.           Z_TYPE_P(new_val) = IS_LONG
  121.  
  122.           Z_LVAL_P(new_val) = num_key; 
  123.  
  124.           zend_hash_next_index_insert(Z_ARRVAL_P(return_value), &new_val, sizeof(zval *), NULL); 
  125.  
  126.           break
  127.  
  128.       } 
  129.  
  130.     } 
  131.  
  132.  //phpfensi.com 
  133.  
  134.     // 移动到下一个 
  135.  
  136.     zend_hash_move_forward_ex(Z_ARRVAL_P(input), &pos); 
  137.  
  138.   } 
  139.  
  140.  
  141. /* }}} */ 

以上是array_keys函数底层的源码。为方便理解,笔者添加了一些中文注释。如果需要查看原始代码,可以点击查看。这个函数的功能就是新建一个临时数组,然后将键值对重新复制到新的数组,如果复制过程中有重复的键值出现,那么就用新的值替换。

这个函数的主要步骤是地57和63行调用的zend_hash_next_index_insert函数。该函数将元素插入到数组中,如果出现重复的值,则使用新的值更新原键值指向的值,否则直接插入,时间复杂度是O(n)。

  1. /* {{{ proto array array_flip(array input) 
  2.  
  3.   Return array with key <-> value flipped */ 
  4.  
  5. PHP_FUNCTION(array_flip
  6.  
  7.  
  8.   // 定义变量 
  9.  
  10.   zval *array, **entry, *data; 
  11.  
  12.   char *string_key; 
  13.  
  14.   uint str_key_len; 
  15.  
  16.   ulong num_key; 
  17.  
  18.   HashPosition pos;  
  19.  
  20.   // 解析数组参数 
  21.  
  22.   if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "a", &array) == FAILURE) { 
  23.  
  24.     return
  25.  
  26.   }
  27.   
  28.  
  29.   // 初始化返回数组 
  30.  
  31.   array_init_size(return_value, zend_hash_num_elements(Z_ARRVAL_P(array))); 
  32.  
  33.   // 重置指针 
  34.  
  35.   zend_hash_internal_pointer_reset_ex(Z_ARRVAL_P(array), &pos); 
  36.  
  37.   // 遍历每个元素,并执行键<->值交换操作 
  38.  
  39.   while (zend_hash_get_current_data_ex(Z_ARRVAL_P(array), (void **)&entry, &pos) == SUCCESS) { 
  40.  
  41.     // 初始化一个结构体 
  42.  
  43.     MAKE_STD_ZVAL(data); 
  44.  
  45.     // 将原数组的值赋值为新数组的键 
  46.  
  47.     switch (zend_hash_get_current_key_ex(Z_ARRVAL_P(array), &string_key, &str_key_len, &num_key, 1, &pos)) { 
  48.  
  49.       case HASH_KEY_IS_STRING: 
  50.  
  51.         ZVAL_STRINGL(data, string_key, str_key_len - 1, 0); 
  52.  
  53.         break
  54.  
  55.       case HASH_KEY_IS_LONG: 
  56.  
  57.         Z_TYPE_P(data) = IS_LONG
  58.  
  59.         Z_LVAL_P(data) = num_key; 
  60.  
  61.         break
  62.  
  63.     }  
  64.     // 将原数组的键赋值为新数组的值,如果有重复的,则使用新值覆盖旧值 
  65.  
  66.     if (Z_TYPE_PP(entry) == IS_LONG) { 
  67.  
  68.       zend_hash_index_update(Z_ARRVAL_P(return_value), Z_LVAL_PP(entry), &data, sizeof(data), NULL); 
  69.  
  70.     } else if (Z_TYPE_PP(entry) == IS_STRING) { 
  71.  
  72.       zend_symtable_update(Z_ARRVAL_P(return_value), Z_STRVAL_PP(entry), Z_STRLEN_PP(entry) + 1, &data, sizeof(data), NULL); 
  73.  
  74.     } else { 
  75.  
  76.       zval_ptr_dtor(&data); /* will free also zval structure */ 
  77.  
  78.       php_error_docref(NULL TSRMLS_CC, E_WARNING, "Can only flip STRING and INTEGER values!"); 
  79.  
  80.     } 
  81.  
  82.     // 下一个 
  83.  
  84.     zend_hash_move_forward_ex(Z_ARRVAL_P(array), &pos); 
  85.  
  86.   } 
  87.  
  88.  
  89. /* }}} */ 

上面就是是array_flip函数的源码。点击链接查看原始代码。这个函数主要的做的事情就是创建一个新的数组,遍历原数组。在26行开始将原数组的值赋值为新数组的键,然后在37行开始将原数组的键赋值为新数组的值,如果有重复的,则使用新值覆盖旧值。整个函数的时间复杂度也是O(n)。因此,使用了array_flip之后再使用array_keys的时间复杂度是O(n)。

接下来,我们看看array_unique函数的源码。点击链接查看原始代码。

  1. /* {{{ proto array array_unique(array input [, int sort_flags]) 
  2.  
  3.   Removes duplicate values from array */ 
  4.  
  5. PHP_FUNCTION(array_unique
  6.  
  7.  
  8.   // 定义变量 
  9.  
  10.   zval *array, *tmp; 
  11.  
  12.   Bucket *p; 
  13.  
  14.   struct bucketindex { 
  15.  
  16.     Bucket *b; 
  17.  
  18.     unsigned int i; 
  19.  
  20.   }; 
  21.  
  22.   struct bucketindex *arTmp, *cmpdata, *lastkept; 
  23.  
  24.   unsigned int i; 
  25.  
  26.   long sort_type = PHP_SORT_STRING; 
  27.  
  28.   
  29.  
  30.   // 解析参数 
  31.  
  32.   if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "a|l", &array, &sort_type) == FAILURE) { 
  33.  
  34.     return
  35.  
  36.   } 
  37.  
  38.   
  39.  
  40.   // 设置比较函数 
  41.  
  42.   php_set_compare_func(sort_type TSRMLS_CC); 
  43.  
  44.   
  45.  
  46.   // 初始化返回数组 
  47.  
  48.   array_init_size(return_value, zend_hash_num_elements(Z_ARRVAL_P(array))); 
  49.  
  50.   // 将值拷贝到新数组 
  51.  
  52.   zend_hash_copy(Z_ARRVAL_P(return_value), Z_ARRVAL_P(array), (copy_ctor_func_t) zval_add_ref, (void *)&tmp, sizeof(zval*)); 
  53.  
  54.   
  55.  
  56.   if (Z_ARRVAL_P(array)->nNumOfElements <= 1) {  /* 什么都不做 */ 
  57.  
  58.     return
  59.  
  60.   } 
  61.  
  62.   
  63.  
  64.   /* 根据target_hash buckets的指针创建数组并排序 */ 
  65.  
  66.   arTmp = (struct bucketindex *) pemalloc((Z_ARRVAL_P(array)->nNumOfElements + 1) * sizeof(struct bucketindex), Z_ARRVAL_P(array)->persistent); 
  67.  
  68.   if (!arTmp) { 
  69.  
  70.     zval_dtor(return_value); 
  71.  
  72.     RETURN_FALSE; 
  73.  
  74.   } 
  75.  
  76.   for (i = 0, p = Z_ARRVAL_P(array)->pListHead; p; i++, p = p->pListNext) { 
  77.  
  78.     arTmp[i].b = p; 
  79.  
  80.     arTmp[i].i = i; 
  81.  
  82.   } 
  83.  
  84.   arTmp[i].b = NULL; 
  85.  
  86.   // 排序 
  87.  
  88.   zend_qsort((void *) arTmp, i, sizeof(struct bucketindex), php_array_data_compare TSRMLS_CC); 
  89.  
  90.   
  91.  
  92.   /* 遍历排序好的数组,然后删除重复的元素 */ 
  93.  
  94.   lastkept = arTmp; 
  95.  
  96.   for (cmpdata = arTmp + 1; cmpdata->b; cmpdata++) { 
  97.  
  98.     if (php_array_data_compare(lastkept, cmpdata TSRMLS_CC)) { 
  99.  
  100.       lastkept = cmpdata; 
  101.  
  102.     } else { 
  103.  
  104.       if (lastkept->i > cmpdata->i) { 
  105.  
  106.         p = lastkept->b; 
  107.  
  108.         lastkept = cmpdata; 
  109.  
  110.       } else { 
  111.  
  112.         p = cmpdata->b; 
  113.  
  114.       } 
  115.  
  116.       if (p->nKeyLength == 0) { 
  117.  
  118.         zend_hash_index_del(Z_ARRVAL_P(return_value), p->h); 
  119.  
  120.       } else { 
  121.  
  122.         if (Z_ARRVAL_P(return_value) == &EG(symbol_table)) { 
  123.  
  124.           zend_delete_global_variable(p->arKey, p->nKeyLength - 1 TSRMLS_CC); 
  125.  
  126.         } else { 
  127.  
  128.           zend_hash_quick_del(Z_ARRVAL_P(return_value), p->arKey, p->nKeyLength, p->h); 
  129.  
  130.         } 
  131.  
  132.       } 
  133.  
  134.     } 
  135.  
  136.   } 
  137.  
  138.   pefree(arTmp, Z_ARRVAL_P(array)->persistent); 
  139.  
  140.  
  141. /* }}} */ 

可以看到,这个函数初始化一个新的数组,然后将值拷贝到新数组,然后在45行调用排序函数对数组进行排序,排序的算法是zend引擎的块树排序算法。接着遍历排序好的数组,删除重复的元素。整个函数开销最大的地方就在调用排序函数上,而快排的时间复杂度是O(nlogn),因此,该函数的时间复杂度是O(nlogn)。

结论:

因为array_unique底层调用了快排算法,加大了函数运行的时间开销,导致整个函数的运行较慢。这就是为什么array_keys比array_unique函数更快的原因。

Tags: array_keys array_unique

分享到: