检查整数数组是否可以使用 Python 中的递归进行均衡

Question

给定大小为 5 且具有任意正值的 integer array 和 m，我的任务是创建一个程序来检查是否有可能使这个数组相等（所有整数都相等）使用以下过程：

以m为当前位置，它会跳转到另一个整数，通过跳转，该位置的整数会根据该整数位置与m的距离而减少。

Example: [2, 2, 4, 2, 2] with m = 0 (first position)
          
we jump to  the third position to equalize the integer array, that means 4 minus (distance from current position to 4)
   
   v (m = 0)
  [2, 2, 4, 2, 2]

          v jump to index 2 (3rd position)
  [2, 2, (4 - abs(2-0)), 2, 2]

= [2, 2, 2, 2, 2] (all integer is equal)

since there is a way to equalize the array, return True.

数组中的整数只能是正数。如果没有办法使数组相等，return False.

我设法编写了一个代码，但它不适用于隐藏的案例（这是我们学校的问题）。

def canEqual(array, m):

    if min(array) == max(array): return True     # Array can be Equalized, Return True

    for integer in array:   
        if integer < 1: return False             # If there is an integer in the array
                                                 # less than 1, then return False since
                                                 # it cannot be.

    for index, integer in enumerate(array):           # Treats the min value in the array as the
        if integer > min(array) and index != m:       # basis for equalizing. If an element is greater
            temp = array[:]                           # than the minimum, the program will jump to this position
            temp[index] = integer-abs(m-index)        # and the cycle continues until either the array
                                                      # is equalized or there exists a non positive integer.

            if canEqual(temp, index): return canEqual(temp, index)

我不确定我解决这个问题的方法是否正确。

Answer 1

看一看（并尝试通过调查更好地理解它），取消注释这两个打印以更好地理解该过程，如果您觉得不明白，请寻求澄清。

def canEqual(cur_arr, m):
    if min(cur_arr) < 0:
        return False
    if min(cur_arr) == max(cur_arr):
        return True
    else:
        for iter, item in enumerate(cur_arr):
            if iter!=m:
                cur_arr[iter] = cur_arr[iter]-abs(m-iter)
                # print(iter)
                # print(cur_arr)
                if canEqual(cur_arr, iter):
                    return True
                else:
                    cur_arr[iter] = cur_arr[iter]+abs(m-iter)
    return False
print(canEqual([2, 2, 4, 2, 2], 2))

输出：

True

检查整数数组是否可以使用 Python 中的递归进行均衡

Check if an Integer Array can be Equalized using Recursion in Python

python

arrays

recursion

search

integer