【CSP-J 2021】插入排序

https://www.luogu.com.cn/problem/P7910?contestId=54994

分析

排序做成$O(n)$复杂度即可。

#include <bits/stdc++.h>
using namespace std;  
#define LL long long
struct ss
{
	int index;
	int num;
}A[8010],B[8010];
int n,Q,f,x,y,a[8010];
void sort_bub()
{
	for(int i=1;i<=n;i++)
	{
		for(int j=i;j>=2;j--)
		{
			if(B[j].num<B[j-1].num)swap(B[j],B[j-1]);
		}
	} 
	for(int i=1;i<=n;i++)A[B[i].index].index=i;
}
int main()
{
//	freopen("sort3.in","r",stdin);
//	freopen("output.txt","w",stdout);
	scanf("%d%d",&n,&Q);
	for(int i=1;i<=n;i++)scanf("%d",&A[i].num),B[i].num=A[i].num,B[i].index=i;
	sort_bub();  //冒泡排序 并且 给A数组赋值 
	while(Q--)
	{
		scanf("%d",&f);
		if(f==1) //修改 
		{
			scanf("%d%d",&x,&y);
			A[x].num=y;  //改掉A数组相应位置的数字 
			int p=A[x].index,btmp=B[p].num;
			B[p].num=y;  //改掉B数组相应位置的数字
			//更新B数组 
			if(btmp<y) //变大了 
			{
				int i=p+1;
				while(i<=n&&B[i].num<y)
				{
					A[B[i].index].index--;
					swap(B[i-1],B[i]);
					i++;
				}
				while(i<=n&&B[i].num==y&&x>B[i].index)
				{
					A[B[i].index].index--;
					swap(B[i-1],B[i]);
					i++;
				}
				A[B[i-1].index].index=i-1;
			}
			else if(btmp>y)
			{
				int i=p-1;
				while(i>=1&&B[i].num>y)
				{
					A[B[i].index].index++;
					swap(B[i],B[i+1]);
					i--;
				}
				while(i>=1&&B[i].num==y&&x<B[i].index)
				{
					A[B[i].index].index++;
					swap(B[i],B[i+1]);
					i--;
				}
				A[B[i+1].index].index=i+1;
			}
		}
		else  //查找 
		{
			scanf("%d",&x);
			printf("%d\n",A[x].index); 
		}
	}
	return 0;
}