现在的位置: 首页 > 综合 > 正文

Heap sort Sample

2014年03月22日 ⁄ 综合 ⁄ 共 1323字 ⁄ 字号 评论关闭
  1. static void u32_swap(void *a, void *b, int size)
  2. {
  3.     u32 t = *(u32 *)a;
  4.     *(u32 *)a = *(u32 *)b;
  5.     *(u32 *)b = t;
  6. }
  7. static void generic_swap(void *a, void *b, int size)
  8. {
  9.     char t;
  10.     do {
  11.         t = *(char *)a;
  12.         *(char *)a++ = *(char *)b;
  13.         *(char *)b++ = t;
  14.     } while (--size > 0);
  15. }
  16. void sort(void *base, size_t num, size_t size,
  17.       int (*cmp)(const void *, const void *),
  18.       void (*swap)(void *, void *, int size))
  19. {
  20.     /* pre-scale counters for performance */
  21.     int i = (num/2 - 1) * size, n = num * size, c, r;
  22.     if (!swap)
  23.         swap = (size == 4 ? u32_swap : generic_swap);
  24.     /* heapify */
  25.     for ( ; i >= 0; i -= size) {
  26.         for (r = i; r * 2 + size < n; r  = c) {
  27.             c = r * 2 + size;
  28.             if (c < n - size && cmp(base + c, base + c + size) < 0)
  29.                 c += size;
  30.             if (cmp(base + r, base + c) >= 0)
  31.                 break;
  32.             swap(base + r, base + c, size);
  33.         }
  34.     }
  35.     /* sort */
  36.     for (i = n - size; i > 0; i -= size) {
  37.         swap(base, base + i, size);
  38.         for (r = 0; r * 2 + size < i; r = c) {
  39.             c = r * 2 + size;
  40.             if (c < i - size && cmp(base + c, base + c + size) < 0)
  41.                 c += size;
  42.             if (cmp(base + r, base + c) >= 0)
  43.                 break;
  44.             swap(base + r, base + c, size);
  45.         }
  46.     }
  47. }

抱歉!评论已关闭.