CSP/信奥赛C++刷题训练:经典二分例题(4)
保龄球
题目描述
DL 算缘分算得很烦闷,所以常常到体育馆去打保龄球解闷。因为他保龄球已经打了几十年了,所以技术上不成问题,于是他就想玩点新花招。
DL 的视力真的很不错,竟然能够数清楚在他前方十米左右每个位置的瓶子的数量。他突然发现这是一个炫耀自己好视力的借口——他看清远方瓶子的个数后从某个位置发球,这样就能打倒一定数量的瓶子。
-
◯ ◯ ◯ \bigcirc \bigcirc \bigcirc ◯◯◯
-
◯ ◯ ◯ ◯ \bigcirc \bigcirc \bigcirc\ \bigcirc ◯◯◯ ◯
-
◯ \bigcirc ◯
-
◯ ◯ \bigcirc\ \bigcirc ◯ ◯
如上图,每个 “ ◯ \bigcirc ◯” 代表一个瓶子。如果 DL 想要打倒 3 3 3 个瓶子就在 1 1 1 位置发球,想要打倒 4 4 4 个瓶子就在 2 2 2 位置发球。
现在他想要打倒 m m m 个瓶子。他告诉你每个位置的瓶子数,请你给他一个发球位置。
输入格式
第一行包含一个正整数 n n n,表示位置数。
第二行包含 n n n 个正整数 a i a_i ai ,表示第 i i i 个位置的瓶子数,保证各个位置的瓶子数不同。
第三行包含一个正整数 Q Q Q,表示 DL 发球的次数。
第四行至文件末尾,每行包含一个正整数 m m m,表示 DL 需要打倒 m m m 个瓶子。
输出格式
共 Q Q Q 行。每行包含一个整数,第 i i i 行的整数表示 DL 第 i i i 次的发球位置。若无解,则输出 0 0 0。
样例 #1
样例输入 #1
5
1 2 4 3 5
2
4
7
样例输出 #1
3
0
提示
【数据范围】
对于 50 % 50\% 50% 的数据, 1 ≤ n , Q ≤ 1000 , 1 ≤ a i , m ≤ 1 0 5 1 \leq n, Q \leq 1000, 1 \leq a_i, m \leq 10^5 1≤n,Q≤1000,1≤ai,m≤105。
对于 100 % 100\% 100% 的数据, 1 ≤ n , Q ≤ 100000 , 1 ≤ a i , m ≤ 1 0 9 1 \leq n,Q \leq 100000, 1 \leq a_i, m \leq 10^9 1≤n,Q≤100000,1≤ai,m≤109。
用二分求解
满分代码
#include<bits/stdc++.h>
using namespace std;
//二分查找
const int N=1e5+10;
int n,q,m;
struct boal{int data,id;
}a[N];
//排序规则:按瓶子数量排序
bool cmp(boal b1,boal b2){return b1.data<b2.data;
}
int f(int x){int l=1,r=n,mid;while(l<=r){mid=(l+r)/2;if(a[mid].data==x) return mid;else if(a[mid].data>x) r=mid-1;else l=mid+1;}return -1;
}
int main(){cin>>n;for(int i=1;i<=n;i++){cin>>a[i].data;a[i].id=i;}cin>>q;sort(a+1,a+n+1,cmp);while(q--){cin>>m;if(f(m)==-1) cout<<0<<endl;else cout<<a[f(m)].id<<endl;}return 0;
}
文末彩蛋:
点击王老师青少年编程主页有更多精彩内容