CSP-J复赛冲刺必刷题 | P7910 插入排序

学习C++从娃娃抓起！记录下CSP-J备考学习过程中的题目，记录每一个瞬间。

附上汇总贴：CSP-J复赛冲刺必刷题 | 汇总_热爱编程的通信人的博客-CSDN博客

---

【题目描述】

插入排序是一种非常常见且简单的排序算法。小 Z 是一名大一的新生，今天 H 老师刚刚在上课的时候讲了插入排序算法。

假设比较两个元素的时间为 O(1)，则插入排序可以以 O(n²) 的时间复杂度完成长度为 n 的数组的排序。不妨假设这 n 个数字分别存储在 a1,a2,…,an 之中，则如下伪代码给出了插入排序算法的一种最简单的实现方式：

这下面是 C/C++ 的示范代码：

for (int i = 1; i <= n; i++)
        for (int j = i; j >= 2; j--)
                if (a[j] < a[j-1]) {
                        int t = a[j-1];
                        a[j-1] = a[j];
                        a[j] = t;
                }

这下面是 Pascal 的示范代码：

for i:=1 to n do
    for j:=i downto 2 do
        if a[j]<a[j-1] then
            begin
                t:=a[i];
                a[i]:=a[j];
                a[j]:=t;
            end;

为了帮助小 Z 更好的理解插入排序，小 Z 的老师 H 老师留下了这么一道家庭作业：

H 老师给了一个长度为 n 的数组 a，数组下标从 1 开始，并且数组中的所有元素均为非负整数。小 Z 需要支持在数组 a 上的 Q 次操作，操作共两种，参数分别如下：

1 x v：这是第一种操作，会将 aa 的第 x 个元素，也就是 ax 的值，修改为 v。保证 1≤x≤n，1≤v≤10^9。注意这种操作会改变数组的元素，修改得到的数组会被保留，也会影响后续的操作。

2 x：这是第二种操作，假设 H 老师按照上面的伪代码对 a 数组进行排序，你需要告诉 H 老师原来 a 的第 x 个元素，也就是 ax，在排序后的新数组所处的位置。保证 1≤x≤n。注意这种操作不会改变数组的元素，排序后的数组不会被保留，也不会影响后续的操作。

H 老师不喜欢过多的修改，所以他保证类型 1 的操作次数不超过 5000。

小 Z 没有学过计算机竞赛，因此小 Z 并不会做这道题。他找到了你来帮助他解决这个问题。

【输入】

第一行，包含两个正整数 n,Q，表示数组长度和操作次数。

第二行，包含 n 个空格分隔的非负整数，其中第 i 个非负整数表示 ai。

接下来 Q 行，每行 2∼3 个正整数，表示一次操作，操作格式见【题目描述】。

【输出】

对于每一次类型为 2 的询问，输出一行一个正整数表示答案。

【输入样例】

3 4
3 2 1
2 3
1 3 2
2 2
2 3

【输出样例】

1
1
2

【代码详解】

#include <bits/stdc++.h>
using namespace std;
struct Node {
    int id, num;
}a[10000];
int n, q, b[10000], op, x, v;
bool cmp(Node &x, Node &y)
{
    if (x.num==y.num) return x.id<y.id;
    return x.num<y.num;
}
int main()
{
    cin >> n >> q;
    for (int i=1; i<=n; i++) {
        cin >> a[i].num;
        a[i].id = i;
    }
    sort(a+1, a+n+1, cmp);  //a数组排序后成新的a数组
    for (int i=1; i<=n; i++) b[a[i].id] = i; //如a=[10,6,9,7,8]，即a[1]=10,a[2]=6,a[3]=9,a[4]=7,a[5]=8，
                                             //排序后顺序为a[2],a[4],a[5],a[3],a[1]，用新的数组b记录，b的下标为a是id，值为b的id
                                             //b[2]=1,b[4]=2,b[5]=3,b[3]=4,b[1]=5
    while (q--) {
        cin >> op;
        if (op==2) {
            cin >> x;
            cout << b[x] << endl; //得到排序后原来a中的第x个元素在b中的位置，如x=4，b[4]=2，即原来a中的第个元素，在b中为第2个元素
        } else {
            cin >> x >> v;
            int t = b[x];  //改变原来a中的第x个元素的值，如x=5，b[5]=3，则改变的是排序后a数组中第3个元素的值
            a[t].num = v;  
            for (int i=t; i>1; i--) {
                if (a[i].num < a[i-1].num || (a[i].num==a[i-1].num&&a[i].id<a[i-1].id)) {
                    b[a[i].id]--, b[a[i-1].id]++;
                    swap(a[i], a[i-1]);
                }
            }
            for (int i=t; i<n; i++) {
                if (a[i].num>a[i+1].num || (a[i].num==a[i+1].num&&a[i].id>a[i+1].id)) {
                    b[a[i].id]++, b[a[i+1].id]--;
                    swap(a[i], a[i+1]);
                } 
            }
        }
    }
    return 0;
}

【运行结果】

3 4
3 2 1
2 3
1
1 3 2
2 2 
1
2 3
2