Codeforces Round 925 (Div. 3) D. Divisible Pairs (Java)

Codes Round 925 (Div. 3) D. Divisible Pairs (Java)

比赛链接：Codeforces Round 925 (Div. 3)

D题传送门：D.Divisible Pairs

题目：D.Divisible Pairs

题目描述

输出格式

For each test case, output a single integer — the number of beautiful pairs in the array $ a $ .

样例 #1

样例输入 #1

7
6 5 2
1 2 7 4 9 6
7 9 5
1 10 15 3 8 12 15
9 4 10
14 10 2 2 11 11 13 5 6
9 5 6
10 7 6 7 9 7 7 10 10
9 6 2
4 9 7 1 2 2 13 3 15
9 2 3
14 6 1 15 12 15 8 2 15
10 5 7
13 3 3 2 12 11 3 7 13 14

样例输出 #1

2
0
1
3
5
7
0

分析：

题目要我们求美丽对的数量。

由美丽对的定义可得

• $a_i+a_j$ % $x=0$ 等价于 $（a_i$% x + $a_j$% x）%x = 0
• $a_i-a_j$ % $y=0$ 等价于 $a_i$%y = $a_j$% y

我们定义一个HashMap的集合，集合的键 List 用来记录对 a_i%x 和 a_i%y 的答案，集合的值用来记录对应键出现的次数。

定义一个变量 t 来记录 $a_i$

定义 ArrayList< Integer > a 先记录 t%x ，再记录 t%y
定义 ArrayList< Intege r> b 先记录 (x-a.get(0))%x ，再记录 t%y

计算 map 中 b 出现的次数，然后将 a 加入到 map 中

代码：

import java.util.*;

public class Main{
    public static void main(String[] args) {     
    	Scanner sc = new Scanner(System.in);
    	int tr = sc.nextInt();
    	while(tr-->0) {
    		int n = sc.nextInt();
    		int x = sc.nextInt();int y = sc.nextInt();
    		Map<List<Integer>,Long> map = new HashMap<>();
    		long ans = 0;
    		for(int i = 0;i < n;i++) {
    			int t = sc.nextInt();
    			ArrayList<Integer> a = new ArrayList<>();
    			ArrayList<Integer> b = new ArrayList<>();
    			a.add(t%x);a.add(t%y);
    			b.add((x-a.get(0))%x);b.add(t%y);
    			if(map.containsKey(b)) {
    				ans += map.get(b);
    			}
    			map.put(a,map.getOrDefault(a, (long) 0)+1);
    		}
    		System.out.println(ans);
    	}
    	sc.close();
    }
}