给出N个整数,你来判断一下是否能够选出4个数,他们的和为0,可以则输出"Yes",否则输出"No"。
Input
第1行,1个数N,N为数组的长度(4 <= N <= 1000) 第2 - N + 1行:A[i](-10^9 <= A[i] <= 10^9)
Output
如果可以选出4个数,使得他们的和为0,则输出"Yes",否则输出"No"。
Sample
Inputcopy | Outputcopy |
---|---|
5 -1 1 -5 2 4 | Yes |
这个题,学校icpc的一道选拔赛, 哈希表不会写, 直接双重循环加二分,数据给的比较小, 所以n^2logn是没问题的。
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long int ll;
const int N = 1e06 + 10;
ll q[N];
int main() {
int n, m;
cin >> n;
for (int i = 0; i < n; i++) {
cin >> q[i];
}
sort(q, q + n);
for (int i = 0; i < n; i++) {
for (int j = i + 1; j < n; j++) {
ll t = -(q[i] + q[j]);
int l = j + 1, r = n - 1;
while (l <= r) {
if (q[l] + q[r] == t) {
cout << "Yes";
return 0;
} else if (q[l] + q[r] < t) {
l++;
} else {
r--;
}
}
}
}
cout << "No";
return 0;
}
icpc下次一定哈哈哈