51nod 4个数和为0

传送门

题意

输入n个数，判断是否存在四个数的和为0.(4 <= n <= 1000)

思路

算了一下，最坏情况下是1.7e8。

代码

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 1000 + 10;
typedef long long ll;
ll arr[maxn];
ll CD[maxn*maxn];
int main() {
	int n;
	while (~scanf("%d", &n)) {
		for (int i = 0; i < n; i++) {
			scanf("%lld", &arr[i]);
		}
		sort(arr, arr+n);
		int flag = 0;
		for (int i = 0; i < n; i++) {
			for (int j = i+1; j < n; j++) {
				ll tmp = -(arr[i] + arr[j]);
				int l = j+1, r = n-1;
				while (l < r) {
					if (arr[l] + arr[r] == tmp) {
						flag = 1;
						break;
					}
					else if (arr[l] + arr[r] < tmp) {
						l++;
					} else {
						r--;
					}
				}
			}
		}
		if (flag == 1) printf("Yes\n");
		else printf("No\n");
	}
	return 0;
}