这是远程学习的第三个月,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;
}