特殊排序
题目描述
有N个元素,编号1.2..N,每一对元素之间的大小关系是确定的,关系不具有传递性。
也就是说,元素的大小关系是N个点与N*(N-1)/2条有向边构成的任意有向图。
然而,这是一道交互式试题,这些关系不能一次性得知,你必须通过不超过10000次提问来获取信息,每次提问只能了解某两个元素之间的关系。
现在请你把这N个元素排成一行,使得每个元素都小于右边与它相邻的元素。
你可以通过我们预设的bool函数compare来获得两个元素之间的大小关系。
例如,编号为a和b的两个元素,如果元素a小于元素b,则compare(a,b)返回true,否则返回false。
将N个元素排好序后,把他们的编号以数组的形式输出,如果答案不唯一,则输出任意一个均可。
数据范围
1 <= N <= 1000
输入样例
[[0, 1, 0], [0, 0, 0], [1, 1, 0]]
输出样例
[3, 1, 2]
代码详解
用插入排序,假设前i-1个数已经排好,将第i个数插入到里面。
代码
// 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 >q;
q.push_back(1);
for(int i=2;i<=N;++i)
{
int l=0,r=q.size()-1;//二分下标位置
while(l<r)
{
int mid=(l+r+1)>>1;//考虑到compare的返回要求,小于才会返回true,所以这里用书上讲的第二种写法,求解小于i的最大的数
if(compare(q[mid],i)) l=mid;
else r=mid-1;
}
q.push_back(i);//现将这个元素插入末尾
for(int j=q.size()-2;j>l;j--)swap(q[j],q[j+1]);//交换元素
if(compare(i,q[r])) swap(q[r],q[r+1]);//如果i<=q[r],就说明这段里面所有数都比i大,那么再交换
}
return q;
}
};