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

x33g5p2x  于2022-03-23 转载在 Python  
字(1.0k)|赞(0)|评价(0)|浏览(209)

一、前言:

今天工作遇到一个问题,根据不同的柜子选取能连接在一起的所有柜子,有可能一组,也有可能几组;把这个问题简单化,就是给你一组列表[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]

三、代码:

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)

结果:

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

相关文章