LeetCode 215. Kth Largest Element in an Array-快速排序的灵活使用

在LeetCode上看到一个题目，第一眼看的感觉就是排序解决，然后思考了下堆排序不太合适，而快排速度很快，然后在快速排序的基础上进行了优化，写下来记录一下。

题目

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.
For example,
Given [3,2,1,5,6,4] and k = 2, return 5.

Note:
You may assume k is always valid, 1 ≤ k ≤ array’s length.
题目分析
- 并不需要全部排序
- 注意小细节，如数据量为1的时候

---

代码（非递归）

 public static int findKthLargest(int[] nums, int k) {
            //把大的放左边，小的放右边，分割成两部分。当k小于中间数，往左查找，大于则往右查找，灵活运用快排

          int left=0;
          int mid=nums.length;
          int right=nums.length-1; //使用最右边的作为划分点

          while(true){

              int i=left-1;
              int j=right;
             mid=nums[right]; //记录划分点

          while(i!=j){
              while(nums[++i]>mid&&i<j) //左边找到小于标记点的
                  if(i==right) break; //一直没找到
             while(nums[--j]<mid&&i<j)//右边找到大于标记点的
              if(j==left) break;
             if(i>=j) break;      //如果此时i>j跳出     
             //交换两个找到的标记点
              int temp=nums[i];
              nums[i]=nums[j];
              nums[j]=temp;


          }

     //将划分点交换
          int temp=nums[i];
          nums[i]=nums[right];
          nums[right]=temp;
//i+1是因为i是从0开始的，当找到的划分点的位置小于我要的k位置，那么就以标记点为最左边元素，在他的右边继续找，而左边的可以不管了，与我们要找的无关，相反如果要找的划分点大于k往右找，而如果等于，那么这个元素就是我们要找的了，退出循环结束
          if(i+1<k){
              left=i+1;
          }else if(i+1>k){
              right=i-1;
          }else{
              break;  
          }
            if(left>=right) break;
          }

          return nums[k-1];
        }

递归的方式就不贴了，比较简单。