【题解】【模拟】—— [CSP-J 2021] 插入排序
[CSP-J 2021] 插入排序
题目描述
插入排序是一种非常常见且简单的排序算法。小 Z 是一名大一的新生,今天 H 老师刚刚在上课的时候讲了插入排序算法。
假设比较两个元素的时间为 O ( 1 ) \mathcal O(1) O(1),则插入排序可以以 O ( n 2 ) \mathcal O(n^2) O(n2) 的时间复杂度完成长度为 n n n 的数组的排序。不妨假设这 n n n 个数字分别存储在 a 1 , a 2 , … , a n a_1, a_2, \ldots, a_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 的示范代码:
为了帮助小 Z 更好的理解插入排序,小 Z 的老师 H 老师留下了这么一道家庭作业:
H 老师给了一个长度为 n n n 的数组 a a a,数组下标从 1 1 1 开始,并且数组中的所有元素均为非负整数。小 Z 需要支持在数组 a a a 上的 Q Q Q 次操作,操作共两种,参数分别如下:
1 x v 1~x~v 1 x v:这是第一种操作,会将 a a a 的第 x x x 个元素,也就是 a x a_x ax 的值,修改为 v v v。保证 1 ≤ x ≤ n 1 \le x \le n 1≤x≤n, 1 ≤ v ≤ 1 0 9 1 \le v \le 10^9 1≤v≤109。注意这种操作会改变数组的元素,修改得到的数组会被保留,也会影响后续的操作。
2 x 2~x 2 x:这是第二种操作,假设 H 老师按照上面的伪代码对 a a a 数组进行排序,你需要告诉 H 老师原来 a a a 的第 x x x 个元素,也就是 a x a_x ax,在排序后的新数组所处的位置。保证 1 ≤ x ≤ n 1 \le x \le n 1≤x≤n。注意这种操作不会改变数组的元素,排序后的数组不会被保留,也不会影响后续的操作。
H 老师不喜欢过多的修改,所以他保证类型 1 1 1 的操作次数不超过 5000 5000 5000。
小 Z 没有学过计算机竞赛,因此小 Z 并不会做这道题。他找到了你来帮助他解决这个问题。
输入格式
第一行,包含两个正整数 n , Q n, Q n,Q,表示数组长度和操作次数。
第二行,包含 n n n 个空格分隔的非负整数,其中第 i i i 个非负整数表示 a i a_i ai。
接下来 Q Q Q 行,每行 2 ∼ 3 2 \sim 3 2∼3 个正整数,表示一次操作,操作格式见【题目描述】。
输出格式
对于每一次类型为 2 2 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 3, 2, 1 3,2,1。
在修改操作之后,假设 H 老师进行了一次插入排序,则原序列的三个元素在排序结束后所处的位置分别是 3 , 1 , 2 3, 1, 2 3,1,2。
注意虽然此时 a 2 = a 3 a_2 = a_3 a2=a3,但是我们不能将其视为相同的元素。
【样例 #2】
见附件中的 sort/sort2.in
与 sort/sort2.ans
。
该测试点数据范围同测试点 1 ∼ 2 1 \sim 2 1∼2。
【样例 #3】
见附件中的 sort/sort3.in
与 sort/sort3.ans
。
该测试点数据范围同测试点 3 ∼ 7 3 \sim 7 3∼7。
【样例 #4】
见附件中的 sort/sort4.in
与 sort/sort4.ans
。
该测试点数据范围同测试点 12 ∼ 14 12 \sim 14 12∼14。
【数据范围】
对于所有测试数据,满足 1 ≤ n ≤ 8000 1 \le n \le 8000 1≤n≤8000, 1 ≤ Q ≤ 2 × 10 5 1 \le Q \le 2 \times {10}^5 1≤Q≤2×105, 1 ≤ x ≤ n 1 \le x \le n 1≤x≤n, 1 ≤ v , a i ≤ 1 0 9 1 \le v,a_i \le 10^9 1≤v,ai≤109。
对于所有测试数据,保证在所有 Q Q Q 次操作中,至多有 5000 5000 5000 次操作属于类型一。
各测试点的附加限制及分值如下表所示。
测试点 | n ≤ n \le n≤ | Q ≤ Q \le Q≤ | 特殊性质 |
---|---|---|---|
1 ∼ 4 1 \sim 4 1∼4 | 10 10 10 | 10 10 10 | 无 |
5 ∼ 9 5 \sim 9 5∼9 | 300 300 300 | 300 300 300 | 无 |
10 ∼ 13 10 \sim 13 10∼13 | 1500 1500 1500 | 1500 1500 1500 | 无 |
14 ∼ 16 14 \sim 16 14∼16 | 8000 8000 8000 | 8000 8000 8000 | 保证所有输入的 a i , v a_i,v ai,v 互不相同 |
17 ∼ 19 17 \sim 19 17∼19 | 8000 8000 8000 | 8000 8000 8000 | 无 |
20 ∼ 22 20 \sim 22 20∼22 | 8000 8000 8000 | 2 × 1 0 5 2 \times 10^5 2×105 | 保证所有输入的 a i , v a_i,v ai,v 互不相同 |
23 ∼ 25 23 \sim 25 23∼25 | 8000 8000 8000 | 2 × 1 0 5 2 \times 10^5 2×105 | 无 |
思路1.直接模拟
1.1.思路解析
观察题目,最简单粗暴的方法就是顺着题意直接枚举。但只需要仔细分析就可以知道, O ( q n 2 + x ) O(qn^2+x) O(qn2+x),其中 x x x是执行操作 1 1 1的次数,由于较小,可以忽略不计。
统计下来就是 O ( q n 2 ) O(qn^2) O(qn2),带入计算得 2 ∗ 1 0 5 ∗ 800 0 2 2*10^5*8000^2 2∗105∗80002,十分惊人。需要换一种思路。
代码就不贴出来了。
思路2.暴力枚举
2.1.思路解析
看看题目标签,有一个枚举
。可以想到这种思路。
找出位置为
x
x
x的数排序后的位置,就是统计有多少个数在排序后在
x
x
x前面。由于插入排序算是稳定的
,排序后在
x
x
x前面的数有这样两个特征之一。
- 值小于位置为 x x x的数。
- 值等于位置为 x x x的数且位置比 x 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 Q Q 次操作中,至多有
5000
次操作属于类型一。
于是,我们可以对操作 2 2 2优化,操作 1 1 1时间复杂度大一些都没关系。
定义一个数组ans
,分别储存数组
a
a
a中的元素排序后的位置。例如ans[i]
表示数组
a
a
a中排序后编号为i
的元素排序后的位置。
我们可以直接使用数组
a
a
a存储排序后的结果。那么怎么定位排序后位置为
x
x
x的数呢?就是用我们前面定义的ans
数组,使用a[ans[x]]
表示排序后位置为
x
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
1
1是还是不能使用
s
o
r
t
sort
sort。烤炉到每次修改都只有修改元素前后的元素位置有细微改变,其他的基本都不动我们只需要用一个
O
(
n
)
O(n)
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;
}
喜欢就订阅此专辑吧!
【蓝胖子编程教育简介】
蓝胖子编程教育,是一家面向青少年的编程教育平台。平台为全国青少年提供最专业的编程教育服务,包括提供最新最详细的编程相关资讯、最专业的竞赛指导、最合理的课程规划等。本平台利用趣味性和互动性强的教学方式,旨在激发孩子们对编程的兴趣,培养他们的逻辑思维能力和创造力,让孩子们在轻松愉快的氛围中掌握编程知识,为未来科技人才的培养奠定坚实基础。
欢迎扫码关注蓝胖子编程教育