#HK5389. 「OOI 2021 Day 2」我要回家

「OOI 2021 Day 2」我要回家

题目描述

题目译自 Open Olympiad in Informatics 2021 Day2 T1 「Прыжки по шкафам

在远程学习的第三个月,女学生娜斯佳感到无聊,决定回故乡待上几周散散心。为了让她的旅途更有趣,她的朋友送给她一个整数数组 aa

在旅途的第五个小时,娜斯佳想起了这个礼物。为了娱乐自己,她决定检查是否可以选择四个不同的编号 x,y,z,wx, y, z, w,使得 ax+ay=az+awa_x + a_y = a_z + a_w

在忙于此事时,娜斯佳没有注意到时间流逝,她的火车已经到达了目的地站。她没能完成检查,但仍然想知道答案。你能帮助她吗?

输入格式

第一行包含一个整数 nn (4n200000)(4 \leq n \leq 200000),表示数组中的数字数量。

第二行包含 nn 个整数 a1,a2,,ana_1, a_2, \ldots, a_n (1ai5106)(1 \leq a_i \leq 5 \cdot 10^{6})

输出格式

在第一行输出 YES,如果存在四个这样的编号;否则输出 NO

如果存在这样的编号,则在第二行输出 x,y,z,wx, y, z, w (1x,y,z,wn)(1 \leq x, y, z, w \leq n)

如果存在多种可能的答案,可以输出任意一种。

6
2 1 5 2 7 4

YES
2 3 1 6

在第一个样例中,a2+a3=1+5=2+4=a1+a6a_2 + a_3 = 1 + 5 = 2 + 4 = a_1 + a_6。请注意,答案 2 3 4 6 也是正确的。

5
1 3 1 9 20

NO

在第二个样例中,无法选择四个这样的编号。答案 1 2 2 3 不符合要求,因为编号不全不同,尽管 a1+a2=1+3=3+1=a2+a3a_1 + a_2 = 1 + 3 = 3 + 1 = a_2 + a_3

数据范围与提示

详细子任务附加限制及分值如下表所示。其中子任务 00 是样例。

子任务 分值 附加限制 子任务依赖 备注
11 3030 n60n \leq 60
22 3030 n600n \leq 600
33 4040