2022牛客多校十 I-Yet Another FFT Problem?（鸽巢原理）

题目链接：登录—专业IT笔试面试备考平台_牛客网

题目：

样例输入： 

4 4
2 4 8 16
3 9 27 81

样例输出：

1 3 1 2

题意：给定一个n，m，分别代表a数组和b数组的长度，接下来给出a数组和b数组，问是否存在i,j,k,l满足1<=i,j<=n，1<=k,l<=m，i!=j,k!=l使得|aj-ai|=|al-ak|，如果存在则输出，否则输出-1.

分析：原等式ai-aj=bk-bl（绝对值只需要交换坐标顺序即可消除）可以转化为ai+bl=bk+aj,其中i!=j,l!=k。

我们看一下数组元素的范围发现是1e7的，那么ai+bl的范围就是2e7，于是我们可以直接用一个map存储已经出现的t=ai+bl,值互不相同的t最多只有2e7个，这是由鸽巢原理得到的，结论也是比较显然的，那么当我们遍历到一个t=bk+aj，且t在之前已经出现过时，那么我们就找到了i,l,k,j满足ai+bl=bk+aj，只是我们需要在过程中记录每个元素出现的位置。但是如果要是数组里面有元素值相同的元素那么就会产生错误，因为我们可能是用一个i=j找到了bl=bk但是l!=k，这样显然是不满足题意的，所以我们可以先对原数组进行去重，但是简单的去重还会影响答案，最简单的例子就是a数组中存在两个元素值相同的数，而b数组中也存在两个元素值相同的数，那么显然这四个数就是我们要找的数，如果我们一开始就将其去重，则会忽略掉这种答案，所以我们需要特殊处理一下差为0的情况。

细节见代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<map>
#include<queue>
#include<vector>
#include<cmath>
using namespace std;
const int N=2e7+10;
vector<int>a,b;
int mpa[N],mpb[N];
int mp1[N],mp2[N];
int main()
{
	int n,m;
	cin>>n>>m;
	int t,id11=0,id12=0,id21=0,id22=0;//分别记录a，b数组中元素值相同的元素的位置 
	a.reserve(n+2);b.reserve(m+2);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&t);
		if(mpa[t])//如果有相同的两个数需要当作特殊情况考虑一下 
		{
			id11=mpa[t];
			id12=i;
		}
		mpa[t]=i;
		a.push_back(t);
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&t);
		if(mpb[t])
		{
			id21=mpb[t];
			id22=i;
		}
		mpb[t]=i;
		b.push_back(t);
	}
	if(id11&&id21)//特判存在差为0的情况 
	{
		printf("%d %d %d %d",id11,id12,id21,id22);
		return 0;
	}
	sort(a.begin(),a.end());
	a.erase(unique(a.begin(),a.end()),a.end());
	sort(b.begin(),b.end());
	b.erase(unique(b.begin(),b.end()),b.end());
	n=a.size();m=b.size();
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<m;j++)
		{
			int t=a[i]+b[j];
			if(mp1[t])
			{
				printf("%d %d %d %d",mpa[a[i]],mp1[t],mpb[b[j]],mp2[t]);
				return 0;
			}
			mp1[t]=mpa[a[i]];mp2[t]=mpb[b[j]];
		}
	}
	puts("-1");
	return 0;
}