哈希表-三数之和

 力扣题号：15. 三数之和

一、题目描述

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

二、示例

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

示例 2：

输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。

示例 3：

输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0 。

三、求解思路

四、代码实现

#define _CRT_SECURE_NO_WARNINGS 1
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

#define MAX 99999999

void PrintArr(int* A, int n);

int cmp(const void* a,const void* b) {
	return *(int*)a - *(int*)b;
}

// 自定义三元组
typedef struct threearray {
	int ** arr= NULL;
	int size = 0;
}ThreeArray;

// set元素
typedef struct set {
	int key;
}SetNode;

// 自定义哈希set
typedef struct hash_set {
	SetNode arr[1005];
}HashSet;
// 映射函数
void voidreflection(HashSet& H, int n) {

	int n1 = n % 1000;
	if (n1 < 0) {
		n1 = -n1;
	}
	while (H.arr[n1].key != MAX && H.arr[n1].key != n) {
		n1++;
		n1 %= 1000;
	}
	if (H.arr[n1].key == n) { // 元素重复
		return;
	}
	H.arr[n1].key = n;
}
// 查找哈希表中是否有该元素
int find(HashSet& H, int n) {
	int n1 = n % 1000;
	if (n1 < 0) {
		n1 = -n1;
	}
	while (H.arr[n1].key != n && H.arr[n1].key != MAX) {
		n1++;
		n1 %= 1000;
	}
	if (H.arr[n1].key == n) {
		return 1;
	}
	return 0;
}
// 哈希表中删除元素
void erase(HashSet& H, int n) {
	int n1 = n % 1000;
	if (n1 < 0) {
		n1 = -n1;
	}
	while (H.arr[n1].key != n && H.arr[n1].key != MAX) {
		n1++;
		n1 %= 1000;
	}
	if (H.arr[n1].key == n) {
		H.arr[n1].key = MAX;
		return;
	}
	return;
}

// 三数之和_哈希法
void solution_hash(int *arr, int n, ThreeArray& result) {
	// 先对数组进行排序
	qsort(arr, n, sizeof(int), cmp);
	PrintArr(arr, n);
	// 定义哈希set
	HashSet set;
	for (int i = 0; i < 1005; i++) {
		set.arr[i].key = MAX;
	}
	result.size = 0;


	// 找出a + b + c = 0
	// a = nums[i], b = nums[j], c = -(a + b)
	for (int i = 0; i <n; i++) {
		// 排序之后如果第一个元素已经大于零，那么不可能凑成三元组
		if (arr[i] > 0) {
			break;
		}
		if (i > 0 && arr[i] == arr[i - 1]) { //三元组元素a去重
			continue;
		}
		// 重新初始化set
		for (int i = 0; i < 1005; i++) {
			set.arr[i].key = MAX;
		}

		for (int j = i + 1; j < n; j++) {
			if (j > i + 2
				&& arr[j] == arr[j - 1]
				&& arr[j - 1] == arr[j - 2]) { // 三元组元素b去重
				continue;
			}
			int c = 0 - (arr[i] + arr[j]);
			if (find(set,c)) {
				int* res = (int*)malloc(3 * sizeof(int));
				res[0] = arr[i]; res[1] = arr[j]; res[2] = c;
				if (result.size == 0) {
					result.arr = (int**)malloc(sizeof(int*));
					result.arr[0] = res;
					result.size++;
				}
				else {
					result.arr = (int**)realloc(result.arr,sizeof(int*)*(result.size+1));
					result.arr[result.size] = res;
					result.size++;
				}
				erase(set,c);// 三元组元素c去重
			}
			else {
				voidreflection(set, arr[j]);
			}
		}
	}
}

// 三数之和_双指针法
void solution_doublepoint(int* arr, int n, ThreeArray& result) {
	
	// 先对数组进行排序
	qsort(arr, n, sizeof(int), cmp);
	result.size = 0;
	PrintArr(arr, n);

	for (int i = 0; i < n; i++) {
		if (arr[i] > 0) {
			break;
		}
		if (i > 0 && arr[i] == arr[i - 1]) {
			continue;
		}

		int left = i + 1;
		int right = n - 1;
		while (left < right) { 
			if (arr[i] + arr[left] + arr[right] > 0) {
				right--;
			}
			else if (arr[i] + arr[left] + arr[right] < 0) {
				left++;
			}
			else { // 保存到三元组
				int* res = (int*)malloc(3 * sizeof(int));
				res[0] = arr[i]; res[1] = arr[left]; res[2] = arr[right];
				if (result.size == 0) {
					result.arr = (int**)malloc(sizeof(int*));
					result.arr[0] = res;
					result.size++;
				}
				else {
					result.arr = (int**)realloc(result.arr, sizeof(int*) * (result.size + 1));
					result.arr[result.size] = res;
					result.size++;
				}

				// 去重逻辑应该放在找到一个三元组之后，对b 和 c去重
				while (left < right && arr[right] == arr[right - 1]) {
					right--;
				}
				while (left < right && arr[left] == arr[left + 1]) {
					left++;
				}

				// 找到答案时，双指针同时收缩
				right--;
				left++;
			}
		}
	}

	return;
}

// 打印数组
void PrintArr(int* A, int n) {
	for (int i = 0; i < n; i++) {
		printf("%4d ", A[i]);
	}
	printf("\n");
	for (int i = 0; i < n; i++) {
		printf("-----");
	}
	printf("\n");
}

// 打印三元组
void PrintThreeArray(ThreeArray& A) {
	for (int i = 0; i < A.size; i++) {
		for (int j = 0; j < 3; j++) {
			printf("%4d ", A.arr[i][j]);
		}
		printf("\n");
	}
	printf("\n");
}

int main() {
	
	int a1[] = { 0,52,98,-99,15,-52,52,52,52,52,52,105, 1, -106,400,395,84,77,89,512,45,88,9,1034 };
	int n1 = sizeof(a1) / sizeof(int);

	ThreeArray res;
	printf("初始数组：\n");
	PrintArr(a1, n1);
	printf("\n");

	// 哈希表法
	solution_hash(a1, n1, res);
	printf("\n哈希表法：\n\n");
	PrintThreeArray(res);

	// 双指针法
	ThreeArray res2;
	solution_doublepoint(a1, n1, res2);
	printf("\n双指针法：\n\n");
	PrintThreeArray(res2);

	return 0;
}

五、运行结果