1. 程式人生 > >HDU 4334 Trouble(hash + 列舉)

HDU 4334 Trouble(hash + 列舉)

題意:

給五個數的集合,問能否從每個集合中取一個數,使五個數之和為0.

思路:

集合大小是200,直接列舉的複雜度是200^5,一定會超時。

直接列舉的上限是3層,我們可以將列舉剩下兩個集合各任取一個元素可能組成的元素和,並將其作hash處理,使我們能很快判斷枚舉出來的三個集合元素和在剩下的兩個集合裡是否有相應元素匹配。

code:

/*
* @author Novicer
* language : C++/C
*/
#include<iostream>
#include<sstream>
#include<fstream>
#include<vector>
#include<list>
#include<deque>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<cmath>
#include<ctime>
#include<iomanip>
#define INF 2147483647
#define cls(x) memset(x,0,sizeof(x))
#define rise(i,a,b) for(int i = a ; i <= b ; i++)
using namespace std;
const double eps(1e-8);
typedef long long lint;

const int maxn = 200 + 5;
const lint fix = 2*1e15 + 20;
const int key = 100003;
lint s[5][maxn];
lint table[key + 10];

int n;

bool check(){
	if(s[0][0] > 0 && s[1][0] > 0 && s[2][0] > 0 && s[3][0] > 0 && s[4][0] > 0)
		return false;
	if(s[0][n-1] < 0 && s[1][n-1] < 0 && s[2][n-1] < 0 && s[3][n-1] < 0 && s[4][n-1] < 0)
		return false;
	return true;
}

void addintohash(lint x){
    lint p = x % key;
    while(table[p] != -1 && table[p] != x){
        p ++;
        if(p == key) p = 0;
    }
    table[p] = x;
}
bool hash1(lint x){
    lint p = x % key;
    while(table[p] !=x && table[p] != -1){
        p ++;
        if(p == key) p = 0;
    }
    if(table[p] != x) return false;
    return true;
}
bool solve(){
	bool flag = false;
	for(int i = 0 ; i < n ; i++){
		for(int j = 0 ; j < n ; j++){
			for(int k = 0 ; k < n ; k++){
				lint tmp = s[0][i] + s[1][j] + s[2][k];
				if(tmp >= -fix && tmp <= fix && hash1(0 - tmp + fix)){
					flag = true;
					return true;
				}
			}
		}
	}
	return false;
}
int main(){
//	freopen("input.txt","r",stdin);
	int t; cin >> t;
	while(t--){
		memset(table,-1,sizeof(table));
		cin >> n;
		for(int i = 0 ; i < 5 ; i++){
			for(int j = 0 ; j < n ; j++){
				scanf("%I64d",&s[i][j]);
			}
			sort(s[i] , s[i]+n);
		}
		for(int i = 0 ; i < n ; i++){
			for(int j = 0 ; j < n ; j++){
				addintohash(s[3][i]+ s[4][j] + fix);
			}
		}
		if(!check()){
			cout << "No\n";
			continue;
		}
		else{
			if(solve()){
				cout << "Yes\n";
			}
			else{
				cout << "No\n";
			}
		}
		
	}
		
	return 0;
}