hash.c
上传用户:zjbsled
上传日期:2007-01-08
资源大小:4k
文件大小:9k
开发平台:

C/C++

  1. /* +++Date last modified: 05-Jul-1997 */
  2. #include <string.h>
  3. #include <stdlib.h>
  4. #include "hash.h"
  5. /*
  6. ** public domain code by Jerry Coffin, with improvements by HenkJan Wolthuis.
  7. **
  8. ** Tested with Visual C 1.0 and Borland C 3.1.
  9. ** Compiles without warnings, and seems like it should be pretty
  10. ** portable.
  11. */
  12. /*
  13. ** These are used in freeing a table.  Perhaps I should code up
  14. ** something a little less grungy, but it works, so what the heck.
  15. */
  16. static void (*function)(void *) = (void (*)(void *))NULL;
  17. static hash_table *the_table = NULL;
  18. /* Initialize the hash_table to the size asked for.  Allocates space
  19. ** for the correct number of pointers and sets them to NULL.  If it
  20. ** can't allocate sufficient memory, signals error by setting the size
  21. ** of the table to 0.
  22. */
  23. hash_table *construct_table(hash_table *table, size_t size)
  24. {
  25.       size_t i;
  26.       bucket **temp;
  27.       table -> size  = size;
  28.       table -> table = (bucket * *)malloc(sizeof(bucket *) * size);
  29.       temp = table -> table;
  30.       if ( temp == NULL )
  31.       {
  32.             table -> size = 0;
  33.             return table;
  34.       }
  35.       for (i=0;i<size;i++)
  36.             temp[i] = NULL;
  37.       return table;
  38. }
  39. /*
  40. ** Hashes a string to produce an unsigned short, which should be
  41. ** sufficient for most purposes.
  42. */
  43. static unsigned hash(char *string)
  44. {
  45.       unsigned ret_val = 0;
  46.       int i;
  47.       while (*string)
  48.       {
  49.             i = *( int *)string;
  50.             ret_val ^= i;
  51.             ret_val <<= 1;
  52.             string ++;
  53.       }
  54.       return ret_val;
  55. }
  56. /*
  57. ** Insert 'key' into hash table.
  58. ** Returns pointer to old data associated with the key, if any, or
  59. ** NULL if the key wasn't in the table previously.
  60. */
  61. void *insert(char *key, void *data, hash_table *table)
  62. {
  63.       unsigned val = hash(key) % table->size;
  64.       bucket *ptr;
  65.       /*
  66.       ** NULL means this bucket hasn't been used yet.  We'll simply
  67.       ** allocate space for our new bucket and put our data there, with
  68.       ** the table pointing at it.
  69.       */
  70.       if (NULL == (table->table)[val])
  71.       {
  72.             (table->table)[val] = (bucket *)malloc(sizeof(bucket));
  73.             if (NULL==(table->table)[val])
  74.                   return NULL;
  75.             (table->table)[val] -> key = strdup(key);
  76.             (table->table)[val] -> next = NULL;
  77.             (table->table)[val] -> data = data;
  78.             return (table->table)[val] -> data;
  79.       }
  80.       /*
  81.       ** This spot in the table is already in use.  See if the current string
  82.       ** has already been inserted, and if so, increment its count.
  83.       */
  84.       for (ptr = (table->table)[val];NULL != ptr; ptr = ptr -> next)
  85.             if (0 == strcmp(key, ptr->key))
  86.             {
  87.                   void *old_data;
  88.                   old_data = ptr->data;
  89.                   ptr -> data = data;
  90.                   return old_data;
  91.             }
  92.       /*
  93.       ** This key must not be in the table yet.  We'll add it to the head of
  94.       ** the list at this spot in the hash table.  Speed would be
  95.       ** slightly improved if the list was kept sorted instead.  In this case,
  96.       ** this code would be moved into the loop above, and the insertion would
  97.       ** take place as soon as it was determined that the present key in the
  98.       ** list was larger than this one.
  99.       */
  100.       ptr = (bucket *)malloc(sizeof(bucket));
  101.       if (NULL==ptr)
  102.             return 0;
  103.       ptr -> key = strdup(key);
  104.       ptr -> data = data;
  105.       ptr -> next = (table->table)[val];
  106.       (table->table)[val] = ptr;
  107.       return data;
  108. }
  109. /*
  110. ** Look up a key and return the associated data.  Returns NULL if
  111. ** the key is not in the table.
  112. */
  113. void *lookup(char *key, hash_table *table)
  114. {
  115.       unsigned val = hash(key) % table->size;
  116.       bucket *ptr;
  117.       if (NULL == (table->table)[val])
  118.             return NULL;
  119.       for ( ptr = (table->table)[val];NULL != ptr; ptr = ptr->next )
  120.       {
  121.             if (0 == strcmp(key, ptr -> key ) )
  122.                   return ptr->data;
  123.       }
  124.       return NULL;
  125. }
  126. /*
  127. ** Delete a key from the hash table and return associated
  128. ** data, or NULL if not present.
  129. */
  130. void *del(char *key, hash_table *table)
  131. {
  132.       unsigned val = hash(key) % table->size;
  133.       void *data;
  134.       bucket *ptr, *last = NULL;
  135.       if (NULL == (table->table)[val])
  136.             return NULL;
  137.       /*
  138.       ** Traverse the list, keeping track of the previous node in the list.
  139.       ** When we find the node to delete, we set the previous node's next
  140.       ** pointer to point to the node after ourself instead.  We then delete
  141.       ** the key from the present node, and return a pointer to the data it
  142.       ** contains.
  143.       */
  144.       for (last = NULL, ptr = (table->table)[val];
  145.             NULL != ptr;
  146.             last = ptr, ptr = ptr->next)
  147.       {
  148.             if (0 == strcmp(key, ptr -> key))
  149.             {
  150.                   if (last != NULL )
  151.                   {
  152.                         data = ptr -> data;
  153.                         last -> next = ptr -> next;
  154.                         free(ptr->key);
  155.                         free(ptr);
  156.                         return data;
  157.                   }
  158.                   /*
  159.                   ** If 'last' still equals NULL, it means that we need to
  160.                   ** delete the first node in the list. This simply consists
  161.                   ** of putting our own 'next' pointer in the array holding
  162.                   ** the head of the list.  We then dispose of the current
  163.                   ** node as above.
  164.                   */
  165.                   else
  166.                   {
  167.                         data = ptr->data;
  168.                         (table->table)[val] = ptr->next;
  169.                         free(ptr->key);
  170.                         free(ptr);
  171.                         return data;
  172.                   }
  173.             }
  174.       }
  175.       /*
  176.       ** If we get here, it means we didn't find the item in the table.
  177.       ** Signal this by returning NULL.
  178.       */
  179.       return NULL;
  180. }
  181. /*
  182. ** free_table iterates the table, calling this repeatedly to free
  183. ** each individual node.  This, in turn, calls one or two other
  184. ** functions - one to free the storage used for the key, the other
  185. ** passes a pointer to the data back to a function defined by the user,
  186. ** process the data as needed.
  187. */
  188. static void free_node(char *key, void *data)
  189. {
  190.       (void) data;
  191.       if (function)
  192.             function(del(key,the_table));
  193.       else  del(key,the_table);
  194. }
  195. /*
  196. ** Frees a complete table by iterating over it and freeing each node.
  197. ** the second parameter is the address of a function it will call with a
  198. ** pointer to the data associated with each node.  This function is
  199. ** responsible for freeing the data, or doing whatever is needed with
  200. ** it.
  201. */
  202. void free_table(hash_table *table, void (*func)(void *))
  203. {
  204.       function = func;
  205.       the_table = table;
  206.       enumerate( table, free_node);
  207.       free(table->table);
  208.       table->table = NULL;
  209.       table->size = 0;
  210.       the_table = NULL;
  211.       function = (void (*)(void *))NULL;
  212. }
  213. /*
  214. ** Simply invokes the function given as the second parameter for each
  215. ** node in the table, passing it the key and the associated data.
  216. */
  217. void enumerate( hash_table *table, void (*func)(char *, void *))
  218. {
  219.       unsigned i;
  220.       bucket *temp;
  221.       for (i=0;i<table->size; i++)
  222.       {
  223.             if ((table->table)[i] != NULL)
  224.             {
  225.                   for (temp = (table->table)[i];
  226.                         NULL != temp;
  227.                         temp = temp -> next)
  228.                   {
  229.                         func(temp -> key, temp->data);
  230.                   }
  231.             }
  232.       }
  233. }
  234. #ifdef TEST
  235. #include <stdio.h>
  236. void printer(char *string, void *data)
  237. {
  238.       printf("%s: %sn", string, (char *)data);
  239. }
  240. int main(void)
  241. {
  242.       hash_table table;
  243.       char *strings[] = {
  244.             "The first string",
  245.             "The second string",
  246.             "The third string",
  247.             "The fourth string",
  248.             "A much longer string than the rest in this example.",
  249.             "The last string",
  250.             NULL
  251.             };
  252.       char *junk[] = {
  253.             "The first data",
  254.             "The second data",
  255.             "The third data",
  256.             "The fourth data",
  257.             "The fifth datum",
  258.             "The sixth piece of data"
  259.             };
  260.       int i;
  261.       void *j;
  262.       construct_table(&table,200);
  263.       for (i = 0; NULL != strings[i]; i++ )
  264.             insert(strings[i], junk[i], &table);
  265.       for (i=0;NULL != strings[i];i++)
  266.       {
  267.             printf("n");
  268.             enumerate(&table, printer);
  269.             del(strings[i],&table);
  270.       }
  271.       for (i=0;NULL != strings[i];i++)
  272.       {
  273.             j = lookup(strings[i], &table);
  274.             if (NULL == j)
  275.                   printf("n'%s' is not in table",strings[i]);
  276.             else  printf("nERROR: %s was deleted but is still in table.",
  277.                   strings[i]);
  278.       }
  279.       free_table(&table, NULL);
  280.       return 0;
  281. }
  282. #endif /* TEST */