题目描述
请你编写一个程序,判断给定的整数 n 是否存在于给定的集合中。
输入描述
有多组测试数据,第一行有一个整数 k,代表有 k 组测试数据。
每组数据第一行首先是一个正整数 m,表示集合中元素的数量(1 <= m <= 1000)。
接下来一行包含 m 个整数,表示集合中的元素。
最后一行包含一个整数 n,表示需要进行判断的目标整数。
输出描述
包含多组输出,每组输出占一行。
如果集合中存在 m,输出“YES”,否则输出“NO”。
输入示例
2
5
1 2 3 4 5
3
6
1 2 3 4 5 6
7
输出示例
YES
NO
题目分析
哈希表常使用的数据结构有数组、set集合、map映射
set
和数学中的集合一样,C++中的集合set用于允许存储一组不重复的元素, 并且元素的值按照有序排列, set基于红黑树实现,支持高效的关键字查询操作, 可以用来检查一个给定关键字是否在set中。
无序集合unordered-set类似于集合(Set),但不会按照元素的值进行排序,而是由哈希函数的结果决定的。
multiset 则是一个用于存储一组元素,允许元素重复,并按照元素的值进行有序排列的集合。
Set的使用
使用集合set需要先引入头文件
// 引入<unordered_set>头文件
#include <unordered_set>
// 引入set头文件
#include <set>
创建一个集合的写法如下
// 创建一个存储整数的无序集合
unordered_set<int> mySet;
// 创建一个存储整数的set
set<int> mySet;
// 创建一个存储整数的 multiset
multiset<int> myMultiSet;
想要向集合中插入元素需要使用insert()方法
// 向集合中插入元素
mySet.insert(1);
mySet.insert(2);
mySet.insert(3);
想要往集合中删除元素需要使用erase方法
mySet.erase(1);
创建一个集合,将输入的数据使用insert()方法插入到集合中
判断某个元素是否在集合中,需要使用到find()方法,如果找到了,返回指向元素的迭代器,如果不存在,则最终指向end()的迭代器,通过判断find()的返回结果可以判定元素是否在集合中。
完整代码
#include <iostream>
#include <unordered_set>int main () {int k;std::cin >> k;while (k--) {int m;std::cin >> m;std::unordered_set<int> unset;int val;while (m--) {std::cin >> val;unset.insert(val);}int n;std::cin >> n;if(unset.find(n) != unset.end()) {std::cout << "YES" << std::endl;}else {std::cout << "NO" << std::endl;}}
}
迭代器
迭代器iterator提供了一种类似指针的接口,可以用来遍历访问容器(比如数组、集合)中的元素,并执行各种操作。
可以理解为,迭代器和下标运算符的作用一样,用来访问容器中的元素,并且迭代器可以从一个元素移动到另外一个元素。
迭代器都拥有名为begin()和end()的成员,表示指向第一个元素和最后一个元素的下一个元素的迭代器(尾后迭代器),如果容器为空,则begin和end返回的是同一个迭代器。
可以使用比较运算符来判断两个迭代器是否相等,如果迭代器想要从一个元素移动到另外一个元素,可以使用递增++运算符和递减–运算符,表示向前(后)移动一个位置。
通过解引用*可以获取迭代器所指的对象,下面的示例表示了vector的遍历。
#include <iostream>
#include <vector>
using namespace std; int main() {vector<int> myVector = {1, 2, 3, 4, 5};// 使用迭代器遍历容器// vector<int>::iterator it 用于创建一个能读取<vector>int 元素的迭代器it,最初指向begin()// ++it表示迭代器的移动for (vector<int>::iterator it = myVector.begin(); it != myVector.end(); ++it) {cout << *it << " "; // 通过解引用获取迭代器所指的对象}cout << endl; return 0;
}
r.begin(); it != myVector.end(); ++it) {cout << *it << " "; // 通过解引用获取迭代器所指的对象}cout << endl; return 0;
}