第一周题单
F - 找球号 (计蒜客 - T1413)
在蒜国里流行着一种游戏。
游戏规则为:在一堆球中,每个球上都有一个整数编号 i ( 0 ≤ i ≤ 1 0 9 ) i (0≤i≤10^9) i(0≤i≤109),编号可重复,现在说一个随机整数 k ( 0 ≤ k ≤ 1 0 9 + 100 ) k (0≤k≤10^9+100) k(0≤k≤109+100),判断编号为k的球是否在这堆球中(存在为“YES”,否则为“NO”),先答出者为胜。现在有一个人想玩玩这个游戏,但他又很懒。他希望你能帮助他
取得胜利。
Input
输入格式第一行有两个整数 m , n ( 0 ≤ m ≤ 1 0 6 , 0 ≤ n ≤ 1 0 6 ) m, n(0≤m≤10^6,0≤n≤10^6) m,n(0≤m≤106,0≤n≤106);
m m m表示这堆球里有 m m m个球, n n n表示这个游戏进行 n n n次。
接下来输入 m + n m+n m+n个整数,前 m m m个分别表示这 m m m个球的编号 i i i,后 n n n个分别表示每次游戏中的随机整数 k k k。
Output
输出“YES”或“NO”。
Simple1
Input
6 4
23 34 46 768 343 343
2 4 23 343
Output
NO
NO
YES
YES
题解过程
本题就是一个查找问题,但是如果用数组直接查找的话时间复杂度是
O
(
n
)
O(n)
O(n),本题的数据量最大为
1
0
12
10^{12}
1012,会超时。所以要采用二分查找的方法,时间复杂度为
O
(
l
o
g
2
n
)
O(log_2 n)
O(log2n)
当然也可以用set完成本题,set是用二叉搜索树实现的,访问元素的时间复杂度为
O
(
l
o
g
2
n
)
O(log_2 n)
O(log2n)
二分↓
代码部分(cpp)#include <bits/stdc++.h>
using namespace std;
bool cmp(int a, int b)
{
return a < b;
}
bool search(int balls[], int n, int t)
{
int l = 0, r = n - 1;
while (l <= r)
{
int mid = l + (r - l) / 2;
if (balls[mid] < t)
l = mid + 1;
else if (balls[mid] > t)
r = mid - 1;
else
return 1;
}
return 0;
}
void solution()
{
int m, n;
cin >> m >> n;
int balls[m];
for (int i = 0; i < m; i++)
cin >> balls[i];
sort(balls, balls + m, cmp);
for (int i = 0; i < n; i++)
{
int k;
cin >> k;
if (search(balls, m, k))
cout << "YES\n";
else
cout << "NO\n";
}
}
int main(int argc, char const *argv[])
{
solution();
return 0;
}
set↓
代码部分(cpp)#include <cstdio>
#include <iostream>
#include <algorithm>
#include <set>
using namespace std;
void solution()
{
set<int> balls;
int m, n;
cin >> m >> n;
for (int i = 0; i < m; i++)
{
int temp;
scanf("%d", &temp);
balls.insert(temp);
}
for (int i = 0; i < n; i++)
{
int k;
scanf("%d", &k);
if (balls.find(k) == balls.end())
printf("NO\n");
else
printf("YES\n");
}
}
int main(int argc, char const *argv[])
{
solution();
return 0;
}
刚开始学习,欢迎指正