您的位置:首页 > 其它

(lintcode)第31题,数组划分

2017-12-19 21:55 369 查看
给出一个整数数组 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 hereif(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;}}

                                            
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  lintcode 数组划分