网易--小易记单词

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

输入描述:

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

输出描述:

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

输入例子:

3 4
apple orange strawberry
strawberry orange grapefruit watermelon

输出例子:

136

#include <iostream>
#include <algorithm>
#include "string.h"
#include "stdio.h"
#include <vector>
#include <deque>
#include <stack>
#include<map>
#include<utility>
#include "math.h"
#include <set>
using namespace std;

int main() {
    int n;
    int m;
    set<string> str1;
    set<string> str2;
    cin>>n>>m;
    for(int i = 0;i<n;i++)
    {
        string s;
        cin>>s;
        str1.insert(s);
    }
    for(int i = 0;i<n;i++)
    {
        string s;
        cin>>s;
        if(str1.find(s)!=str1.end())
            ans+=s.size()*s.size();
    }
    
    cout<<ans<<endl;
    return 0;

}