欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LintCode 31.数组划分

程序员文章站 2022-03-05 13:45:06
...

描述:

给出一个整数数组 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操作,对于快速排序,脑海中一定要有的一个图如下:

LintCode 31.数组划分

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

 

上一篇: 背包问题 LintCode

下一篇: 七、模块