在 C++ 中查詢由給定 N 個範圍生成的序列中的第 k 個元素


在這個問題中,我們給定了一個 N 個整數範圍,這些範圍在區間 L - R 之間,作為一個矩陣 range[N][2] 和一個整數 k。我們的任務是*找到由給定 N 個範圍生成的序列中的第 k 個元素*。

讓我們舉一個例子來理解這個問題,

Input : ranges[][] = {{1, 3}, {5, 7}}, k = 4
Output : 5

**說明** -

The series is {1, 2, 3, 5, 6, 7}
The 4th element is 5.

解決方案方法

解決這個問題的一個簡單方法是為給定的範圍建立一個整數序列,然後找到其陣列中索引 k 處的元素。

示例

程式說明我們解決方案的工作原理

#include <iostream>
using namespace std;
int findKthSmallestEleSeries(int n, int k, int range[][2]){
   int rangeVal[10000];
   int rangeSize = 0;
   for(int i = 0; i < n; i++){
      for(int j = range[i][0]; j <= range[i][1]; j++){
         rangeVal[rangeSize] = j;
         rangeSize++;
      }
   }
   return rangeVal[k-1];
}
int main(){
   int L[] = { 1, 5 };
   int R[] = { 3, 8 };
   int range[][2] = {{1, 3}, {5, 8}};
   int n = sizeof(L) / sizeof(int);
   int k = 4;
   cout<<k<<"-th element of the series generated by ranges is "<<
   findKthSmallestEleSeries(n, k, range); 
   return 0;
}

輸出

4-th element of the series generated by ranges is 5

這種方法很好,但如果範圍太大,可能會導致記憶體問題。因此,我們需要推匯出一種比儲存陣列所有元素更好的方法。

另一種方法

解決此問題的另一種方法是使用二分查詢和計數器陣列。計數器陣列將儲存到給定範圍為止的整數計數,count[i] = 到第 i 個範圍為止的字元計數。

然後對於 k,我們將找到第 i 個範圍,其中包含第 k 個最小值。

然後我們將在這個範圍內找到該值。

示例

程式說明我們解決方案的工作原理

#include <iostream>
using namespace std;
int findKthSmallestEleSeries(int n, int k, int range[][2]){
   int start = 1;
   int end = n;
   int count[n + 1];
   count[0] = 0;
   for (int i = 0; i < n; i++)
      count[i + 1] = count[i] + (range[i][1] - range[i][0]) + 1;
   int index = -1;
   int mid;
   while (start <= end) {
      mid = (start + end) / 2;
      if (count[mid] > k) {
         index = mid;
         end = mid - 1;
      }
      else if (count[mid] < k) 
         start = mid + 1;
      else {
         index = mid;
         break;
      }
   }
   start = range[index - 1][0];
   end = range[index - 1][1];
   int indexK = k - count[index - 1];
   while (start <= end) {
      mid = (start + end) / 2;
      if ((mid - range[index - 1][0]) + 1 == indexK) {
         return mid;
      }
      else if ((mid - range[index - 1][0]) + 1 > indexK)
         end = mid - 1;
      else
         start = mid + 1;
   }
   return -1;
}
int main(){
   int L[] = { 1, 5 };
   int R[] = { 3, 8 };
   int range[][2] = {{1, 3}, {5, 8}};
   int n = sizeof(L) / sizeof(int);
   int k = 4;
   cout<<k<<"-th element of the series generated by ranges is "<<
   findKthSmallestEleSeries(n, k, range);
   return 0;
}

輸出

4-th element of the series generated by ranges is 5

更新於: 2022年1月28日

200 次檢視

開啟你的 職業生涯

透過完成課程獲得認證

開始學習
廣告