Lintcode 恢復旋轉陣列

題目

給定一個旋轉排序陣列，在原地恢復其排序。

樣例

什麼是旋轉陣列？

比如，原始陣列為[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]

class Solution {
public:


    void Reverse(vector<int> &nums,int l,int h)
    {
        int temp=0;
        while(l<h)
        {
            temp=nums[l];
            nums[l]=nums[h];
            nums[h]=temp;
            l++;
            h--;
        }

    }
    void recoverRotatedSortedArray(vector<int> &nums) {
        // write your code here
        if(nums.empty()||nums.size()==1) return;
        int i=0;

        //找到分界點
        while(i<nums.size()-1)
        {
            if(nums[i]<=nums[i+1])
                i++;
            else
                break;
        }

        //反轉
        if(i<nums.size()-1)
        {
            Reverse(nums,0,i);
            Reverse(nums,i+1,nums.size()-1);
            Reverse(nums,0,nums.size()-1);
        }

    }
};

STL擴充

C++的STL提供了reverse,可以用於反轉陣列。

reverse(引數1，引數2);

引數1：要反轉陣列第一個元素的迭代器；
引數2：要反轉陣列的最後一個元素的下一個位置。

上題程式碼可改寫成

class Solution {
public:

    void recoverRotatedSortedArray(vector<int> &nums) {
        // write your code here
        if(nums.empty()||nums.size()==1) return;
        int i=0;

        //找到分界點
        while(i<nums.size()-1)
        {
            if(nums[i]<=nums[i+1])
                i++;
            else
                break;
        }

        //反轉
        if(i<nums.size()-1)
        {
            //STL::reverse
            reverse(nums.begin(),nums.begin()+i+1);
            reverse(nums.begin()+i+1,nums.end());
            reverse(nums.begin(),nums.end());
        }

    }
};