原题链接:https://codeforces.com/problemset/problem/1945/E
题意翻译
现给定一个长度为 n 的排列 p 以及一个数 x。
定义“在 p 中二分查找 x”为以下操作:
初始时有 l=1,r=n+1。
- 如果当前 r−l=1,退出二分查找,并且认定 pl 为二分查找的结果。
- 定义 m=floor((l+r)/2),floor表示向下取整。
- 如果 pm≤x,将 l 赋值为 m,否则将 r 赋值为 m。
不断重复以上三个操作直到得到结果。
显然当 p 未排序时二分查找的结果不一定为 x,现在你希望进行不超过两次交换操作,使得操作后的排列 p 能使得二分查找的结果为 x。
一次交换操作为:选择 1≤i,j≤n,交换 pi,pj。
你需要给出交换的参数,容易证明两次操作总是足够的。
输入输出描述
输入输出样例
输入
5
6 3
1 2 3 4 5 6
6 5
3 1 6 5 2 4
5 1
3 5 4 2 1
6 3
4 3 1 5 2 6
3 2
3 2 1
输出
0
1
3 4
2
2 4
1 5
2
4 5
2 4
1
1 3
解题思路:
这个题目考察的就是二分查找的实现原理,我们可以考虑分类讨论。
(1)二分查找结束时,pl==x,
- 此时已经符合pl==x的要求,所以不需要进行交换,交换次数为0
(2)二分查找结束时,pl>x
- 如果查找的过程中遇到了p(mid)<=x的情况,根据题目描述的查找方式,那么pl必然会停留在<=x的位置,不可能pl>x,这里最终的pl>x说明查找过程中遇到的p(mid)都是大于x的,所以说我们x并不是某一个p(mid),所以我们直接将x和pl进行交换即可,x没有参与这个过程并且交换之后只有pl变为x,右边的原来遇到的p(mid)都是大于x的,所以并不会影响这个过程,所以可以直接将x和pl进行交换,交换次数为1。
(3)二分查找结束时,pl<x
- 此时如果我们的x并不是某个p(mid),那么直接交换x和pl即可,因为x和pl都是<=x的,所以他们在二分查找过程中带来的效益都是l=m,所以不影响这个过程,那么如果x是某个p(mid),其实也是一样的,虽然此时的x是操作过程中的某个p(mid),但是其实x和pl也都是<=x的,他们在二分查找过程中带来的影响是一样的,就算交换他们也是不会影响到二分查找的过程的,所以直接交换x和pl即可,交换次数为1。
可以发现其实对于(2)(3)俩种情况都是可以直接交换x和pl的,所以我们合并这俩种情况,综合分析我们可以得出如下结果。
- 如果pl==x,那么已经满足要求,交换次数为0
- 如果pl!=x,那么直接交换x和pl即可,交换次数为1
时间复杂度:O(n),找到x所在位置需要遍历一遍数组p。
空间复杂度:O(1),不考虑输入所占空间,只使用了常数个额外变量。
cpp代码如下:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N=2e5+10;
int T,n,x;
int p[N];
void solve()
{
cin>>n>>x;
int pos=0;
for(int i=1;i<=n;i++)
{
cin >> p[i];
if(p[i]==x){
pos=i;
}
}
int l=1,r=n+1;
while(l<r-1){
int mid=l+r>>1;
if(p[mid]<=x)l=mid;
else r=mid;
}
if(p[l]==x)cout<<0<<'\n';
else {
cout<<1<<'\n';
cout<<l<<' '<<pos<<'\n';
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
cin>>T;
while(T--)
{
solve();
}
return 0;
}