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

Codeforces 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 = 0 && $a_j$% x = 0
• $a_i-a_j$ % $y=0$ 等价于 $a_i$%y + $a_j$% y = y

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

定义一个变量 t 来存储 $a_i$

定义 ArrayList a 先存储 t%x 再 存储 t%y
定义 ArrayList 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();
    }
}