01:查找最接近的元素
- 總時間限制:
- 1000ms
- 內存限制:
- 65536kB
- 描述
-
在一個非降序列中,查找與給定值最接近的元素。
- 輸入
-
第一行包含一個整數n,為非降序列長度。1 <= n <= 100000。
第二行包含n個整數,為非降序列各元素。所有元素的大小均在0-1,000,000,000之間。
第三行包含一個整數m,為要詢問的給定值個數。1 <= m <= 10000。
接下來m行,每行一個整數,為要詢問最接近元素的給定值。所有給定值的大小均在0-1,000,000,000之間。 - 輸出
- m行,每行一個整數,為最接近相應給定值的元素值,保持輸入順序。若有多個值滿足條件,輸出最小的一個。
- 樣例輸入
-
3 2 5 8 2 10 5
- 樣例輸出
-
8 5
最基本的二分,找點感覺。
但是有小坑。。
二分邊界問題,我這種二分的寫法,while到最后找到的是兩個最接近的值,二選一。
但是當n=1的時候,最后的兩個值其實是同一個,a[l+1]是不存在的。。。
特判n=1直接輸出即可。
還是就是數組長度不要和尋值個數看錯。。數組是開到10W的,不然OJ返回的結果是TLE。。。
#include <iostream> #include <cstdio> #include <cmath> const int maxn = 100000+10; int a[maxn],n; using namespace std; int search(int x) { int l = 1; int r = n; int ans = -1; int mid = (1+n)/2; while(l < r-1) { if(x < a[mid]) { r = mid; } else { l = mid; } mid = (l+r)/2; } if(abs(a[l]-x)<=abs(a[l+1]-x)&&l<n) ans = a[l]; else { ans = a[l+1]; } // if(x-a[l]>a[l+1]-x&&l<n) // l++; return ans; } int main() { // freopen("in.txt","r",stdin); scanf("%d",&n); for(int i = 1; i <= n; i++) { scanf("%d",&a[i]); } int T; scanf("%d",&T); while(T--) { int t; scanf("%d",&t); if(n==1) { printf("%d\n",a[1]); } else { printf("%d\n",search(t)); } } return 0; }