[Aizu] ITP2_8_C: Map: Range Search

前言

ITP系列的题目, 只是为了了解基础的用法的, 所以为了提高效率, 不再记录完整的内容, 只侧重于本题中所使用到的一些知识

题目

传送门: ITP2_8_C: Map: Range Search

求解

用法

STL中map是通过红黑树实现的(不明觉厉).
详细请参阅文章C++中的hash_map和map的区别 - 追逐更好的自己

map的常用方法(只涉及到本题的部分)

方法名	描述	示例
insert((key, val))	向map中插入元素(key, val)	my_map.insert(make_pair(key, val));
erase(key)	删除含有key的元素	my_map.erase(key);
find(key)	搜索与key一致的元素, 并返回指向该元素的迭代器, 如果不存在, 则返回指向末尾的迭代器end()	map<string, int>::iterator it = my_map.find(key);
lower_bound(key)	返回指向第一个不小于key的元素的迭代器	it1 = my_map.lower_bound(key);
upper_bound(key)	返回指向第一个大于key的元素的迭代器	it2 = my_map.upper_bound(key);
count(key)	返回与key键匹配的元素数	if (my_map.count(key)) {...}

编码

自己尝试使用STL的代码

#include <bits/stdc++.h>
using namespace std;

int main(void) {
	ios::sync_with_stdio(false);
	cin.tie(0);

	int q, com, x;
	string key, L, R;
	map<string, int> my_map;
	map<string, int>::iterator it1, it2, it;

	cin >> q;
	while (q--) {
		cin >> com >> key;
		switch (com) {
			case 0:
				cin >> x;
				if (my_map.end() != my_map.find(key)) {
					my_map.erase(key);
				}
				my_map.insert(make_pair(key, x));
				break;
			case 1:
				it = my_map.find(key);
				if (it != my_map.end()) {
					cout << it->second << endl;
				} else {
					cout << 0 << endl;
				}
				break;
			case 2:
				my_map.erase(key);
				break;
			case 3:
				L = key;
				cin >> R;
				it1 = my_map.lower_bound(L);
				it2 = my_map.upper_bound(R);
				for (it = it1; it != it2; it++) {
					cout << it->first << " " << it->second << endl;
				}
				break;
		}
	}
}

别人使用STL的代码

传送门: #3431194 Solution for ITP2_8_C by shioree

#include <bits/stdc++.h>
using namespace std;
 
int main() {
  int num_query; cin >> num_query;
  map<string, int> dic;
   
  while (num_query--) {
    int command; scanf("%d", &command);
    char key[20]; scanf("%s", key);
     
    switch (command) {
      case 0:
        int value; scanf("%d", &value);
        if (dic.count(key)) dic.at(key) = value;
        else dic.insert(make_pair(key, value));
        break;
         
      case 1:
        if (dic.count(key)) printf("%d
", dic.at(key));
        else printf("%d
", 0);
        break;
         
      case 2:
        dic.erase(key);
        break;
         
      case 3:
        char key2[20]; scanf("%s", key2);
        map<string, int>::iterator it = dic.lower_bound(key);
        map<string, int>::iterator last = dic.upper_bound(key2);
        while(it != last) {
          printf("%s %d
", ((*it).first).c_str(), (*it).second);
          it++;
        }
        break;
    }
  }
}

记录

应该尝试使用count()方法去快速判断有无, 而不是使用find(), 它之后还需要一次与end()的比较, 而count()的返回值可以直接作为判断的条件