python – 建立一個保留重複值的列表的補碼

給定列表a = [1,2,2,3],其子列表b = [1,2]以這樣一種排序(a)==排序(b補碼)的方式找到一個補全b的列表.在上面的例子中,補碼將是[2,3]的列表.

使用列表解析是很誘人的：

complement = [x for x in a if x not in b]

或設定：

complement = list(set(a) - set(b))

然而,這兩種方式都將返回complement = [3].

一個明顯的做法是：

complement = a[:]
for element in b:
complement.remove(element)

但是,這種感覺非常不滿意,而且不是非常棒的.我錯過了一個明智的成語嗎？

正如下面所指出的那樣,效能是O(n ^ 2)是否有更有效的方式？

只有更多的宣告性和因此的Pythonic方式才能進入我的腦海,並提高大b(和a)的效能是使用某種減法計數器：

from collections import Counter

class DecrementCounter(Counter):

def decrement(self,x):
if self[x]:
self[x] -= 1
return True
return False

現在我們可以使用列表解析：

b_count = DecrementCounter(b)
complement = [x for x in a if not b_count.decrement(x)]

這裡我們跟蹤b中的計數,對於我們檢視的每個元素是否是b_count的一部分.如果確實如此,我們減少計數器並忽略該元素.否則我們將其新增到補全.請注意,只有當我們確信這樣的補充存在時,這才有效.

構建補碼後,可以檢查補碼是否存在：

not bool(+b_count)

如果這是False,那麼這樣的補碼不能被構造(例如a = [1]和b = [1,3]).所以全面實施可能是：

b_count = DecrementCounter(b)
complement = [x for x in a if not b_count.decrement(x)]
if +b_count:
raise ValueError('complement cannot be constructed')

如果字典查詢在O(1)中執行(通常情況下,僅在極少數情況下為O(n)),則該演算法執行在O(| a | | b |)中(因此,列表).而刪除方法通常會在O(| a |×| b |)中執行.