网易2017春招笔试——小易记单词

小易记单词

题目

小易参与了一个记单词的小游戏。游戏开始系统提供了m个不同的单词，小易记忆一段时间之后需要在纸上写出他记住的单词。小易一共写出了n个他能记住的单词，如果小易写出的单词是在系统提供的，将获得这个单词长度的平方的分数。注意小易写出的单词可能重复，但是对于每个正确的单词只能计分一次。

输入描述:

输入数据包括三行：
第一行为两个整数n(1 ≤ n ≤ 50)和m(1 ≤ m ≤ 50)。以空格分隔
第二行为n个字符串，表示小易能记住的单词，以空格分隔，每个单词的长度小于等于50。
第三行为m个字符串，系统提供的单词，以空格分隔，每个单词的长度小于等于50。

输出描述:

输出一个整数表示小易能获得的分数

输入例子:

3 4
apple orange strawberry
strawberry orange grapefruit watermelon

输出例子:

136

---

分析：

用一个set存储能记住的单词，另一个set存储系统提供的单词，然后循环判断记住的每一个单词是不是在系统提供的集合里即可。

代码如下：

#include <iostream>
#include <string>
#include <set>
using namespace std;
int main() {
    int n, m;
    set<string> s1, s2;
    cin >> n >> m;
    for (int i = 0; i < n; i++) {
        string s;
        cin >> s;
        s1.insert(s);
    }
    for (int i = 0; i < m; i++) {
        string s;
        cin >> s;
        s2.insert(s);
    }
    int ans = 0;
    for (set<string>::iterator it = s1.begin(); it != s1.end(); it++) {
        if (s2.find(*it) != s2.end()) {
            ans += it->length() * it->length();
        }
    }
    cout << ans << endl;
}