前言
最近在面试中一遇到算法题就懵了,总是能巧妙避过最优解方法给出最朴素、最贪心的答案,然后面试完脑子又能一闪而过更好的思路。
想起那么一句话叫“事前猪一样,事后诸葛亮”?
题目
给已排序数组(有正有负)按照绝对值大小进行排序,给出尽可能最优的时间复杂度和空间复杂度
思路
数组大概是这样,{-20, -9, -4, -1, -1, 0, 3, 5, 19}
如果负数且有正数存在,那么绝对值最小的一定在中间,绝对值最大的一定在左右两侧。那么可以有两种思路
额外设置一个大小为n的数组(空间复杂度 O ( n ) O(n) O(n))
- 先找到绝对值最小的(二分查找,时间复杂度 O ( l o g ( n ) ) O(log(n)) O(log(n))),分别向左右两边进行遍历、比较,按绝对值从小到大赋值到新数组(时间复杂度 O ( n ) O(n) O(n)),总时间复杂度为 O ( n ) O(n) O(n)
- 设置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),但具体思路还没想到,以后补上。