看题目的数据,其中可以发现两个数相加结果最大也只是5e6,根据抽屉原理,只要有5e6+1个数就至少会有两个数相加得到相同的值。那么也就是说只要枚举出来的两个数相加的答案最多有5e6+1个就一定会得到至少两组坐标满足相加相等,直接暴力即可。
#include<iostream>
using namespace std;
typedef long long ll;
const int Max = 5e6 + 5;
int lst[Max];
int ls[Max][2];
int main()
{
int n;cin >> n;
for (int i = 1;i <= n;i++)cin >> lst[i];
for (int i = 1;i < n;i++)
{
for (int j = i + 1;j <= n;j++)
{
int sum = lst[i] + lst[j];
if (ls[sum][0] != 0&&ls[sum][1]!=i&&ls[sum][1]!=j&&ls[sum][0]!=i)
{
cout << "YES" << endl;
cout << ls[sum][0] << " " << ls[sum][1] << " " << i << " " << j;
return 0;
}
else ls[sum][0] = i, ls[sum][1] = j;
}
}
cout << "NO" << endl;
}