2023河南萌新联赛第（五）场：郑州轻工业大学 I - 双指针

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

给定一个数组 $a$ 和数组 $b$ ，两个数组的长度都为 $n$, 请选择分别两个下标 $i,j$ 且 $i,j$ 满足条件 $1\le i<j\le n$, 求有多少对 $i,j$ 满足 $a_i\times a_j=b_i\times b_j$ 。

输入描述:

本题包含多组数据

第一行包含一个正整数 $T$（$1\le T\le 1\times 10^5$）。

对于每组数据：

第一行包含一个正整数 $n(1\le n\le 2\times 10^5)$ 。

接下来 $2$ 行

第1行包含 $n$ 个整数 $a_1,a_2,a_3...a_n(1\le a_i\le 2\times 10^5)$ 。

第2行包含 $n$ 个整数 $b_1,b_2,b_3...b_n(1\le b_i\le 2\times 10^5)$ 。

${\sum }_{i=1}^{T}n\le 2\times 10^5$

输出描述:

对于每组数据：

输出一行表示有多少对 $i,j$ 满足 $a_i\times a_j=b_i\times b_j$ 。

输入

2
5
3 2 2 2 3
3 1 2 4 2
5
2 1 1 1 3
4 2 4 3 1

输出

2
1

import java.io.*;
import java.util.HashMap;public class Main {public static int gcd(int a, int b) {return b == 0 ? a : gcd(b, a % b);}public static void main(String[] args) throws IOException {BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));int T = Integer.parseInt(bf.readLine());while (T-- > 0) {HashMap<String, Integer> cnt = new HashMap<>();int n = Integer.parseInt(bf.readLine());int[] a = new int[n];int[] b = new int[n];String[] str = bf.readLine().split(" ");for (int i = 0; i < n; i++) {a[i] = Integer.parseInt(str[i]);}str = bf.readLine().split(" ");for (int i = 0; i < n; i++) {b[i] = Integer.parseInt(str[i]);}long res = 0;for (int i = 0; i < n; i++) {int d = gcd(a[i], b[i]);a[i] /= d;b[i] /= d;String s1 = a[i] + " " + b[i];String s2 = b[i] + " " + a[i];res += cnt.getOrDefault(s2, 0);cnt.put(s1, cnt.getOrDefault(s1, 0) + 1);}bw.write(res + "\n");}bw.close();}
}