C. Going Home(抽屉原理)

Problem - C - Codeforces

这是远程学习的第三个月，Nastya厌倦了呆在宿舍，于是她决定回到家乡。为了让她的旅行更有趣，Nastya的一个朋友给了她一个整数数组a。回家几个小时后，娜斯佳想起了礼物。为了娱乐自己，她决定检查，是否有四个不同的指标x, y, z, w使得az + ay = az + aw她的火车已经到达目的地，但她仍然没有找到答案。你能帮她解开这个谜吗?输入第一行包含单个整数n (4 <n< 200 000)——数组的大小。第二行包含n个整数a1, a2，..， an (1 < ai < 2.5-106)。输出如果有四个索引，则打印“YES”，否则打印“NO”。如果存在这样的索引，打印这些索引，y, z和w (1 <， y, z, w < n)。如果有多个答案，打印其中任何一个。

Examples

input

Copy

6
2 1 5 2 7 4

output

Copy

YES
2 3 1 6 

input

Copy

5
1 3 1 9 20

output

Copy

NO

请注意在第一个例子中，a2 + a3 =1+5 = 2+4 = a1 + a6。注意，还有其他答案，例如，2 3 4 6。在第二个例子中，我们不能选择四个指标。答案1 2 2 3是错误的，因为尽管如此，指数应该是不同的a1 +a2=1+3=3+1=a2 + a3

题解:
看到数据这么怪,那么肯定有问题

最大两个数相加和是5e6

一直用两个不同数相加,记录

也就是说最多5e6次一定能找到答案,不会t,(服了这个思路)

 

#include<iostream>
#include<string>
#include<vector>
#include<cstring>
#include<algorithm>
#include<queue>
#include<cmath>
#include<map>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
int a[2000050];
PII p[5000050];
void solve()
{
	int n;
	scanf("%d",&n);
	for(int i = 1;i <= n;i++)
	{
		scanf("%d",&a[i]);
	}
	map<PII,int> f;
	for(int i = 1;i < n;i++)
	{
		for(int j = i + 1;j <= n;j++)
		{
			int s = a[i] + a[j];
			if(p[s].first)
			{
				if(p[s].first != i&&p[s].second != j&&p[s].first != j&&p[s].second != i)
				{
					printf("YES\n");
					printf("%d %d %d %d",i,j,p[s].first,p[s].second);
					return ;
				}
			}
			p[s] = {i,j};
		}
	}
	printf("NO");
}
signed main() 
{
//	ios::sync_with_stdio(0);
//	cin.tie(0);cout.tie(0);
	int t = 1;
//    scanf("%d",&t);
	while (t--) 
	{
		solve();
	}
	return 0;
}