题目连接: AcWing 113. 特殊排序
问题描述
分析
这道题可以用二分来解决,因为最多能进行10000次询问,所以询问的时间复杂度应该为
n
l
o
g
n
n\ log n
n logn,这道题说大小关系不具有传递性,这是什么意思呢?
有传递性:
a>b,b>c,可以推出a>c
无传递性:
a>b,b>c,不能推出a>c,a与c的大小关系未知
这样的情况也是能进行二分的
在一个序列f[l,r]中找到数 x应该插到的位置,mid=l+r>>1
(1)如果f[mid]>x,即compare(x,f[mid])==true,那么在[l,mid]这个区间内一定有合法位置让x插入,
推论:
首先,f[mid-1]<f[mid],这两的大小关系是已知的,x<f[mid],如果x>f[mid-1],那么x可以插入到mid-1和mid之间
如果x<f[mid-1],那么按照上面的思路考虑x与f[mid-2]的关系,依次类推,可以知道,如果x小于[l,mid]的每一个数,那么x可以插在l位置上,否则,x在[l,mid]中有位置插入
这种方法可以保障当f[mid]>x时,在[l,mid]上一定能找到合法位置让x插入
(2)如果f[mid]>x,即compare(x,f[mid])==false,那么在[mid,r]这个区间内一定有合法位置让x插入,
推论:
按照上面的分析方式
首先,f[mid]<f[mid+1],这两的大小关系是已知的,x>f[mid],如果x<f[mid+1],那么x可以插入到mid和mid+1之间
如果x>f[mid+1],那么按照上面的思路考虑x与f[mid+2]的关系,依次类推,可以知道,如果x大于[mid,r]的每一个数,那么x可以插在r+1位置上,否则,x在[mid,r]中有位置插入
这种方法可以保障当f[mid]<x时,在[mid,r]上一定能找到合法位置让x插入
代码如下
这代码的时间复杂度是
n
2
n^2
n2因为类似插入排序,插入算法是
n
2
n^2
n2,但是查询算法是
n
l
o
g
n
nlogn
nlogn的
// Forward declaration of compare API.
// bool compare(int a, int b);
// return bool means whether a is less than b.
class Solution {
public:
vector<int> specialSort(int N) {
vector<int> res;
res.push_back(1);
for(int i=2;i<=N;i++){
int l=0,r=res.size()-1;
while(l<r){
int mid=l+r+1>>1;
if(compare(res[mid],i)) l=mid;
else r=mid-1;
}
res.push_back(i);
for(int j=res.size()-2;j>r;j--) swap(res[j],res[j+1]);
if(compare(i,res[r])) swap(res[r],res[r+1]);
}
return res;
}
};
还有种更简单的做法,而且时间复杂度是
n
l
o
g
n
nlogn
nlogn,就是用归并排序来做,我们可以发现归并排序也不需要传递的大小关系
代码如下
class Solution {
public:
vector<int> specialSort(int N) {
vector<int>v;
for(int i=1;i<=N;++i)v.push_back(i);
stable_sort(v.begin(),v.end(),compare);
return v;
}
};