题意:求集合A和B中有几个相同的元素,其中集合中元素已经按升序排好,集合的元素个数n<=1000000
分析:集合中的元素已经按升序排好,只要对B中每个元素在A中进行二分查找就解决了。
代码:
public static void main(String[] args) {
int n,m; //从键盘录入A,B两个集合中的元素个数
Scanner input = new Scanner(System.in);
System.out.println("请输入A中元素个数:");
boolean flag;
do{ //依题意,元素个数应小于1000000
n=input.nextInt();
if(n>1000000){
System.out.println("元素个数应小于1000000,请重新输入:");
flag=true;
}
else{
flag = false;
}
} while(flag);
System.out.println("请输入B中元素个数:");
do{
m=input.nextInt();
if(m>1000000){
System.out.println("元素个数应小于1000000,请重新输入:");
flag=true;
}
else{
flag = false;
}
} while(flag);
int[] a = new int[n]; //A集合
int[] b = new int[m]; //B集合
ArrayList<Integer> x = new ArrayList<Integer>(); //定义一个ArrayList来存放A,B中相同的元素
System.out.println("A中元素值:");
for(int i = 0;i<n;i++){ //给A中每个元素赋值
a[i] = input.nextInt();
}
System.out.println("B中元素值:");
for(int i=0;i<m;i++){
b[i] = input.nextInt();
int result = binarySearch(a, b[i]); //对B中每个元素在A中进行二分查找
if(result != -1) { //结果不为-1,说明A中有此元素,将此元素加入x中
x.add(b[i]);
}
}
System.out.print("A,B中相同的元素有:");
for(Integer res : x){ //将结果打印出来
System.out.print(res+" ");
}
}
binarySearch方法的实现见于https://blog.csdn.net/Pmithie/article/details/82872877
让我们运行测试一下吧
关注我们,免费获取java海量视频学习资源~