Java学习手册：（数据结构与算法-数组）4Sum（leetcode454）

题目：

给出四个整型数组A、B、C、D，寻扎有多少i、j、k、l的组合使得A[i] + B[j] + C[k] + D[l] == 0。其中，A、B、C、D中均含有相同的元素个数N，且0<=N<=500。

方法一：

暴力解法：O(n^4)
500^4 = 625,0000,0000

方法二：

将D中的元素放入查找表：O(n^3)
500^3 = 1,2500,0000

方法三：

将C+D的每一种可能放入查找表：O(n^2)
500^2 = 25,0000
采用Map集合，记录每一种和的可能性，每一种和出现多少次。
时间复杂度：O(n^2)
空间复杂度：O(n^2)

方法三代码：

package com.haobi;

import java.util.HashMap;
import java.util.Map;

public class FourSum {
	public static void main(String[] args) {
		int[] A = {1,-1,2,-2,3,-3};
		int[] B = {1,-1,2,-2,3,-3};
		int[] C = {1,-1,2,-2,3,-3};
		int[] D = {1,-1,2,-2,3,-3};
		System.out.println(fourSumCount(A, B, C, D));
	}
	
	public static int fourSumCount(int[] A, int[] B, int[] C, int[] D) {
		Map<Integer, Integer> record = new HashMap<>();
		//初始化查找表(由C、D数组构建查找表)
		for(int i=0;i<C.length;i++) {
			for(int j=0;j<D.length;j++) {
				if(record.containsKey(C[i]+D[j])) {
					int old_value = record.get(C[i]+D[j]);
					int new_value = old_value + 1;
					record.put(C[i]+D[j], new_value);
				}else {
					record.put(C[i]+D[j], 1);
				}
			}
		}
		int res = 0;
		for(int i=0;i<A.length;i++) {
			for(int j=0;j<B.length;j++) {
				//如果查找表中有目标元素
				if(record.containsKey(0-A[i]-B[j])) {
					res += record.get(0-A[i]-B[j]);
				}
			}
		}
		return res;
	}
}

方法四：

将A+B和C+D的每一种可能放入两个查找表：O(n^2)