利用python寻找列表中相邻元素之差绝对值为1的所有最长链

zhzhx

一、前言：


今天工作遇到一个问题，根据不同的柜子选取能连接在一起的所有柜子，有可能一组，也有可能几组；把这个问题简单化，就是给你一组列表[2, 1, 3, 4, 5, 7, 9, 8, 12, 13, 14, 15]，得到相邻两个数绝对值为1的列表组合，结果：[[5, 4, 3, 2, 1], [7, 8, 9], [12, 13, 14, 15]]

二、思路：


1、遍历列表，寻找每个元素与下一个元素之差绝对值为1的元素，例如：[2,1],[1,2,3]
2、寻找完毕后，发现如下结果：[[2, 1], [3, 2, 1], [4, 3, 2, 1], [5, 4, 3, 2, 1], [7, 8, 9], [12, 13, 14, 15]]，需要把小列表删除，比如：[3,2,1]包含[2,1]，所以删除[2,1]

三、代码：

[Python] 纯文本查看 复制代码

list1 = [2, 1, 3, 4, 5, 7, 9, 8, 12, 13, 14, 15]
 
 
# 递归，寻找符合条件的所有元素元素
def get_li(list1, l, con_list):
    w1_list = list1.copy()
    w2 = list1.copy()
 
    for i in con_list:
        w2.remove(i)
    if len(w2) == 0:
        return con_list
 
    for wl in w2:
        if abs(l - wl) == 1:
            con_list.append(wl)
            return get_li(w1_list, wl, con_list)
    return con_list
 
# 得到最终的链列表
def connect_l(list1):
    existlist = []
    ds_list = []
 
    for l in list1:
        if l not in existlist:
            ws = get_li(list1, l, [l])
            ds_list.append(ws)
 
            for e in ws:
                existlist.append(e)
 
    return ds_list
 
 
li = connect_l(list1)
 
# 删除小列表，如：[3,2,1]包含[1,2],删除[1,2]得到最优列表组合
ret_li = li.copy()
for i in range(len(li) - 1):
    if [c for c in li[i] if c in li[i + 1]]:
        if li[i] in ret_li:
            ret_li.remove(li[i])
 
print(ret_li)

结果：

[Python] 纯文本查看 复制代码

[[5, 4, 3, 2, 1], [7, 8, 9], [12, 13, 14, 15]]

cloud2010

题目看着面熟

wkdxz

[Python] 纯文本查看 复制代码

#我试了下先排序再查找

ls = [2, 1, 3, 4, 5, 7, 9, 8, 12, 13, 14, 15]
sortedLs = sorted(ls)

smallLs = []
bigLs = []
for n in range(len(sortedLs)):
    if n > 0 and abs(sortedLs[n] - sortedLs[n - 1]) == 1:
        if sortedLs[n - 1] not in smallLs:
            smallLs.append(sortedLs[n - 1])
        if sortedLs[n] not in smallLs:
            smallLs.append(sortedLs[n])
    else:
        smallLs = []
    if smallLs and smallLs not in bigLs:
        bigLs.append(smallLs)

print(bigLs)

CrushIndex

感谢分享