插入排序是一种非常常见且简单的排序算法。小 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 的示范代码
for i:=1 to n do
for j:=i downto 2 do
if a[j]
为了帮助小 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 的第 xx 个元素,也就是 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 的询问,输出一行一个正整数表示答案。
输入样例
3 4
3 2 1
2 3
1 3 2
2 2
2 3
输出样例
1
1
2
根据题目要求,可以直接模拟两种操作:
p[i]
表示排序后i
位置的数在原数组p[i]
位置上;np[i]
表示原数组i
位置的数在排序后的数组np[i]
位置上除此之外,由于题目中要求操作 2 2 2不会改变数组的元素,排序后的数组不会被保留,也不会影响后续的操作。因此在每次排序前,需要将数组信息备份下来,排序输出后在还原回来。
#include
#include
#include
using namespace std;
const int N = 10010;
//p[i]表示排序后i位置的数在原数组p[i]位置
//np[i]表示原数组i位置的数在排序后的数组np[i]
int a[N], b[N], p[N], np[N], bp[N], bnp[N];
int n, q;
int main()
{
cin >> n >> q;
for(int i = 1; i <= n; i ++)
{
scanf("%d", &a[i]);
np[i] = p[i] = i;
}
while(q --)
{
int op, x, v;
scanf("%d%d", &op, &x);
if(op == 1)
{
scanf("%d", &v);
a[x] = v;
}
else
{
//备份
memcpy(b, a, sizeof a);
memcpy(bp, p, sizeof p);
memcpy(bnp, np, sizeof np);
//插入排序
for(int i = 2; i <= n; i ++)
for(int j = i; j > 1 && a[j] < a[j - 1]; j --)
{
swap(a[j], a[j - 1]);
np[p[j]] = j - 1, np[p[j - 1]] = j;
swap(p[j], p[j - 1]);
}
printf("%d\n", np[x]);
//还原
memcpy(a, b, sizeof b);
memcpy(p, bp, sizeof bp);
memcpy(np, bnp, sizeof bnp);
}
}
}
通过分析插入排序的算法实现可以发现,在排序之后如果修改了某个位置上的数字,只需在该位置向前(或者向后)再调整一遍即可完成排序,时间复杂度为 O ( n ) \mathcal O(n) O(n)。
根据这一发现,那么在每次操作 1 1 1后,都进行一次调整(向前或者向后,二选一),在调整过程中维护好数组 p p p和 n p np np,那么对于操作 2 2 2就可以直接输出np[x]
。
注意:在调整过程中,由于插入排序是稳定的排序算法,那么对于相等的元素要保持之前的先后关系,即原来位置在前的还是在前。
#include
#include
using namespace std;
const int N = 10010;
//p[i]表示排序后i位置的数在原数组p[i]位置
//np[i]表示原数组i位置的数在排序后的数组np[i]
int a[N], p[N], np[N];
int n, q;
//向前调整
void forward(int k)
{
for(int i = k; i > 1 && a[i] <= a[i - 1]; i --)
{
//如果相等,需要判断原来位置的前后关系,从而保证插入排序的稳定性,即原来在前的还是在前
if(a[i] == a[i - 1] && p[i] > p[i - 1]) continue;
swap(a[i], a[i - 1]);
np[p[i]] = i - 1;
np[p[i - 1]] = i;
swap(p[i], p[i - 1]);
}
}
//向后调整
void backward(int k)
{
for(int i = k; i < n && a[i] >= a[i + 1]; i ++)
{
//如果相等,需要判断原来位置的前后关系,从而保证插入排序的稳定性,即原来在后面的还是在后面
if(a[i] == a[i + 1] && p[i] < p[i + 1]) continue;
swap(a[i], a[i + 1]);
np[p[i]] = i + 1;
np[p[i + 1]] = i;
swap(p[i], p[i + 1]);
}
}
int main()
{
cin >> n >> q;
for(int i = 1; i <= n; i ++)
{
scanf("%d", &a[i]);
np[i] = p[i] = i;
}
//插入排序
for(int i = 2; i <= n; i ++) forward(i);
while(q --)
{
int op, x, v;
scanf("%d%d", &op, &x);
if(op == 1)
{
scanf("%d", &v);
int k = np[x];
a[k] = v;
//更新后,向前或向后进行调整
forward(k);
backward(k);
}
else printf("%d\n", np[x]);
}
}