演算法競賽入門經典—例題5-1-大理石在哪?
阿新 • • 發佈:2018-12-21
題目描述:
現有N個大理石,每個大理石上寫了一個非負整數、首先把各數從小到大排序;然後回答Q個問題。每個問題問是否有一個大理石寫著某個整數x,如果是,還要回答哪個大理石上寫著x。排序後的大理石從左到右編號為1~N。
(在樣例中,為了節約篇幅,所有大理石的數合併到一行,所有問題也合併到一行。)
樣例輸入:
4 1
2 3 5 1
5
5 2
1 3 3 3 1
2 3
樣例輸出:
CASE# 1:
5 found at 4
CASE# 2:
2 not found
3 found at 3
程式設計提示
- 題目意思已經非常清楚了:先排序,再查詢。使用algorithm標頭檔案中的sort和low_bound很容易完成這兩項操作。
- lower_bound的作用是“查詢大於或者等於X的第一個位置”。
程式碼
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=10000;
int main()
{
int n,q,x,a[maxn],kase=0;
while(scanf("%d%d",&n,&q)==2&&n)
{
printf("CASE# %d:\n",++kase);
for(int i=0;i<n;i++) scanf("%d",& a[i]);
sort(a,a+n);//排序
while(q--)
{
scanf("%d",&x);
int p=lower_bound(a,a+n,x)-a;//在已排序陣列a中尋找x
if(a[p]==x) printf("%d found at %d\n",x,p+1);
else printf("%d not found\n",x);
}
}
return 0;
}