题目描述
给定一个旋转排序数组,在原地恢复其排序。
说明
什么是旋转数组?
比如,原始数组为[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)时间复杂度
分析
根据题目中给出的说明,它的旋转数组都有一个共同的特点,那就是它将原数组分成两部分,然后将这两部分进行了翻转。也就是说可以通过移位实现数组的还原。所以关键就是找到移位数。
代码
class Solution {
public:
/*
* @param nums: An integer array
* @return: nothing
*/
void recoverRotatedSortedArray(vector<int> &nums) {
// write your code here
int i;
for(i = 0; i < nums.size() - 1; i++) {
if(nums[i] > nums[i + 1]) break;
}
reverse(nums, 0, i);
reverse(nums, i + 1, nums.size() - 1);
reverse(nums, 0, nums.size() - 1);
}
void reverse(vector<int> &nums, int i, int j) {
while(i < j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
i++, j--;
}
}
};
优化
直接就完成了挑战的要求,应该已经是合格的算法了,那就这样吧。