25-代码随想录第454题.四数相加II

25-代码随想录第454题.四数相加II

给定四个包含整数的数组列表 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 。所有整数的范围在 -2^28 到 2^28 - 1 之间，最终结果不会超过 2^31 - 1 。

例如:

输入:

A = [ 1, 2]
B = [-2,-1]
C = [-1, 2]
D = [ 0, 2]
输出:

2

代码

def find(A,B,C,D):
    x=0
    for i in A:
        for j in B:
            for m in C:
                for h in D:
                    r=int(i)+int(j)+int(m)+int(h)
                    if r==0:
                        x+=1
    print(x)
#def find1(A,B,C,D):




if __name__=='__main__':
    A=input("A=").split()
    # print(A)
    # for i in A:
    #     print(int(i))
    B = input("B=").split()
    C = input("C=").split()
    D = input("D=").split()
    find(A,B,C,D)
    #find1(A,B,C,D)

结果