Partition Array

最新推荐文章于 2022-05-08 00:35:38 发布

原创最新推荐文章于 2022-05-08 00:35:38 发布 · 419 阅读

0 ·

CC 4.0 BY-SA版权

strStr 同时被 2 个专栏收录

360 篇文章

订阅专栏

两根指针

14 篇文章

订阅专栏

本文介绍了一种使用双指针技术的数组分区算法实现，并详细解释了处理边界情况的方法。该算法通过从两端向中间遍历的方式，实现了高效的数组元素重新排列。

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

这道题用两根指针，向中间逼近。

一定要注意的是corner case时的下标处理。

/** 
     *@param nums: The integer array you should partition
     *@param k: As description
     *return: The index after partition
     */
    public int partitionArray(int[] nums, int k) {
	    //write your code here
	    if ((nums == null || nums.length == 0) || (nums.length == 1 && nums[0] > k)) {
	        return 0;
	    }
	    if (nums.length == 1 && nums[0] > k) {
	        return 1;
	    }
	    int left = 0, right = nums.length - 1;
	    int index = 0;
	    while (left < right) {
	        if (nums[left] < k && nums[right] >= k) {
	            left++;
	            right--;
	            index++;
	        } else if (nums[left] >= k && nums[right] >= k) {
	            right--;
	        } else if (nums[left] < k && nums[right] < k) {
	            left++;
	            index++;
	        } else {
	            int temp = nums[left];
	            nums[left] = nums[right];
	            nums[right] = temp;
	            left++;
	            right--;
	            index++;
	        }
	    }
	    if (left == nums.length - 1) {
	        //1 return index++;
	        return ++index;
	    } else {
	        //return index;
	        if (nums[left] < k) {
	            return ++index;
	        } else 
	        return index;
	    }
	   //return index;
    }