1.解法一
思路:因为数组为升序排列,所以只需要知道
k
的起始位置和个数,即可知道k
的终止位置(起始为值 + 个数 - 1
)。
#include <iostream>
using namespace std;
const int MAXN = 100005;
typedef pair<int, int> PII;
int value[MAXN];
//cnt[i]表示值为i的起始位置和值为i的个数,其中first表示开始位置,second表示个数
PII cnt[MAXN];
int main() {
int n, q;
cin >> n >> q;
for(int i = 0; i < n; i++) {
cin >> value[i];
if(!cnt[value[i]].second) { //如果second为0,则表示value[i]第一次出现,记录起始位置
cnt[value[i]].first = i;
}
cnt[value[i]].second += 1; //value[i]的个数自增1
}
while(q--) {
int k;
cin >> k;
if(!cnt[k].second) { //如果k的个数为0,则表示没有此数字
cout << "-1 -1" << endl;
}else {
//起始位置为first,终点位置为first + 个数 - 1
int l = cnt[k].first, r = cnt[k].second + cnt[k].first - 1;
cout << l << ' ' << r << endl;
}
}
return 0;
}
2.解法二
思路:题目为有序数组,所以用二分查找比较合适,分别查找起点和终点即可。
#include <iostream>
using namespace std;
const int MAXN = 100005;
int value[MAXN];
int main() {
int n, q;
cin >> n >> q;
for(int i = 0; i < n; i++) {
cin >> value[i];
}
while(q--) {
int k;
cin >> k;
//二分查找起点(第一个不小于k的值)
int l = 0, r = n - 1;
while(l < r) {
int mid = (l + r) >> 1;
if(value[mid] >= k) {
r = mid;
}else {
l = mid + 1;
}
}
if(value[r] != k) { //即未找到k
cout << "-1 -1" << endl;
continue;
}
//输出左值
cout << r << ' ';
//二分查找终点(最后一个不大于k的值)
l = 0, r = n - 1;
while(l < r) {
int mid = (l + r + 1) >> 1;
if(value[mid] <= k) {
l = mid;
}else {
r = mid - 1;
}
}
cout << l << endl;
}
return 0;
}