应用二分的思想进行优化
假设当前是第k个点
前k-1个点已经排好序了,通过二分的思想在[0,k-1]找到一个可以插入的位置,然后插入
时间复杂度O(N^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++) { int l=0,r=res.size()-1; //下标从0开始 while(l<r) { int mid=l+r+1>>1; //找到位置:左边全部<=i,右边全部>i if(compare(res[mid],i)) //mid<i l=mid; else r=mid-1; } res.push_back(i); //通过冒泡交换排序,一直移动到应该插入的位置 for(int j=res.size()-2;j>r;j--) swap(res[j+1],res[j]); if(compare(i,res[r])) //如果res小于i,不操作,反之交换 swap(res[r],res[r+1]); //这里不能swap(i),因为i已经在数组中了,再插入会出错 } return res; } };