暑假算法刷题日记 Day 10-CSDN博客

import java.util.*;


public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        String s[] = new String[n];
        for (int i = 0; i < n; i++) {
            s[i] = sc.next();
        }
        Arrays.sort(s,new Comparator<String>() {
            public int compare(String o1, String o2) {
                return (o2+o1).compareTo(o1+o2);
            }
        });
        for (int i = 0; i < n; i++) {
            System.out.print(s[i]);
        }
    }
}

055、求第k小的数

题目描述

输入 nn（1≤n<50000001≤n<5000000 且 nn 为奇数）个数字 aiai（1≤ai<1091≤ai<109），输出这些数字的第 kk 小的数。最小的数是第 00 小。

请尽量不要使用 nth_element 来写本题，因为本题的重点在于练习分治算法。

输入格式

无

输出格式

无

输入输出样例

输入 #1复制

5 1
4 3 2 1 5

输出 #1复制

核心思路

这个题看起来并没有什么难度，但是题目给的样例卡时间，最后两个数据量太大，经典的快排和归并都会超时间。

我们回顾一下快排的思路，确定枢轴的过程是实质上就是把这个元素放到其排序后的正确位置上去，其实就是把第k小的数放在下标为k的位置上，根据这个思路我们可以在快排的代码上做优化。

我们在快排的基础上，确定好枢轴位置后，判断该位置是否是k，是的话直接结束程序，不是的话继续往后，为了节约时间，我们只排序k所在的那个区间。

代码

#include <iostream>  
#include <vector>  
using namespace std;  
  const int N=5e6 +10;
 int nums[N];
void quickSort(int left, int right, int k) {  
    // 判断排序区间长度  
    if (right <= left) {  
        if (right == left && left == k) {  
            cout << nums[k] << endl;  
            exit(0);  
        }  
        return;  
    }  
  
    // 选择基准值（这里使用最左边的值）  
    int pivot = nums[left];  
    int i = left, j = right;  
    while (i < j) {  
        // 从右向左找到第一个小于等于pivot的元素  
        while (nums[j] >= pivot && i < j) j--;  
        // 交换  
        if (i < j) nums[i++] = nums[j];  
  
        // 从左向右找到第一个大于pivot的元素  
        while (nums[i] <= pivot && i < j) i++;  
        if (i < j) nums[j--] = nums[i];  
    }  
    nums[i] = pivot;  
  
    // 判断基准值是否为目标位置  
    if (i == k) {  
        cout << nums[k] << endl;  
        exit(0);  
    }  
  
    // 递归排序  
    if (k < i) quickSort(left, i - 1, k);  
    if (k > i) quickSort(i + 1, right, k);  
}  
  
int main() {  
    int n, k;  
    cin >> n >> k;  
    
    for (int i = 0; i < n; i++) {  
        scanf("%d",&nums[i]);  
    }  
  
    
    
    quickSort( 0, n - 1, k); 
  
    return 0;  
}