利用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]]

蒋德盼

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

list = [2, 1, 3, 4, 5, 7, 9, 8, 12, 13, 14, 15]
list.sort()
result = [];
temp = [];
for i in range(len(list)):
    if i == 0:
        temp.append(list[i])
    elif list[i] - list[i-1] == 1:
        temp.append(list[i])
    else:
        result.append(temp)
        temp = []
        temp.append(list[i])
result.append(temp)
print(result)

cloud2010

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

list1 = [2, 1, 3, 4, 5, 7, 9, 8, 12, 13, 14, 15]
list1.sort()
list2 = list(map(lambda x :x-list1.index(x), list1))
list3 = [list1[list2.index(i) :list2.index(i)+list2.count(i)] for i in set(list2)]
print(list3)

去除数量为 1 的链

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

list1 = [2, 1, 3, 4, 5, 7, 9, 8, 12, 13, 14, 15]
list1.sort()
list2 = list(map(lambda x :x-list1.index(x), list1))
list3 = [list1[list2.index(i) :list2.index(i)+list2.count(i)] for i in set(list2) if list2.count(i) > 1]
print(list3)