c++ - 对负值使用计数排序？

Question

我有以下代码，但这仅适用于无符号整数，我的目标是编写一个适用于所有整数的代码......

void CountingSort(vector<int> & a, vector<int> & b)
{
    int k=*max_element(a.begin(),a.end());
    k++;

    vector <int> c(k);

    for (int i=0;i<a.size();i++)
        c[a[i]]++;
    for (int i=1;i<k;i++)
        c[i]=c[i]+c[i-1];
    for (int i=0;i<a.size();i++)
    {
        b[c[a[i]]-1]=a[i];
        c[a[i]]--;
    }
}

如何将其更改为适用于所有整数类型？

Answer 1

首先计算最小值和最大值：

int k_min=*max_element(a.begin(),a.end());
int k_max=*min_element(a.begin(),a.end());
int k = k_max - k_min + 1;

对以下代码应用一些更改，替换a[i]为a[i] - k_min; 其余的应该很容易。

Answer 2

thnks Anatolyg 我的新代码是：

void Counting_Sort(vector <int>& a, vector <int>& b)
{
int k=*max_element(a.begin(), a.end());
int m=*min_element(a.begin(), a.end());

  int n=k-m+1;
  vector<int>v(n);
for(int i=0; i<a.size(); i++)
   v[a[i]-m]++;
 for(int i=1; i<v.size(); i++)
   v[i]=v[i]+v[i-1];
  for(int i=a.size()-1;i>=0; i--)
  { 
     b[v[a[i]-m]-1] = a[i]; 
     v[a[i]-m]--; 
  }
}