LeetCode-Python-1131. 绝对值表达式的最大值

 

给你两个长度相等的整数数组，返回下面表达式的最大值：

|arr1[i] - arr1[j]| + |arr2[i] - arr2[j]| + |i - j|

其中下标 i，j 满足 0 <= i, j < arr1.length。

 

示例 1：

输入：arr1 = [1,2,3,4], arr2 = [-1,4,5,6]
输出：13

示例 2：

输入：arr1 = [1,-2,-5,0,10], arr2 = [0,-2,-1,-7,-4]
输出：20

 

提示：

• 2 <= arr1.length == arr2.length <= 40000
• -10^6 <= arr1[i], arr2[i] <= 10^6

第一种思路：

此题乍一看很简单，直接上双重循环的暴力解即可，

然而数据规模较大，结果超时。

class Solution(object):
    def maxAbsValExpr(self, arr1, arr2):
        """
        :type arr1: List[int]
        :type arr2: List[int]
        :rtype: int
        """
        res = 0
        for i in range(len(arr1)):
            for j in range(len(arr2)):
                res = max(res, abs(arr1[i] - arr1[j]) + abs(arr2[i] - arr2[j]) + abs(i - j))
        return res

第二种思路：

先思考一下子问题怎么处理：

1. 如何求 abs（arr1[ i ] - arr1[ j ])的最大值？

显然易知， 答案是 max(arr1) - min(arr1)

2. 如何求 abs( arr1[ i ] - arr1[ j ]) + abs(i - j)的最大值？

emmm 并不容易直观地看出答案，不妨分析一番：

表达式 去掉绝对值符号之后，一共有四种可能：

abs( arr1[ i ] - arr1[ j ]) + abs(i - j) 
=  arr1[i] - arr1[j] + i - j =  (arr1[i] + i) - (arr1[j] + j)
=  arr1[i] - arr1[j] - i + j =  (arr1[i] - i) - (arr1[j] - j)
= -arr1[i] + arr1[j] + i - j = -(arr1[i] - i) + (arr1[j] - j)
= -arr1[i] + arr1[j] - i + j = -(arr1[i] + i) + (arr1[j] + j)

再观察一下，可以发现，因为 i 和 j 是可以互换的，第一个分解 等价于 第四个分解， 第二个分解 等价于 第三个分解；

所以此表达式的值等价于：

abs( arr1[ i ] - arr1[ j ]) + abs(i - j) 
=  arr1[i] - arr1[j] + i - j =  (arr1[i] + i) - (arr1[j] + j)
=  arr1[i] - arr1[j] - i + j =  (arr1[i] - i) - (arr1[j] - j)
只需保留前两种分解即可。

因此：
abs( arr1[ i ] - arr1[ j ]) + abs(i - j)的最大值
= max((arr1[i] + i) - (arr1[j] + j),
      (arr1[i] - i) - (arr1[j] - j))

现在的表达式里只由两部分组成：

a. arr1[i] + i

b. arr1[i] - i

我们不妨把 arr1[i] + i 看做一个整体 表达式A，把 arr1[i] - i 看做一个整体 表达式B， 那么问题变成了：

abs( arr1[ i ] - arr1[ j ]) + abs(i - j)的最大值
= max((arr1[i] + i) - (arr1[j] + j),
      (arr1[i] - i) - (arr1[j] - j))

= max(max(A) - min(A),
      max(B) - min(B))

所以我们只需要找到A（arr1[i] + i ）的所有取值和B（arr1[i] - i ）的所有取值，那么就可以计算出 abs( arr1[ i ] - arr1[ j ]) + abs(i - j)的最大值 啦

子问题代码如下：

class Solution(object):
    def maxAbsValExpr(self, arr1, arr2):
        """
        :type arr1: List[int]
        :type arr2: List[int]
        :rtype: int
        """
        A = []
        B = []
        for i, x in enumerate(arr1):
            A.append(x + i)
            B.append(x - i)
        # print A, B
        return max(max(A) - min(A), max(B) - min(B))

回到本题，类似于上面分析的子问题的解法，也可以把绝对值符号去掉，然后用组合的办法解题。

class Solution(object):
    def maxAbsValExpr(self, arr1, arr2):
        """
        :type arr1: List[int]
        :type arr2: List[int]
        :rtype: int
        """
        A, B, C, D, E, F, G, H = [], [], [], [], [], [], [], []
        for i in range(len(arr1)):
            x, y = arr1[i], arr2[i]
            A.append(x + y + i)
            B.append(x + y - i)
            C.append(x - y + i)
            D.append(x - y - i)
            E.append(-x + y + i)
            F.append(-x + y - i)
            G.append(-x - y + i)
            H.append(-x - y - i)
            
        a = max(A) - min(A)
        b = max(B) - min(B)
        c = max(C) - min(C)
        d = max(D) - min(D)
        e = max(E) - min(E)
        f = max(F) - min(F)
        g = max(G) - min(G)
        h = max(H) - min(H)
        
        return max(a, b, c, d, e, f, g, h)

又因为类似子问题，八种分解里存在四对等价的分解，因此代码可以继续优化 

|arr1[i] - arr1[j]| + |arr2[i] - arr2[j]| + |i - j|

 =  (arr1[i] + arr2[i] + i) - (arr1[j] + arr2[j] + j)
 =  (arr1[i] + arr2[i] - i) - (arr1[j] + arr2[j] - j)
 =  (arr1[i] - arr2[i] + i) - (arr1[j] - arr2[j] + j)
 =  (arr1[i] - arr2[i] - i) - (arr1[j] - arr2[j] - j)
 = -(arr1[i] + arr2[i] + i) + (arr1[j] + arr2[j] + j)
 = -(arr1[i] + arr2[i] - i) + (arr1[j] + arr2[j] - j)
 = -(arr1[i] - arr2[i] + i) + (arr1[j] - arr2[j] + j)
 = -(arr1[i] - arr2[i] - i) + (arr1[j] - arr2[j] - j)

优化为四个表达式：
A = arr1[i] + arr2[i] + i
B = arr1[i] + arr2[i] - i
C = arr1[i] - arr2[i] + i
D = arr1[i] - arr2[i] - i

class Solution(object):
    def maxAbsValExpr(self, arr1, arr2):
        """
        :type arr1: List[int]
        :type arr2: List[int]
        :rtype: int
        """
        A, B, C, D, E, F, G, H = [], [], [], [], [], [], [], []
        for i in range(len(arr1)):
            x, y = arr1[i], arr2[i]
            A.append(x + y + i)
            B.append(x + y - i)
            C.append(x - y + i)
            D.append(x - y - i)
            # E.append(-x + y + i)
            # F.append(-x + y - i)
            # G.append(-x - y + i)
            # H.append(-x - y - i)
            
        a = max(A) - min(A)
        b = max(B) - min(B)
        c = max(C) - min(C)
        d = max(D) - min(D)
        # e = max(E) - min(E)
        # f = max(F) - min(F)
        # g = max(G) - min(G)
        # h = max(H) - min(H)
        
        return max(a, b, c, d)