洛谷-set and map-P5266 【深基17.例6】学籍管理

【深基17.例6】学籍管理

题目描述

您要设计一个学籍管理系统，最开始学籍数据是空的，然后该系统能够支持下面的操作（不超过 $10^5$ 条）：

• 插入与修改，格式1 NAME SCORE：在系统中插入姓名为 NAME(由字母和数字组成不超过 20 个字符的字符串，区分大小写) ，分数为 $\text{SCORE}$（$0<\text{SCORE}<2^{31}$） 的学生。如果已经有同名的学生则更新这名学生的成绩为 SCORE。如果成功插入或者修改则输出OK。
• 查询，格式2 NAME：在系统中查询姓名为 NAME 的学生的成绩。如果没能找到这名学生则输出Not found，否则输出该生成绩。
• 删除，格式3 NAME：在系统中删除姓名为 NAME 的学生信息。如果没能找到这名学生则输出Not found，否则输出Deleted successfully。
• 汇总，格式4：输出系统中学生数量。

输入格式

输出格式

样例 #1

样例输入 #1

5
1 lxl 10
2 lxl
3 lxl
2 lxl
4

样例输出 #1

OK
10
Deleted successfully
Not found
0

思路

本题主要可以通过map来解决，map<key,value> 的两个值，通过key来记录name，value来记录score。
首先插入与修改只需要直接将名字和姓名直接插入就行。
其次是查询，我们需要从map的key中找到name，可以用find函数也可以用count函数，因为map中每个name和score是一一对应的，所以count的返回值只能是0或1，所以count也可以用来判断。对于map容器使用count函数时查找的是map容器的key，而不是value。
最后是删除，我们可以使用find函数来查找需要删除name的位置，找到后使用erase删除。

关于count函数和find函数，查找是否存在时可以用count或者find，查找有几个时用count，查找要找的元素的数量时用find。

代码

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

int main(){
    int n;
    cin>>n;
    map<string,int> a; //string记录name，int记录score
    
    for(int i=0;i<n;i++){
        int type,socre;
        string name;
        cin>>type;
        if(type==1){
            cin>>name>>socre;
            a[name]=socre;
            cout<<"OK"<<endl;
        }
        else if(type==2){
            cin>>name;
            if(a.count(name)) cout<<a[name]<<endl;
            else cout<<"Not found"<<endl;
        }
        else if(type==3){
            cin>>name;
            auto it=a.find(name);//因为该函数会返回一个输入迭代器，所以it的类型是map<string,int>::iterator  类型
            if(it==a.end()){//当find函数找不到时会返回end。
                cout<<"Not found"<<endl;
            }
            else{
                a.erase(it);
                cout<<"Deleted successfully"<<endl;
            }
        }
        else if(type==4){
            cout<<a.size()<<endl;
        }
    }
    return 0;
}