Ai
1 Star 2 Fork 5

LilithSangreal/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
find-k-th-smallest-pair-distance.cpp 931 Bytes
一键复制 编辑 原始数据 按行查看 历史
kamyu 提交于 2017-10-30 00:02 +08:00 . Create find-k-th-smallest-pair-distance.cpp
// Time: O(nlogn + nlogw), n = len(nums), w = max(nums) - min(nums)
// Space: O(1)
// Binary search with sliding window solution
class Solution {
public:
int smallestDistancePair(vector<int>& nums, int k) {
sort(nums.begin(), nums.end());
int left = 0, right = nums.back() - nums.front() + 1;
while (left < right) {
const auto mid = left + (right - left) / 2;
if (possible(mid, nums, k)) {
right = mid;
} else {
left = mid + 1;
}
}
return left;
}
private:
bool possible(const int guess, const vector<int>& nums, const int k) {
int count = 0, left = 0;
for (int right = 0; right < nums.size(); ++right) {
while ((nums[right] - nums[left]) > guess) {
++left;
}
count += right - left;
}
return count >= k;
}
};
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/LilithSangreal/LeetCode-Solutions.git
git@gitee.com:LilithSangreal/LeetCode-Solutions.git
LilithSangreal
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助