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

题目：

 

思路：

        这里我们可以把公式 ai * aj = bi * bj  转换成  ai / bi = bj / ai

这样就可以顺着序号来找答案，统计好对应的计算结果，最后累加，去除我们重复部分即可

代码详解如下：

#include <iostream>
#include <vector>
#include <unordered_map>
#define endl '\n'
#define int long long
#define umap unordered_map
#define ___G std::ios::sync_with_stdio(false),cin.tie(0), cout.tie(0)
using namespace std;

inline void solve()
{
	int n = 0;
	// r 用于记录好对应位置的 a / b 的结果
	umap<double, int>r;

	vector<int>a, b;

	cin >> n;

	for (int i = 0, x; i < n; ++i)
	{
		cin >> x;
		a.emplace_back(x);
	}
	for (int i = 0, x; i < n; ++i)
	{
		cin >> x;
		b.emplace_back(x);
	}

	// 开始记录对应的结果
	for (int i = 0; i < n; ++i)
	{
		r[1.0 * a[i] / b[i]]++;
	}

	// 统计符合 ai * bj == aj * bi 的结果
	int ans = 0;

	for (int i = 0; i < n; ++i)
	{
		// 计算 bj / aj 结果
		double tem = 1.0 * b[i] / a[i];

		// 如果对应的位置 ai * bi == aj * bj  i == j
		// 说明会有重复对 ，所以我们这里要去掉我们这个重复对
		if (1.0 * b[i] / a[i] == 1.0 * a[i] / b[i])
			ans--;

		// 累加对应的 bj / aj数量
		ans += r[1.0 * b[i] / a[i]];
	}

	// 由于我们操作是不管顺序，凡是符合的就配对
	// 所以所配对的数量，肯定是一前一后，重复一次
	// 所以我们求得的总和 要 / 2
	cout << ans / 2 << endl;

	return ;
}

signed main()
{
	___G;
	int _t;
	cin >> _t;
	while (_t--)
	{
		solve();
	}

	return 0;
}

 最后提交：