题目
有 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]
代码
// 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 ++){//遍历需要插入的数
// 从已有的序列中查找i需要插入的位置,即满足与当前数比较
// 符合 compare(k,i) && compare(i,k+1)
// 思路:当前数k与i相比,如果满足compare(a,i) 那么画一条上升趋势的线段 如果满足compare(i,b),那么画一条下降趋势的线段,需要查找的即使b这样的极大值点,i需要插在b后面一个位置
// 对已有序列进行二分查找 i需要插入到查找出来的边界后面一个位置
int l = 0, r = res.size()-1;
while(l < r){
int mid = l + (r-l+1>>1);
if(compare(res[mid], i)) l = mid;//以mid为根据,需要的答案在区间[mid,r],则更新左边界l为mid
else r = mid - 1;//以mid为根据,需要的答案在区间[l,mid-1],则更新右边界r为mid-1,此模板需注意防止死循环
}
res.resize(res.size()+1);
//把r+1~res.size()-2位置的元素都往后移动一位,空出res[r+1]的位置
for(int j = res.size()-2; j >= r+1; j--) res[j+1] = res[j];
//把i插入到res[r+1]的位置,即r的后面一个位置
res[r+1] = i;
//如果r为0,即二分查找出的边界为0,而插入的位置使r+1,那么需要再特判一下r与r+1位置元素的关系,如果需要交换则交换
if(compare(res[1], res[0]))swap(res[0],res[1]);
return res;
}
};