题目描述:
给定一个整数的数组,找出其中的pair(a,b),使得a+b=0,并返回这样的pair数目。(a,b)和(b,a)是同一组。
输入:
整数数组
输出
找到的pair数目
样例输入
-1,2,4,5,-1
样例输出
1
思路:
找到数组中的和为0的两个数,将其保存在map中,根据题目要求,[1,-1]和[-1,1]这样的算是一种,所以需要map的一份拷贝数据,如果map中存在一个key是另外一个value的情况,将拷贝的map中的内容删除一个即可。
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
public class Test1 {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
while (sc.hasNext()){
String s = sc.nextLine();
String ss = s.replaceAll(" ","");
String[] temp = ss.trim().split(",");
long[] array = new long[temp.length];
for (int i = 0; i < array.length; i++) {
array[i] = Integer.parseInt(temp[i]);
}
System.out.println(getResult(array));
}
}
public static long getResult(long[] array){
Map<Long, Long> map = new HashMap<>();
for (int i = 0; i < array.length; i++) {
for (int j = i; j < array.length; j++) {
if(array[i] + array[j] == 0 &&!map.containsKey(array[i])&&!map.containsKey(array[j])){
map.put(array[i],array[j]);
}
}
}
Map<Long,Long> hashmap = new HashMap<>();
for (Long key : map.keySet()) {
hashmap.put(key,map.get(key));
}
for (Long key : map.keySet()) {
if(map.containsKey(map.get(key))&&hashmap.containsKey(map.get(key))){
hashmap.remove(key);
}
}
return hashmap.size();
}
}