给出一个整数数组 nums 和一个整数 k。划分数组(即移动数组 nums 中的元素),使得:
nums 和一个整数 k。划分数组(即移动数组 nums 中的元素),使得:
- 所有小于k的元素移到左边
- 所有大于等于k的元素移到右边
返回数组划分的位置,即数组中第一个位置 i,满足 nums[i] 大于等于 k。
注意事项
你应该真正的划分数组 nums,而不仅仅只是计算比 k 小的整数数,如果数组 nums 中的所有元素都比 k 小,则返回 nums.length。
样例
给出数组 nums = [3,2,2,1]
和 k = 2
,返回 1
.
思路:
左右各一个游标,不断向中间靠拢,左边的找到比k大或者相等的数为止,右边找到比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.length==0) return 0; if(nums.length==1) return 1; int i,j; int flag=0; for(i=0,j=nums.length-1;i!=j&&i<nums.length&&j>=0;){ while(nums[i]<k&&i<nums.length-1){ i++; } while(nums[j]>=k&&j>0){ j--; } if(i>=j) break; int temp=nums[i]; nums[i]=nums[j]; nums[j]=temp; flag=1; } if(i==0) return 0; if(flag==0) return i+1; return i; } }
如果有所帮助,脸皮厚求个赞~
此文章仅代表自己(本菜鸟)学习积累记录,或者学习笔记,如有侵权,请联系作者删除。人无完人,文章也一样,文笔稚嫩,在下不才,勿喷,如果有错误之处,还望指出,感激不尽~
技术之路不在一时,山高水长,纵使缓慢,驰而不息。
公众号:秦怀杂货店