div2-707 C Going Home

Description

在一个$array$ $a$ 中是否存在四个不同的下标$x,y,z,w$，使得$a_x+a_y=a_z+a_w$

Solution

不妨设$sum=a[i]+a[j],1\le i,j\le n,i\ne j$，对于一个长度为$n$的$array$ $a$，从中可以选出$\frac{n\ast (n-1)}{2}$个$sum$，当然里面可能有相同的

观察数据范围可以看出，$n$最大值为$2e5$，$a[i]$最大值为$2.5e6$，则$sum\in [2,5000000]$

如果 $\frac{n\ast (n-1)}{2}>5000000$，那么必然存在至少两个相同的$sum$

因此可以直接枚举$\frac{n\ast (n-1)}{2}$这么多个$sum$，如果出现相同的，则直接输出即可，时间复杂度为$O(min(n^2,n+c))$~~（详见Editorial）~~因为最多枚举到$5e6$个就会终止，所以其实一定不会超时

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxn = 5e6+10;

struct Node{
	int x,y;
}p[maxn];

int n, a[200000+10];

int main()
{	
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=1;i<=n-1;i++)
	{
		for(int j=i+1;j<=n;j++)
		{
			int sum = a[i] + a[j];
			if(p[sum].x==0 || p[sum].y==0)
			{
				p[sum].x = i;
				p[sum].y = j;
			}
			else
			{
				if(p[sum].x!=i && p[sum].y!=i && p[sum].x!=j && p[sum].y!=j)
				{
					printf("YES\n");
					printf("%d %d %d %d",p[sum].x,p[sum].y,i,j);
					return 0;
				}
			}
		}
	}
	cout<<"NO";
	return 0;
}