113. 特殊排序

题目

有 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;
    }
};