给5个数组， 问你能否从每个数组里取出一个数， 使得5个数相加为0.数组长度小于200
将第一个数组和第二个数组合并， 第三个数组和第四个数组合并。将合并好的两个数组排序。
然后枚举第五个数组里的数， 将这个数取反， 记为x。 然后两个指针， 一个指针pos1指向合并后第一个数组的开始， 一个指针pos2指向第二个数组的末尾。 如果这两个数相加大于x， 那么pos2--， 如果小于， pos1++， 相等说明有解。
#include <iostream> #include <vector> #include <cstdio> #include <cstring> #include <algorithm> #include <cmath> #include <map> #include <set> #include <string> #include <queue> #include <stack> #include <bitset> using namespace std; #define pb(x) push_back(x) #define ll long long #define mk(x, y) make_pair(x, y) #define lson l, m, rt<<1 #define mem(a) memset(a, 0, sizeof(a)) #define rson m+1, r, rt<<1|1 #define mem1(a) memset(a, -1, sizeof(a)) #define mem2(a) memset(a, 0x3f, sizeof(a)) #define rep(i, n, a) for(int i = a; i<n; i++) #define fi first #define se second typedef pair<int, int> pll; const double PI = acos(-1.0); const double eps = 1e-8; const int mod = 1e9+7; const int inf = 1061109567; const int dir[][2] = { {-1, 0}, {1, 0}, {0, -1}, {0, 1} }; int n, cnt, num; ll a[4][205], x[160005], y[160005], b[205]; int solve() { for(int k = 0; k < n; k++) { ll tmp = -b[k]; int i = 0, j = num-1; while(i < cnt && j >= 0) { if(x[i] + y[j] == tmp) { return 1; } if(x[i] + y[j] > tmp) { j--; } else { i++; } } } return 0; } void pre() { for(int i = 0; i < n; i++) { for(int j = 0; j < n; j++) x[cnt++] = a[0][i] + a[1][j]; } sort(x, x+cnt); /*cnt = unique(x, x+cnt)-x;*/ for(int i = 0; i < n; i++) { for(int j = 0; j < n; j++) y[num++] = a[2][i] + a[3][j]; } sort(y, y+num); /*num = unique(y, y+num)-y;*/ } void read() { for(int i = 0; i < 4; i++) { for(int j = 0; j < n; j++) scanf("%I64d", &a[i][j]); } for(int i = 0; i < n; i++) scanf("%I64d", &b[i]); } int main() { int t; cin>>t; while(t--) { scanf("%d", &n); cnt = num = 0; read(); pre(); if(solve()) { puts("Yes"); } else { puts("No"); } } return 0; }