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 |)中執行.
http://stackoverflow.com/questions/42273686/create-a-complement-of-list-preserving-duplicate-values