且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

一个神奇的实现:计算数组尾部对称长度

更新时间:2022-09-09 16:16:40

灵光一闪,得到这一句:

all(map(lambda x:x==-1, series[-i:-mid] * series[-mid:])) # 对折,相乘,全是-1

爽到死!

完整代码如下:

import numpy as np
import matplotlib.pyplot as plt


# 函数:返回数组尾部对称长度
def symmetry(series):
    for i in range(50, 0, -2):
        mid = i // 2
        is_symmetry = all(map(lambda x:x==-1, series[-i:-mid][::-1] * series[-mid:])) # 对折,相乘,全是-1
        if is_symmetry:
            return i
    return 0


# 计算对称长度
series = np.random.choice([-1, 1], 100)
print(symmetry(series))


# 作图验证
fig, axes = plt.subplots(1, 1, sharex=True)
axes.plot(np.cumsum(series), 'rd-')
plt.show()

效果图
一个神奇的实现:计算数组尾部对称长度

本文转自罗兵博客园博客,原文链接:http://www.cnblogs.com/hhh5460/p/6581244.html,如需转载请自行联系原作者