LintCode 31. 数组划分

最新推荐文章于 2021-04-22 07:25:34 发布

老汉在此

最新推荐文章于 2021-04-22 07:25:34 发布

阅读量180

点赞数

CC 4.0 BY-SA版权

分类专栏： java开发文章标签： java lintcode

本文链接：https://blog.youkuaiyun.com/laisinanvictor/article/details/79948499

java开发专栏收录该内容

63 篇文章

订阅专栏

本文介绍了一种数组划分算法，该算法可以将整数数组中的元素按照指定整数k进行划分，确保所有小于k的元素位于数组左侧，而所有大于等于k的元素位于右侧，并返回划分后数组中第一个大于等于k的元素的位置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给出一个整数数组 nums 和一个整数 k。划分数组（即移动数组 nums 中的元素），使得：

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

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

public class Solution {
    /**
     * @param nums: The integer array you should partition
     * @param k: An integer
     * @return: The index after partition
     */
    public int partitionArray(int[] nums, int k) {
        // write your code here
        if (nums == null) {
            return -1;
        }

        int start = 0;
        int end = nums.length;
        int[] tmp = new int[nums.length+1];
        tmp[0] = k;
        for (int i = 0; i < nums.length; i++) {
            tmp[i+1] = nums[i];
        }
        while (start < end) {
            while (start < end && tmp[end] >= k) {
                end--;
            }
            if (start < end) {
                tmp[start++] = tmp[end];
            }
            while (start < end && tmp[start] < k) {
                start++;
            }
            if (start < end) {
                tmp[end--] = tmp[start];
            }
        }
        return start;
    }
}