目录
SGI-STL中关于键值对的定义:
template <class T1, class T2>
struct pair
{
typedef T1 first_type;
typedef T2 second_type;
T1 first;
T2 second;
pair(): first(T1()), second(T2())
{}
pair(const T1& a, const T2& b): first(a), second(b)
{}
};
T: set 中存放元素的类型,实际在底层存储 <value, value> 的键值对。Compare : set 中元素默认按照小于来比较Alloc : set 中元素空间的管理方式,使用 STL 提供的空间配置器管理
set中的底层使用二叉搜索树(红黑树)来实现
set中只放value,但在底层实际存放的是由<value, value>构成的键值对
set中的元素不可以重复。
set中查找某个元素,时间复杂度为:logN
set中的元素不允许修改(因为底层时二叉搜索树)
count和find的作用一样,都是用于查找set中是否存在某个元素。
其实count是为容器multiset设计的,其与set的区别在于multiset允许插入重复元素,此时count会返回被搜索元素的个数。
void test()
{
set<int>s;
for (int i = 1; i < 10; i++)
s.insert(i);
for (int i = 1; i < 10; i++)
{
cout << i;
if (s.count(i))
cout << "is an element of s. \n";
else
cout << "is not an element of s. \n";
}
}
lower_bound返回大于等于目标值的迭代器,upper_bound返回大于目标值的迭代器。这里通常可以配合erase使用。
void test()
{
set<int>s;
for (int i = 1; i < 10; i++)
s.insert(i);
auto it1=s.lower_bound(1);
auto it2=s.upper_bound(8);
s.erase(it1,it2);//左开右闭
for(auto ch:s)
{
cout<<ch<<" ";
}
}
返回值为一个pair类型的变量,pair中存放的有两个变量,第一个为插入数据的迭代器,第二个如果插入成功为true,失败为false。
void test()
{
set<int>s;
for (int i = 1; i < 10; i++)
s.insert(i);
pair<set<int>::iterator,bool> p=s.insert(1);
cout<<*p.first<<" "<<p.second<<endl;
auto p2=s.insert(10);
cout<<*p2.first<<" "<<p2.second<<endl;
}
void test()
{
map<string, string>dict;
dict.insert(pair<string,string>("sort", "排序"));
dict.insert(pair<string,string>("test", "测试"));
for (auto& x : dict)
{
cout << x.first << ":" << x.first << " ";
}
cout << endl;
}
这里也可以使用make_pair.
void test()
{
map<string, string>dict;
dict.insert(make_pair("sort", "排序"));
dict.insert(make_pair("test", "测试"));
for (auto& x : dict)
{
cout << x.first << ":" << x.first << " ";
}
cout << endl;
}
void test()
{
string arr[] = { "西瓜","西瓜","香蕉","苹果","梨" };
map<string, int>s;
for (auto x : arr)
{
s[x]++;
}
for (auto x : s)
{
cout << x.first << ":" << x.second << endl;
}
}
这里数组下标为key,获得值为value。如果key不存在,则插入,并且初始化value,并++,
存在就直接++。
void TestSet()
{
?int array[] = { 2, 1, 3, 9, 6, 0, 5, 8, 4, 7 };
// 注意:multiset在底层实际存储的是<int, int>的键值对
multiset<int> s(array, array + sizeof(array)/sizeof(array[0]));
for (auto& e : s)
cout << e << " ";
cout << endl;
return 0;
}
class Solution {
public:
class Compare
{
public:
? ? ? ? // 在set中进行排序时的比较规则
bool operator()(const pair<string, int>& left, const
pair<string, int>& right)
{
return left.second > right.second;
}
};
vector<string> topKFrequent(vector<string>& words, int k)
{
// 用<单词,单词出现次数>构建键值对,然后将vector中的单词放进去,统计每
个单词出现的次数
map<string, int> m;
for (size_t i = 0; i < words.size(); ++i)
++(m[words[i]]);
// 将单词按照其出现次数进行排序,出现相同次数的单词集中在一块
multiset<pair<string, int>, Compare> ms(m.begin(), m.end());
// 将相同次数的单词放在set中,然后再放到vector中
set<string> s;
size_t count = 0; ? // 统计相同次数单词的个数
size_t leftCount = k;
? ? ? ? vector<string> ret;
for (auto& e: ms)
{
if (!s.empty())
{
// 相同次数的单词已经全部放到set中
if (count != e.second)
{
if (s.size() < leftCount)
{
ret.insert(ret.end(), s.begin(), s.end());
leftCount -= s.size();
s.clear();
}
else
{
break;
}
}
}
count = e.second;
s.insert(e.first);
}
for (auto& e : s)
{
if (0 == leftCount)
break;
ret.push_back(e);
leftCount--;
}
return ret;
}
};
class Solution {
public:
? ?vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
? // 先去重
? ? ? ?set<int> s1;
? ? ? ?for(auto e : nums1)
? ? ? {
? ? ? ? ? ?s1.insert(e);
? ? ? }
? ? ? ?set<int> s2;
? ? ? ?for(auto e : nums2)
? ? ? {
? ? ? ? ? ?s2.insert(e);
? ? ? }
? ? ? ?
? ? ? ?// set排过序,依次比较,小的一定不是交集,相等的是交集
? ? ? ?auto it1 = s1.begin();
? ? ? ?auto it2 = s2.begin();
? ? ? ?vector<int> ret;
? ? ? ?while(it1 != s1.end() && it2 != s2.end())
? ? ? {
? ? ? ? ? ?if(*it1 < *it2)
? ? ? ? ? {
? ? ? ? ? ? ? ?it1++;
? ? ? ? ? }
? ? ? ? ? ?else if(*it2 < *it1)
? ? ? ? ? {
? ? ? ? ? ? ? ?it2++;
? ? ? ? ? }
? ? ? ? ? ?else
? ? ? ? ? {
? ? ? ? ? ? ? ?ret.push_back(*it1);
? ? ? ? ? ? ? ?it1++;
? ? ? ? ? ? ? ?it2++;
? ? ? ? ? }
? ? ? }
? ? ? ?return ret;
? }
};