LintCode 31. partitionArray 數組劃分

31. partitionArray 數組劃分

題目連接

lintcode 31 partitionArray 數組劃分

題目描述

給出一個整數數組 nums 和一個整數 k。劃分數組（即移動數組 nums 中的元素），使得：

全部小於k的元素移到左邊
全部大於等於k的元素移到右邊
返回數組劃分的位置，即數組中第一個位置 i，知足 nums[i] 大於等於 k。

1. 注意事項

你應該真正的劃分數組 nums，而不單單只是計算比 k 小的整數數，若是數組 nums 中的全部元素都比 k 小，則返回 nums.length。

1. 樣例

給出數組 nums = [3,2,2,1] 和 k = 2，返回 1.

1. 挑戰

使用 O(n) 的時間複雜度在數組上進行劃分。

分析

簡單來講就是快排跑一圈

代碼

github

class Solution {
public:
    /**
     * @param nums: The integer array you should partition
     * @param k: An integer
     * @return: The index after partition
     */
    int partitionArray(vector<int> &nums, int k) {
        // write your code here
        int i,j;
        int n = nums.size();
        if(n==0) return 0;
        i = 0;
        j=n-1;
        while(i<j)
        {
            while(i<=j && nums[i]<k) i++;
            while(i<=j && nums[j]>=k) j--;
            if(i<j)
            {
                int tmp = nums[i];
                nums[i] = nums[j];
                nums[j] = tmp;
                i++;
                j--;
            }
        }
        return i;
    }
};

測試數據