描述
说明
什么是旋转数组?
比如,原始数组为[1,2,3,4], 则其旋转数组可以是[1,2,3,4], [2,3,4,1], [3,4,1,2], [4,1,2,3]
样例
[4, 5, 1, 2, 3] -> [1, 2, 3, 4, 5]
挑战
给定一个旋转排序数组,在原地恢复其排序。
什么是旋转数组?
比如,原始数组为[1,2,3,4], 则其旋转数组可以是[1,2,3,4], [2,3,4,1], [3,4,1,2], [4,1,2,3]
样例
[4, 5, 1, 2, 3] -> [1, 2, 3, 4, 5]
挑战
使用O(1)的额外空间和O(n)时间复杂度
分析
对于该题目,要求O(1)的额外空间和O(n)时间复杂度,所以不能够新建数组或者多次循环调用。首先,判断该数组是否已经排好序;如果没有排好序,会在递增的途中突然减小,找到那个位置。将该位置之前的翻转,再将该位置之后的翻转,最后将整个数组翻转,示例如下:
[4, 5, 1, 2, 3]突然减小的位置是2,所以将[4, 5]进行翻转,得到[5, 4, 1, 2, 3];
将位置2及其之后的翻转,得到[5, 4, 3, 2, 1];
对整个数组进行翻转,得到[1, 2, 3, 4, 5]。
这里,翻转的子函数为交换两端位置,然后不断地向中间逼近,直到两端的位置相等。
程序
class Solution { public: /** * @param nums: An integer array * @return: nothing */ void recoverRotatedSortedArray(vector<int> &nums) { // write your code here // 判断是否已经排好序 int pos = 1, end = nums.size(); for(; (pos < end) && (nums[pos] >= nums[pos - 1]); pos++); if(pos == end) return; reverse_nums(nums,0, pos-1); reverse_nums(nums,pos, end-1); reverse_nums(nums,0, end-1); } void reverse_nums(vector<int> &nums, int begin, int end){ while(begin < end){ swap(nums[begin++], nums[end--]); } } };