Codeforces Round 875 (Div. 2) D.The BOSS Can Count Pairs（数论）

Martin925

已于 2023-05-30 16:46:34 修改

阅读量300

点赞数 1

分类专栏： codeforces 文章标签：算法 c++ 数据结构

于 2023-05-30 16:24:42 首次发布

本文链接：https://blog.csdn.net/qq_62026149/article/details/130950694

版权

codeforces 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

该篇文章描述了一个算法问题，给定两个整数数组a和b，要求找出满足a[i]*a[j]=b[i]+b[j]条件的无序对(i,j)的数量。解决方案包括对整数对进行排序，然后枚举a[i]的值，计算对应b[i]的个数，使用哈希表避免重复计数，最后累加答案。

摘要由CSDN通过智能技术生成

题目大意：给定两个整数数组a，b，数组长度<=n。求使得a[i]*a[j]==b[i]+b[j]成立的无序对(i,j)总共有多少对。两个数组元素的取值范围为1<=a<=n,1<=b<=n（a，b代表数组中的元素）。

思路：由于1<=a[i]<=n,1<=b[j]<=n，a[i]*a[j]==b[i]+b[j]，可得b[i]+b[j]<=2*n，可得a[i]*b[j]<=2*n。

1、创建整数对并从小到大排序（为了方便描述，作者将整数对数组pa[i].first描述成a[i]，pa[i].second描述成b[i]），然后枚举x：1~ $\sqrt{2n}$ 作为a[i]。

2、枚举j：1~n。

3、此时可以求得b[i]，若1<=b[i]<=n，那么最终答案加上a[i]=x情况下，与a[i]对应的b[i]值的个数mp[b[i]]。

如果此时的a[j]==x，mp[b[i]]++（这一步的目的是为了使得枚举的j只能与小于j的i匹配，避免因顺序原因重复贡献）。

4、输出答案。

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int MAXN=2e5+100;
int a[MAXN];
int b[MAXN];
pair<int,int> pa[MAXN];
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	int t;cin>>t;
	while(t--)
	{
		int n;cin>>n;
		for(int i=1;i<=n;i++)
		{
			cin>>a[i];
		}
		for(int i=1;i<=n;i++)
		{
			cin>>b[i];
		}
		for(int i=1;i<=n;i++)
		{
			pa[i]=make_pair(a[i],b[i]);
		}
		int num=0;
		sort(pa+1,pa+1+n,less<pair<int,int>>());
		int ans=0;
		for(int x=1;x*x<=2*n;x++)
		{
			vector<int>mp(n+10);
			for(int j=1;j<=n;j++)
			{
				int bi=x*pa[j].first-pa[j].second;
				if(bi>=1&&bi<=n)
				{
					ans+=mp[bi];
				}
				if(pa[j].first==x)
				{
					mp[pa[j].second]++;
				}
			}

		}
		cout<<ans<<"\n";


	}


	return 0;

}