这是一个很水的题目。标准二分查找模板题。
题目相关
题目链接
计蒜客,https://nanti.jisuanke.com/t/T1873。
题目描述
蒜头君所在的学校又迎来了一年一度的开花活动,有 n 名学生被评为文学优秀奖,m 名学生被评为体育优秀奖。现已知两个奖项获奖同学的编号,每个同学都有唯一的编号。只有同时被评为文学优秀奖和体育优秀奖的学生才能开花,蒜头君想知道开花的名单,请你帮他统计一下。
输入格式
第一行两个整数 n,m (1 ≤ n, m ≤ 10^5),分别表示文学优秀奖和体育优秀奖的获奖人数。
第二行 n 个不同的整数,表示获得文学优秀奖的同学编号。
第二行 m 个不同的整数,表示获得体育优秀奖的同学编号。
所有编号为正整数且不超过 10^9。
输出格式
一行若干个空格分隔的整数,表示开花的同学编号,按文学优秀奖的先后次序输出。
样例输入
4 4
5 1 7 3
2 3 4 1
样例输出
1 3
题目分析
题目要求
输出开花的同学编号,按文学优秀奖的先后次序输出。只有同时被评为文学优秀奖和体育优秀奖的学生才能开花。
样例数据分析
根据样例,我们知道获得文学优秀奖的同学有 5 1 7 3,获得体育优秀奖的同学有 2 3 4 1。那么同时获得文学优秀奖和体育优秀奖的同学是 1 和 3,题目要求按照文学优秀奖获奖顺序输出。故而输出的结果是 3 1。
编程思路
由于数据最大可能是 10^5,故算法必须优于 O(nlogn)。同时我们需要在体育优秀奖里搜索获得文学优秀奖名单。所以可以考虑用二分查找,这样算法的复杂度为 O(logn)。
1、读入文学优秀奖同学,存入数组 a。
2、读入体育优秀奖同学,存入数组 b。
3、排序数组 b,因为要在体育优秀奖里搜索文学优秀奖名单。
4、遍历数组 a,查看 a[i] 是否在数组 b 中。
AC 参考代码
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1e5+6;
const int MAXM = 1e5+6;
int a[MAXN] = {};//文学优秀奖
int b[MAXM] = {};//体育优秀奖
int main() {
//读入数据
int n,m,i;
scanf("%d%d", &n, &m);
//读入文学优秀奖
for (i=0; i<n; i++) {
scanf("%d", &a[i]);
}
//读入体育优秀奖
for (i=0; i<m; i++) {
scanf("%d", &b[i]);
}
//排序
sort(b, b+m);
for (i=0; i<n; i++) {
if (binary_search(b, b+m, a[i])) {
printf("%d ", a[i]);
}
}
printf("\n");
return 0;
}
时间复杂度
f(n)=n+m+logn,也就是 O(n)。