面试编程题:对已排序的数组(有正有负),按照绝对值从小到大重新排序

前言

最近在面试中一遇到算法题就懵了,总是能巧妙避过最优解方法给出最朴素、最贪心的答案,然后面试完脑子又能一闪而过更好的思路。

想起那么一句话叫“事前猪一样,事后诸葛亮”?

题目

已排序数组(有正有负)按照绝对值大小进行排序,给出尽可能最优的时间复杂度和空间复杂度

思路

数组大概是这样,{-20, -9, -4, -1, -1, 0, 3, 5, 19}

如果负数且有正数存在,那么绝对值最小的一定在中间绝对值最大的一定在左右两侧。那么可以有两种思路

额外设置一个大小为n的数组(空间复杂度 O ( n ) O(n) O(n)

  1. 先找到绝对值最小的(二分查找,时间复杂度 O ( l o g ( n ) ) O(log(n)) O(log(n))),分别向左右两边进行遍历、比较,按绝对值从小到大赋值到新数组(时间复杂度 O ( n ) O(n) O(n)),总时间复杂度为 O ( n ) O(n) O(n)
  2. 设置l, r指针,从左右两端向里进行遍历、比较,按绝对值从大到小的顺序赋值到新数组(时间复杂度 O ( n ) O(n) O(n))。

方法一在面试的时候花了几十分钟写完,面试完想了想,两分钟写了方法二。

解法

这里贴一下方法二的代码:

void abs_sort(vector<int> & nums){
    if(nums.empty())
        return ;

    int n = nums.size();
    int l = 0, r = n - 1, p = n - 1;
    vector<int> ans(n);
    
    while(l <= r){
        if(abs(nums[l]) > abs(nums[r]))
            ans[p--] = nums[l++];
        else 
            ans[p--] = nums[r--];
    }
    nums = ans;
}

int main() {
    vector<int> nums = {-20, -4, -1, -1, 3, 4, 5, 7, 9, 10};
    abs_sort(nums);
    
    for(auto n : nums)
        cout<<n<<" ";
    cout<<endl;

    return 0;
}

待续

直觉上,面试出的编程题空间复杂度应该还可以再优化到 O ( 1 ) O(1) O(1),但具体思路还没想到,以后补上。

你可能感兴趣的:(面试准备)