擅长:python、mysql、java
<p>我试着想出一种矢量化的方法来使用Numpy实现这一点。这是迄今为止我想到的最好的:</p>
<pre><code>import numpy as np
def balancedSums(arr):
arr = np.array(arr)
ltr = np.cumsum(arr)
rtl = np.cumsum(arr[::-1])[::-1]
if np.any(ltr == rtl):
return "YES"
else:
return "NO"
assert(balancedSums([1, 2, 3]) == "NO")
assert(balancedSums([3, 2, 1, 2, 3]) == "YES")
assert(balancedSums([10]) == "YES")
assert(balancedSums([1, 1]) == "NO")
assert(balancedSums([0]) == "YES")
assert(balancedSums([]) == "NO")
</code></pre>