LintCode 31.数组划分

描述:

给出一个整数数组 nums 和一个整数 k。划分数组(即移动数组 nums 中的元素),使得:

  • 所有小于k的元素移到左边
  • 所有大于等于k的元素移到右边

返回数组划分的位置,即数组中第一个位置 i,满足 nums[i] 大于等于 k

样例:

Example 1:

Input:
[],9
Output:
0

Example 2:

Input:
[3,2,2,1],2
Output:1
Explanation:
the real array is[1,2,2,3].So return 1

挑战

使用 O(n) 的时间复杂度在数组上进行划分。

注意事项

你应该真正的划分数组 nums,而不仅仅只是计算比 k 小的整数数,如果数组 nums 中的所有元素都比 k 小,则返回 nums.length。

代码:

    private void swap(int x,int y){
        int tmp=x;
        x=y;
        y=tmp;
    }
    
    private int __quickSort(int[] nums,int left,int right,int k){
        //if(left>right)return -1;
        int j=left-1;
        for(int i=left;i<=right;i++){
            if(nums[i]<k){
                swap(nums[j+1],nums[i]);
                j++;
            }
            
        }
        return j+1;
    }
    
    public int partitionArray(int[] nums, int k) {
        // write your code here
        if(nums==null||nums.length==0){
            return 0;
        }
        else{
            int left=0;
            int right=nums.length-1;
            return __quickSort(nums,0,right,k);
        }
    }

补充说明:

这个题目其实就是快速排序的Partition操作,对于快速排序,脑海中一定要有的一个图如下:

已知数组的左起始位置和数组的右终止位置,从做开始向后遍历,这道题目不需要选第一个为标定元素,我们只需要将大于等于v的元素都放入后半部分即可,全部遍历完成后,j+1的位置即为我们要找的第一个位置的元素。

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值