18. nth-element

description

给定一个长度为 ​ 的整数数组。你的程序需要按照输入顺序处理如下两种操作：

1. 修改。修改操作带有两个参数 ​ 和 ​，表示将数组中下标为 ​ 的元素修改为 ​；

2. 查询。查询操作带有三个参数 ​ 和 ​，表示查询数组中从下标为 ​ 的元素开始到下标为 ​ 的元素为止（两端均包含）的这个子数组中数值从小到大排第 ​ 的整数。

输入描述

测试用例输入第一行为一个整数 ​，表示数组的长度。

第二行为 ​ 个以空格分隔的整数，表示初始数组的每一项。

第三行为一个整数 ​，表示操作的数量。

从第四行开始的 ​ 行，每行包含若干个以空格分隔的整数。其中第一个整数表示操作的类型，1表示修改操作，2表示查询操作；若操作为修改操作，接下来包含两个整数 ​，含义如前述所示；若操作为查询操作，接下来包含三个整数 ​，含义如前述所示。

保证数组中的每个元素在任意时刻均在 ​ 范围内。

保证所有查询中子数组的长度之和不超过 ​。即 ​。

输出描述

对于输入中的每一个查询操作，你的程序应该在单独的一行上输出查询的结果。

样例说明

对于题目中给出的样例，其第一个操作为 2 1 3 2，表示查询下标从 1 - 3 范围内，即数组 [2, 3, 4] 中数值排名第二大的元素，即为 3；第二个操作为 1 2 2，表示修改下标为 2 的元素为 2；第三个操作为2 1 3 2，表示查询下标从 1 - 3 范围内，即数组 [2, 2, 4] （注意此时下标为 2 的元素已经被修改为 2）中数值排名第二大的元素，即为 2。

注意，计算排名时，相同的整数要进行重复计算。例如样例中的 [2, 2, 4]，计算排名时 2 排第 1 和 2，4 排第 3。

	测试输入	期待的输出	时间限制	内存限制	额外进程
测试用例 1	以文本方式显示 5↵ 1 2 3 4 5↵ 3↵ 2 1 3 2↵ 1 2 2↵ 2 1 3 2↵	以文本方式显示 3↵ 2↵	1秒	64M	0

code

1. 对给定范围的所有元素进行排序的话肯定是行不通的，这样会导致TLE ，然后我们看我们要查找的第rank元素和基准之间的关系，对另一部分进行排序，也就是不需要两边都进行排序
2. 然后这样子写出来的程序还是会有两个用例过不掉，时间超时所以我们使用三路快排，三路快排的提出是为了解决数据中含有大量重复元素的情况，原理很简单，直接上代码：

#include <stdio.h>
#include <stdlib.h>

int* arr = NULL;  //数组

// 修改指定索引处为指定值
void Change()
{
    int idx = -1;
    int target = 0;
    scanf("%d", &idx);
    scanf("%d", &target);
    if(idx != -1) {
        arr[idx] = target;
    }
}

void Swap(int& a, int& b)
{
    int tmp = a;
    a = b;
    b = tmp;
}

void recur(int* tmp, int rank, int left, int right)
{
    //基准随机化
    int end = right - left + 1;
    int mid = rand() % end + left;  //随机化处理，生成一个[left，right]的数字
    Swap(tmp[left], tmp[mid]);
    int pivot = tmp[left];  //基准

    int lt = left;  // less than
    int gt = right + 1;  // great than
    int i = left + 1;  //等于部分的指针

    while(i < gt) {
        if(tmp[i] < pivot) {
            Swap(tmp[i], tmp[lt + 1]);
            i++;
            lt++;
        }
        else if(tmp[i] > pivot) {
            Swap(tmp[i], tmp[gt - 1]);
            gt--;
        }
        else
            i++;
    }
    Swap(tmp[left], tmp[lt]);

    if(rank >= gt) {  //在右边，那么左边也不用排序了，反正都大于它
        recur(tmp, rank, gt, right);
    }
    else if(rank < lt) {  //在左边，右边不用管了，只需要针对左边就行了
        recur(tmp, rank, left, lt - 1);
    }
    else {
        printf("%d\n", tmp[rank]);
        return;
    }
}

// 查找某范围内第几大的元素
void Query()
{
    int* tmp = NULL;
    int begin = -1;
    int end = -1;
    int rank = -1;
    scanf("%d", &begin);
    scanf("%d", &end);
    scanf("%d", &rank);

    int len = end - begin + 1;
    tmp = (int*)malloc(sizeof(int) * len);

    for(int i = 0; i < len; i++) {
        tmp[i] = arr[begin + i];
    }

    rank = rank - 1;  //和下标对应上
    recur(tmp, rank, 0, len - 1);

    free(tmp);
}

int main()
{
    int n;
    int T;
    int dowhat;  //进行什么操作
    freopen("file in.txt", "r", stdin);
    scanf("%d", &n);
    arr = (int*)malloc(sizeof(int) * n);
    for(register int i = 0; i < n; i++)
        scanf("%d", &arr[i]);
    scanf("%d", &T);
    while(T--) {
        scanf("%d", &dowhat);  // 2是查询，1是修改
        switch(dowhat) {
            case 1:
                Change();
                break;
            case 2:
                Query();
                break;
            default:
                break;
        }
    }

    free(arr);
    return 0;
}

summary

1. 整理的快速排序算法笔记：https://blog.csdn.net/qq_41680771/article/details/120841246