算法导论9.3-7习题解答

CLRS 9.3-7 :

给出一个O(n)时间的算法，在给定一个有n个不同数字的集合S以及一个正整数k<=n后，它能确定出S中最接近其中位数的k个数。

算法思想：
1.找到数组a中第n/2小的数median；
2.对a中非median数进行|a[i] - median|，得到一个大小为n - 1的数组distance；
3.寻找distance中第k小的数值；
4.对distance进行一次遍历，找到小于等于k的数，从而对应得到数组a中的k个数。

上述每一步的时间复杂度都为O(n)，因而最后总的时间复杂度为O(n).

#include < iostream >
#include < time.h >
using  namespace std;
// 随机化分割
int randomized_partition( int * a, int p, int r);
int randomized_select( int * a, int p, int r, int i);
int main()
{
  const  int LEN = 12 ;
  int arr[LEN] = { 5 , 7 , 10 , 3 , 6 , 2 , 8 , 9 , 4 , 1 , 11 , 12 };
  int median = randomized_select(arr, 0 , LEN - 1 , (LEN - 1 ) / 2 );
  int distance[LEN - 1 ];
  int distance_cpy[LEN - 1 ];

  int temp = 0 ;
  int middle;
  for ( int i = 0 ; i < LEN; i ++ )
  {
    if (arr[i] != median)
      distance[temp ++ ] = abs(arr[i] - median);
    else
      middle = i;
  }
  for ( int i = 0 ; i < LEN - 1 ; i ++ )
    distance_cpy[i] = distance[i];
  int k;
  while ( true )
  {
    cout << " 输入k： " << endl;
    cin >> k;
    cout << " k邻近值为： " << endl;
    int kth_value = randomized_select(distance_cpy, 0 , LEN - 2 , k - 1 );
    int * k_arr = new int [k];
    int j = 0 ;
    // 这里要注意，比如中间值为6，那么k = 3，有5,7明显符合要求，他们最近,4,8也符合要求，那么在这选择谁
    if (k % 2 )
    {
      bool flag = true ;
      for ( int i = 0 ; i < LEN - 1 ; i ++ )
      {
        if (distance[i] < kth_value)
        {
          if (i < middle)
            k_arr[j ++ ] = median - distance[i];
          else
            k_arr[j ++ ] = median + distance[i];
        }
        else  if (distance[i] == kth_value && flag)
        {
          if (i < middle)
            k_arr[j ++ ] = median - distance[i];
          else
            k_arr[j ++ ] = median + distance[i];
          flag = false ;
        }
      }
    }
    else
    {
      for ( int i = 0 ; i < LEN - 1 ; i ++ )
      {
        if (distance[i] <= kth_value)
        {
          if (i < middle)
            k_arr[j ++ ] = median - distance[i];
          else
            k_arr[j ++ ] = median + distance[i];
        }
      }
    }
    for ( int i = 0 ; i < k; i ++ )
    cout << k_arr[i] << " \t " ;
    cout << endl;
    delete[] k_arr;
  }
}
// 下标为[p, r]之间的元素
int randomized_partition( int * a, int p, int r)
{
  srand(time(NULL));
  int q = rand() % (r - p + 1 ) + p;
  int temp = a[q];
  a[q] = a[r];
  a[r] = temp;
  int j = p;
  for ( int i = p; i < r; i ++ )
  {
    if (a[i] < a[r])
    {
      if (i != j)
      {
        int temp2 = a[i];
        a[i] = a[j];
        a[j] = temp2;
      }
      j ++ ;
    }
  }

  temp = a[j];
  a[j] = a[r];
  a[r] = temp;
  return j;
}
// 迭代版本
int randomized_select( int * a, int p, int r, int i)
{
  int q = randomized_partition(a, p, r);
  while (p != r)
  {
    if (i == q)
      return a[q];
    else  if (i < q)
    {
      r = q - 1 ;
      q = randomized_partition(a, p, r);
    }
    else
    {
      p = q + 1 ;
      q = randomized_partition(a, p, r);
    }
  }
  return a[p];
}