HDU 4334 Trouble（哈希|线性查找）

最新推荐文章于 2021-12-11 18:48:23 发布

转载最新推荐文章于 2021-12-11 18:48:23 发布 · 109 阅读

本文介绍了一种解决五个集合中是否存在元素组合使元素之和为零的问题的算法。通过将五个集合分为三组并利用排序和双指针技巧来降低时间复杂度，最终达到高效解决问题的目的。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定五个集合。问是否能从五个集合各取一个元素，使得元素之和为0.

这道题有两种做法，一种是哈希，然而之前没写过哈希.....比赛后从大神那copy了一份。

这里说还有一种。

对于这五个集合分为三组。1，2组求和为一组，3,4组求和分为一组，5为一组。

那么如今转化为了是否能从前两组中各取一个元素。使得这两个值和为第三组一个元素的相反数。

那么对于第一组我们升序排序。第二组我们降序排序。

对于第三组里的任一元素，假如第一组队首加第二组队首之和大于第三组的元素。那么第二组游标往后移一位，反之第一组移一位，

那么这个查找时间就为O（m），m为数组元素个数。

那么总的时间复杂度为O(n*n*n).

<pre class="cpp" name="code">#include<cstdio>  
#include<cstring>  
#include<cmath>  
#include<cstdlib>  
#include<iostream>  
#include<algorithm>  
#include<vector>  
#include<map>  
#include<queue>  
#include<stack> 
#include<string>
#include<map> 
#include<set>
#define eps 1e-6 
#define LL long long  
#define pii pair<int,int>
using namespace std;  

//const int maxn = 100 + 5;
//const int INF = 0x3f3f3f3f;

LL s[6][205];
LL s12[200*202], s34[200*202];
int n; 
bool check(LL t) {
	int y12 = 0, y34 = n*n-1;
//	cout << y12 << endl << y34 << endl;
	while(y12<n*n && y34>=0) {
		LL tmp = s12[y12] + s34[y34];
//		cout << s12[y12] << endl << s34[y34] << endl;
//		cout << tmp << endl;
		if(tmp == t) return true;
		else if(tmp < t) y12++;
		else y34--;
	}
	return false;
}

int main() {
//	freopen("input.txt", "r", stdin);
	int t; cin >> t;
	while(t--) {
		cin >> n;
		for(int i = 0; i < 5; i++) {
			for(int j = 0; j < n; j++) scanf("%I64d", &s[i][j]);
		}
		int y1 = 0;
		for(int i = 0; i < n; i++) {
			for(int j = 0; j < n; j++) {
				LL tmp = s[0][i]+s[1][j];
				s12[y1++] = tmp;
			}
		}
		int y2 = 0;
		for(int i = 0; i < n; i++) {
			for(int j = 0; j < n; j++) {
				LL tmp = s[2][i]+s[3][j];
				s34[y2++] = tmp;
			}
		}
//		cout << y1 << y2 << endl;
		sort(s12, s12+n*n);
		sort(s34, s34+n*n);
		int tag = 0;
		for(int i = 0; i < n; i++) if(check(-s[4][i])) {
			tag = 1; break;
		}
//		cout << check(4) << endl;
		if(tag) puts("Yes");
		else puts("No");
	} 
	return 0;
}