Beijing University of Posts Find (java)

题目描述
输入数组长度 n 输入数组      a[1...n] 输入查找个数m 输入查找数字b[1...m]   输出 YES or NO  查找有则YES 否则NO 。
输入描述:
输入有多组数据。
每组输入n，然后输入n个整数，再输入m，然后再输入m个整数（1<=m,n<=100）。
输出描述:
如果在n个数组中输出YES否则输出NO。
示例1
输入
复制
5
1 5 2 4 3
3
2 5 6
输出
复制
YES
YES
NO

import java.util.*;
import java.io.*;
import java.math.*;
import java.text.* ;
public class Main
{
	public static void main(String[] args) {
		try {
			BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
			String str;
			while((str=br.readLine()) != null) {
				 int n = Integer.parseInt(str); 
				 String[] parts1 = br.readLine().split(" ");
				 HashSet<String> set = new HashSet<>();
				 for(int i = 0; i < n; i++) {
					 set.add(parts1[i]);
				 }
				 int m = Integer.parseInt(br.readLine()); 
				 String[] parts2 = br.readLine().split(" ");
				 for(int i = 0; i < m; i++) {
					 if(set.contains(parts2[i])) System.out.println("YES");
					 else System.out.println("NO");
				 }
				 
			}
		} catch(IOException e) {
			e.printStackTrace();
		}
	}
}