风歌的blog

已搬家到 blog.imxjf.top
  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

桶排序

Posted on 2005-07-13 18:44  风歌  阅读(657)  评论(1编辑  收藏  举报
#include <iostream.h>
#include <iomanip.h>

// constant size must be defined as the array size for bucketSort to work
const int SIZE = 12;

void bucketSort( int [] );
void distributeElements( int [], int [][ SIZE ], int );
void collectElements( int [], int [][ SIZE ] );
int numberOfDigits( int [], int );
void zeroBucket( int [][ SIZE ] );

int main()
{
   int array[ SIZE ] = { 19, 13, 5, 27, 1, 26, 31, 16, 2, 9, 11, 21 };

   cout << "Array elements in original order:\n";

   for ( int i = 0; i < SIZE; ++i )
      cout << setw( 3 ) << array[ i ];

   cout << '\n';
   bucketSort( array );

   cout << "\nArray elements in sorted order:\n";

   for ( int j = 0; j < SIZE; ++j )
      cout << setw( 3 ) << array[ j ];

   cout << endl;
   return 0;
}

// Perform the bucket sort algorithm
void bucketSort( int a[] )
{
   int totalDigits, bucket[ 10 ][ SIZE ] = { 0 };

   totalDigits = numberOfDigits( a, SIZE );

   for ( int i = 1; i <= totalDigits; ++i ) {
      distributeElements( a, bucket, i );
      collectElements( a, bucket );

      if ( i != totalDigits )
         zeroBucket( bucket );  // set all bucket contents to zero
   }
}

// Determine the number of digits in the largest number
int numberOfDigits( int b[], int arraySize )
{
   int largest = b[ 0 ], digits = 0;

   for ( int i = 1; i < arraySize; ++i )
      if ( b[ i ] > largest )
         largest = b[ i ];

   while ( largest != 0 ) {
      ++digits;
      largest /= 10;
   }

   return digits;
}

// Distribute elements into buckets based on specified digit
void distributeElements( int a[], int buckets[][ SIZE ], int digit )
{
   int divisor = 10, bucketNumber, elementNumber;

   for ( int i = 1; i < digit; ++i )   // determine the divisor
      divisor *= 10;                 // used to get specific digit

   for ( int k = 0; k < SIZE; ++k ) {
      // bucketNumber example for hundreds digit:
      // (1234 % 1000 - 1234 % 100) / 100 --> 2

      bucketNumber = ( a[ k ] % divisor - a[ k ] %
                     ( divisor / 10 ) ) / ( divisor / 10 );

      // retrieve value in buckets[bucketNumber][0] to determine
      // which element of the row to store a[i] in.

      elementNumber = ++buckets[ bucketNumber ][ 0 ];
      buckets[ bucketNumber ][ elementNumber ] = a[ k ];
   }
}

// Return elements to original array
void collectElements( int a[], int buckets[][ SIZE ])
{
   int subscript = 0;

   for ( int i = 0; i < 10; ++i )
      for ( int j = 1; j <= buckets[ i ][ 0 ]; ++j )
         a[ subscript++ ] = buckets[ i ][ j ];
}

// Set all buckets to zero
void zeroBucket( int buckets[][ SIZE ] )
{
   for ( int i = 0; i < 10; ++i )
      for ( int j = 0; j < SIZE; ++j )
         buckets[ i ][ j ] = 0;
}