力扣719.找出第K小的数对距离

力扣719.找出第K小的数对距离

• 二分答案

朴素版

• 双指针遍历数组 超过界限break

  auto check = [&](int mid) -> bool
  {
  	int res=0;
  	for(int i=0;i<n-1;i++)
  		for(int j=i+1;j<n;j++)
  		{
  			if(nums[j] - nums[i] > mid) break;
  			else
  				if(++res >= k) return true;
  		}
  	return false;
  };

优化版

• 双指针遍历 j找到位置以后直接+=j - i - 1个数对

  class Solution {
  public:
      int smallestDistancePair(vector<int>& nums, int k) {
          ranges::sort(nums);
          int n = nums.size();
          auto check = [&](int mid) -> bool
          {
              int res=0;
              for(int i=0,j=0;i<n-1;i++)
              {
                  while(j<n && nums[j] - nums[i] <= mid) j++;
                  res += j - i - 1;
                  if(res >= k) return true;
              }
              return false;
          };
          int l = 0,r = nums[n-1] - nums[0];
          while(l<r)
          {
              int mid = l + r >> 1;
              if(check(mid)) r = mid;
              else l = mid + 1;
          }               
          return l;
      }
  };