洛谷P1102 A-B 数对

P1102 A-B 数对

题目背景

出题是一件痛苦的事情！

相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 A+B Problem，改用 A-B 了哈哈！

题目描述

给出一串正整数数列以及一个正整数 ，要求计算出所有满足 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

第一行，两个正整数 。

第二行， 个正整数，作为要求处理的那串数。

输出格式

一行，表示该串正整数中包含的满足 的数对的个数。

输入输出样例 #1

输入 #1

4 1
1 1 2 3

输出 #1

3

说明/提示

对于 的数据，。

对于 的数据，，，。

2017/4/29 新添数据两组

思路

将A-B=C 转化为B=A-C，遍历数组作为A，减去C的值作为B，B的值个数就是数对的个数

因为会有相同的数，所以需要一个Map来记录相同的数的个数（数，重复出现的个数）

若A-C得到的值B，也存在在数列里面（就是Map的键），就加上对应的个数，如果没有就是默认值0

代码

package LAN;

import java.util.*;


public class ABshudui {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner sc = new Scanner(System.in);
		Map<Integer,Integer> map = new HashMap<>();
		int n = sc.nextInt();
		int c = sc.nextInt();
		int []arr = new int[n];
		for(int i = 0; i< n; i++) {
			arr[i] = sc.nextInt();
			map.put(arr[i],map.getOrDefault(arr[i], 0)+1);
		}
		long res= 0;
		for(int j = 0; j<n;j++) {
			int b = arr[j] - c;
			res+= map.getOrDefault(b,0);
		}
		System.out.print(res);

	}

}