【题解】【模拟】—— [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 的示范代码：

为了帮助小 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

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$	无

思路1.直接模拟

1.1.思路解析

    观察题目，最简单粗暴的方法就是顺着题意直接枚举。但只需要仔细分析就可以知道，$O(q{n}^2+x)$，其中$x$是执行操作$1$的次数，由于较小，可以忽略不计。

    统计下来就是$O(q{n}^2)$，带入计算得$2\ast 10^5\ast 8000^2$，十分惊人。需要换一种思路。

    代码就不贴出来了。

思路2.暴力枚举

2.1.思路解析

    看看题目标签，有一个枚举。可以想到这种思路。

    找出位置为$x$的数排序后的位置，就是统计有多少个数在排序后在$x$前面。由于插入排序算是稳定的，排序后在$x$前面的数有这样两个特征之一。

1. 值小于位置为$x$的数。
2. 值等于位置为$x$的数且位置比$x$小。

2.2.参考代码

#include<bits/stdc++.h>
using namespace std;
int n,q,a[8010];
int main()
{
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    while(q--)
    {
    	int opt;
    	scanf("%d",&opt);
    	if(opt==1)//执行修改操作
    	{
    		int x,v;
    		scanf("%d%d",&x,&v);
    		a[x]=v;//直接修改 
		}
		else
		{
			int x,ans=0;
			scanf("%d",&x);
			for(int i=1;i<=n;i++)//枚举 
			    if(a[i]<a[x]||(a[i]==a[x]&&i<=x))
			        ans++;
			printf("%d\n",ans);
		}
	}
	return 0;
}

解法1.重新设计模拟算法

1.思路解析

    上面的代码还是不能通过此题，需要重新构思。

    观察题目，可以发现：

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

    于是，我们可以对操作$2$优化，操作$1$时间复杂度大一些都没关系。

    定义一个数组ans，分别储存数组$a$中的元素排序后的位置。例如ans[i]表示数组$a$中排序后编号为i的元素排序后的位置。

    我们可以直接使用数组$a$存储排序后的结果。那么怎么定位排序后位置为$x$的数呢？就是用我们前面定义的ans数组，使用a[ans[x]]表示排序后位置为$x$的数。

    但是，为了方便更新ans数组，我们定义一个结构体，储存每一个数，如下：

struct num
{
	int id,val;//储存原数组中的编号和值
}a[MAXN];
bool cmp(num a,num b)//比较a是否小于b
{
	if(a.val==b.val)return a.id<b.id;
	return a.val<b.val;
}

    我们可以使用for(int i=1;i<=n;i++)ans[a[i].id]=i;来更新每个数排序后位置。

    最后，执行操作$1$是还是不能使用$sort$。烤炉到每次修改都只有修改元素前后的元素位置有细微改变，其他的基本都不动我们只需要用一个$O(n)$的for循环，如果修改的值比原值大，就往后移，否则往前移。没必要像sort一样“大动干戈”。

    具体细节请看代码

2.AC代码

#include<bits/stdc++.h>
using namespace std;
#define MAXN 8010
int n,q,ans[MAXN];//用ans[i]存储数组a中的元素排序后的位置
struct num
{
	int id,val;//储存原数组中的编号和值
}a[MAXN];
bool cmp(num a,num b)//比较a是否小于b
{
	if(a.val==b.val)return a.id<b.id;
	return a.val<b.val;
}
int main()
{
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i].val);
		a[i].id=i;
	}
    sort(a+1,a+n+1,cmp);//先排一次序
    for(int i=1;i<=n;i++)ans[a[i].id]=i;//初始化
    while(q--)
    {
    	int opt;
    	scanf("%d",&opt);
    	if(opt==1)//执行修改操作 
    	{
    		int x,v;
    		scanf("%d%d",&x,&v);
    		if(v>a[ans[x]].val)//如果此元素修改之后变大了,向后浮动 
    		{
    			a[ans[x]].val=v;
    			for(int i=ans[x];i<n;i++)
    			    if(cmp(a[i+1],a[i]))//如果a[i]>a[i+1]
    			        swap(a[i],a[i+1]);
    			    else break;
			}
			else//否则就是修改之后变小了 
			{
				a[ans[x]].val=v;
				for(int i=ans[x]-1;i>=1;i--)
    			    if(cmp(a[i+1],a[i]))//如果a[i]>a[i+1]
    			        swap(a[i],a[i+1]);
    			    else break;
			}
			for(int i=1;i<=n;i++)ans[a[i].id]=i;//重新记录答案 
		}
		else//执行查询操作 
		{
			int x;
			scanf("%d",&x);
			printf("%d\n",ans[x]);//直接输出答案 
		}
	}
	return 0;
}

喜欢就订阅此专辑吧！

【蓝胖子编程教育简介】
蓝胖子编程教育，是一家面向青少年的编程教育平台。平台为全国青少年提供最专业的编程教育服务，包括提供最新最详细的编程相关资讯、最专业的竞赛指导、最合理的课程规划等。本平台利用趣味性和互动性强的教学方式，旨在激发孩子们对编程的兴趣，培养他们的逻辑思维能力和创造力，让孩子们在轻松愉快的氛围中掌握编程知识，为未来科技人才的培养奠定坚实基础。

欢迎扫码关注蓝胖子编程教育