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

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$ 。

• 关键词：思维，STL
• 给定一个数组 $a$ 和数组 $b$，问有多少对 $i，j$ 满足 $1\le i<j\le n$ 使得 $a_i\times a_j=b_i\times b_j$ 。
• 将原式转化为 $a_i/b_i=b_j/a_j$，令 $c_i=a_i/b_i$ , $d_j=b_j/a_j$，这样问题就转化成有多少对 $i$，$j$ 满足 $1\le i<j\le n$ 使得 $c_i=d_j$。
• 使用 $map$ 分别统计数组 $c$ 和数组 $d$ 中每个数出现的次数并 计算答案。注意由于 $i<j$ 当 $a_i/b_i$ = $b_i/a_i$ 时处理不当答案 可能会有冗余计算，注意避免这种情况。如果担心 $double$ 精度问题可以用分数代替。注意答案开 $long$ $long$。

输入

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();
    }
}