P7910 [CSP-J 2021]-插入排序

[CSP-J 2021] 插入排序

题目描述

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

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

这下面是 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$ 次操作，操作共两种，参数分别如下：

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

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

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

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

输入格式

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

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

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

输出格式

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

样例 #1

样例输入 #1

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

样例输出 #1

1
1
2

提示

【样例解释 #1】

在修改操作之前，假设 H 老师进行了一次插入排序，则原序列的三个元素在排序结束后所处的位置分别是 $3,2,1$。

在修改操作之后，假设 H 老师进行了一次插入排序，则原序列的三个元素在排序结束后所处的位置分别是 $3,1,2$。

注意虽然此时 $a_2=a_3$，但是我们不能将其视为相同的元素。

【样例 #2】

见附件中的 sort/sort2.in 与 sort/sort2.ans。

该测试点数据范围同测试点 $1\sim 2$。

【样例 #3】

见附件中的 sort/sort3.in 与 sort/sort3.ans。

该测试点数据范围同测试点 $3\sim 7$。

【样例 #4】

见附件中的 sort/sort4.in 与 sort/sort4.ans。

该测试点数据范围同测试点 $12\sim 14$。

【数据范围】

对于所有测试数据，满足 $1\le n\le 8000$，$1\le Q\le 2\times {10}^5$，$1\le x\le n$，$1\le v,a_i\le 10^9$。

对于所有测试数据，保证在所有 $Q$ 次操作中，至多有 $5000$ 次操作属于类型一。

各测试点的附加限制及分值如下表所示。

测试点	$n\le$	$Q\le$	特殊性质
$1\sim 4$	$10$	$10$	无
$5\sim 9$	$300$	$300$	无
$10\sim 13$	$1500$	$1500$	无
$14\sim 16$	$8000$	$8000$	保证所有输入的 $a_i,v$ 互不相同
$17\sim 19$	$8000$	$8000$	无
$20\sim 22$	$8000$	$2\times 10^5$	保证所有输入的 $a_i,v$ 互不相同
$23\sim 25$	$8000$	$2\times 10^5$	无

---

题目链接
P7910 [CSP-J 2021] 插入排序
​

开始讲解

定义

首先我们要定义一个结构体数组，有两个元素，分别是需要存储的量和它的序号，用来存储在排序前他们分别占的位置，必须定义！数组的长度应是8000，因为n的值最大是8000，我们多开五个空间。

struct node
{
    int num,id;//数组里的值原和序号
}a[8005];

还需要定义一个b数组，用来存放他们排序后的序号，b应是一个一维数组。

int b[8005];

输入

输入的时候应该边输入并计算它的序号，序号就是循环的i。

for(i=1;i<=n;i++)//输入元素
{
    cin>>a[i].num;
    a[i].id=i;//计算标号
}

初始化

操作二需要把a数组进行排序，可操作二的次数不确定，会超时，所以我们应该把排序的任务放在操作一之后以及输入之后。

所干的事情应该是把a数组的元素进行排序，可以用sort函数，（详细使用方法见文章讲解C++中的sort(快速排序函数)），并且把a数组的下标存入b数组里。

sort(a+1,a+1+n,cmp);//排序
for(i=1;i<=n;i++)
b[a[i].id]=i;//将a数组的序号顺序存入b数组

进行操作

首先进行q次循环，可以用while。

每次循环，先输入操作类型，然后判断操作类型是一还是二。

操作一

输入x和v先进行修改的任务，只需将原来a数组的第x个元素改v，再重新进行排序，再把存好的a数组的下标存起来，但排序a数组时不能再使用sort函数了，因为这样会超时，那么应该怎么优化呢，不难看出，a数组的元素中，只有一个修改的数是乱的，其余的数排序都是升序排好的，所以我们可以使用两个for循环，一个正着遍历一遍，另一个反着遍历一遍，如果这个数字比前面一个数小，则反着遍历的for可以把它存到适当的位置；反之，如果这个数字比后一个数还要大，则正序遍历的for可以把它存到适当的位置。而这两个for是顺序关系，时间复杂度只有2n，远比sort要快的多。

cin>>x>>v;
a[b[x]].num=v;//改变数组的元素
for(i=1;i<=n-1;i++)
if(cmp(a[i],a[i+1])==0)
swap(a[i],a[i+1]);//向后排序
for(i=n;i>=2;i--)
if(cmp(a[i-1],a[i])==0)
swap(a[i-1],a[i]);//向前排序
for(i=1;i<=n;i++)
b[a[i].id]=i;//对下标重新排序

操作二

如果是操作二的话，只需要输出b数组里的下标就行了。

cin>>x;//序号为x的数排序后在哪里
cout<<b[x]<<endl;//输出

最终代码+注释

#include<bits/stdc++.h>
using namespace std;
struct node
{
    int num,id;//数组里的值原和序号
}a[8005];
bool cmp(node a,node b)//排序规则
{
    if(a.num==b.num)//值相同
    return a.id<b.id;//根据序号升序排
    return a.num<b.num;//值不相同按序号升序排
}
int main()
{
    int b[8005],n,q,i;
    cin>>n>>q;//输入几个数和几次操作
    for(i=1;i<=n;i++)//输入元素
    {
        cin>>a[i].num;
        a[i].id=i;//计算标号
    }
    sort(a+1,a+1+n,cmp);//排序
    for(i=1;i<=n;i++)
    b[a[i].id]=i;//将a数组的序号顺序存入b数组
    while(q--)//进行q次操作
    {
        int s,x,v;
        cin>>s;//第几种操作
        if(s==1)//操作一
        {
            cin>>x>>v;
            a[b[x]].num=v;//改变数组的元素
            for(i=1;i<=n-1;i++)
            if(cmp(a[i],a[i+1])==0)
			swap(a[i],a[i+1]);//向后排序
			for(i=n;i>=2;i--)
            if(cmp(a[i-1],a[i])==0)
			swap(a[i-1],a[i]);//向前排序

            for(i=1;i<=n;i++)
            b[a[i].id]=i;//对下标重新排序
        }
        else//操作二
        {
            cin>>x;//序号为x的数排序后在哪里
            cout<<b[x]<<endl;//输出
        }
    }
    return 0;
}

​