题目描述
狙击手小可的射击能力非常厉害,小达的视力非常厉害。小达观察敌人个数和位置,小可射击,他们俩配合,可以一枪同时击中多个敌人。
1 OOO
2 OOOO
3 O
4 OO
如上图,每个“O”代表一个敌人。如果小可想要打倒 3 个敌人就在 1 位置射击,想要打倒 4 个敌人就在 2 位置射击。
现在他想要打倒 m 个敌人。小达告诉小可每个位置的敌人,请你给小可一个狙击位置。
输入格式
第一行包含一个正整数 n,表示位置数。
第二行包含 n 个正整数,第 i 个数,表示第 i 个位置的敌人数,保证各个位置的敌人数不同。
第三行包含一个正整数 Q,表示小可发球的次数。
第四行开始,一共有 Q 行,每行包含一个正整数 m,表示小可需要打倒 m 个敌人。
输出格式
共 Q 行。每行包含一个整数,第 i 行的整数表示小可第 i 次的狙击位置。若无解,则输出 0。
样例 #1
样例输入 #1
5
1 2 4 3 5
2
4
7
5个位置
每个位置分别有敌人 1 2 4 3 5 个
小可要射击两次
第一次希望打中4个敌人
第二次希望打中7个敌人
样例输出 #1
3
0
提示
【数据范围】
对于 50%的数据,1≤n,Q≤1000,1≤,m≤
对于 100%的数据,1≤n,Q≤100000,1≤,m≤
每行敌人个数不相同
思路
1.写个结构体用来记录数的下标,防止排序后顺序乱套
2.然后排序把数组排成单调数组方便二分查找
3.二分查找(模板)
代码:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6+1;
struct node{
int name;
int ID;
}a[maxn];
bool cmp (node a, node b) {//写的那个cmp排序规则的参数就的是你的结构体类型而不是int(看下面例子)
return a.name < b.name;
}
/*
struct k {
int name;
int ID;
} a[100100];
bool cmp(k a, k b){
return a.name<b.name;
}
*/
int find (int x ,int l ,int r){
if(a[l].name==x) return l;
if(l >= r) return -1;
int mid = l + ((r-l) >> 1);
if(a[mid].name >= x) return find(x,l,mid);
else return find(x,mid+1,r);
}
int main() {
int n,m,q,k=0;
cin >> n;
for (int i = 1; i<=n; i++ ){
cin >>a[i].name;
a[i].ID=i;
}
sort(a+1,a+n+1,cmp);
cin>>m;
for (int i = 1; i<=m; i++ ){//find得出来的结果就是答案,直接判度你的a[ans].name是不是等于q,如果是直接输出a[ans].ID就行不是就输出0
cin >> q ;
int ans = find(q , 1 ,n+1);
if(a[ans].name==q){//不需要变回排序前这个题其实就是二分查找一个数的下标
cout<<a[ans].ID<<endl;
}else{
cout<<k<<endl;
}
}
return 0;
}