[编程题] 小易记单词

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

第一行为两个整数n(1 ≤ n ≤ 50)和m(1 ≤ m ≤ 50)。以空格分隔

第二行为n个字符串，表示小易能记住的单词，以空格分隔，每个单词的长度小于等于50。

第三行为m个字符串，系统提供的单词，以空格分隔，每个单词的长度小于等于50。

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

输入例子1:
3 4
apple orange strawberry
strawberry orange grapefruit watermelon

输出例子1:
136

解题思路

用数组表示小易记住的单词，用一个HashSet表示系统中的单词，并且小易每写出一个单词则将系统中的此单词移除。

import java.util.HashSet;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();

        String[] hasRemeber = new String[n];
        HashSet<String> systemWord = new HashSet<>();
        for (int i = 0; i < n; i++) {
            hasRemeber[i] = sc.next();
        }
        for (int i = 0; i < m; i++) {
            String str = sc.next();
            systemWord.add(str);
        }
        int score = 0;
        for (int i = 0; i < n; i++) {
            if (systemWord.contains(hasRemeber[i])) {

                score += hasRemeber[i].length() * hasRemeber[i].length();
                systemWord.remove(hasRemeber[i]);
            }
        }
        System.out.println(score);
    }
}