数组-恢复旋转排序数组-简单

最新推荐文章于 2024-11-10 23:30:05 发布

原创最新推荐文章于 2024-11-10 23:30:05 发布 · 621 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

LintCode 专栏收录该内容

94 篇文章

订阅专栏

本文介绍了一种在O(1)额外空间和O(n)时间复杂度内恢复旋转排序数组的方法。通过查找数组中突然减小的位置，进行三次翻转操作实现数组的排序。

描述

给定一个旋转排序数组，在原地恢复其排序。

说明
什么是旋转数组？
比如，原始数组为[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--]);
        }
    }
};