Подтвердить что ты не робот

Быстрый алгоритм вычисления дельта двух списков

У меня есть два списка имен альбомов, упорядоченных по некоторым баллам.

albums_today = ['album1', 'album2', 'album3']
albums_yesterday = ['album2', 'album1', 'album3']

Как я могу рассчитать изменение порядка списка и получить что-то вроде

{'album1':1, 'album2':-1, 'album3':0}
4b9b3361

Ответ 1

D = dict((title, rank) for rank, title in enumerate(albums_yesterday))
for rank, title in enumerate(albums_today):
    D[title] = D[title] - rank

Ответ 2

>>> albums_today = ['album1', 'album2', 'album3']
>>> albums_yesterday = ['album2', 'album1', 'album3']
>>> D = dict((k,v) for v,k in enumerate(albums_yesterday))
>>> dict((k,D[k]-v) for v,k in enumerate(albums_today))
{'album1': 1, 'album3': 0, 'album2': -1}

В Python2.7 или Python3 его можно записать еще проще

>>> albums_today = ['album1', 'album2', 'album3']
>>> albums_yesterday = ['album2', 'album1', 'album3']
>>> D = {k:v for v,k in enumerate(albums_yesterday)}
>>> {k:D[k]-v for v,k in enumerate(albums_today)}
{'album1': 1, 'album3': 0, 'album2': -1}

Ответ 3

вы также можете использовать тот же алгоритм, что и выше, и просто использовать один хэш файл.

def findDelta1(today,yesterday):
 results = {}
 ypos = 0
 for i,title in enumerate(today):
      if title in results:
           results[title] = results[title] - i
      else:
           for ypos in xrange(ypos,len(yesterday)):
                if yesterday[ypos] == title:
                     results[title] = ypos - i
                     ypos = ypos + 1
                     break
                else:
                     results[yesterday[ypos]] = ypos
 return results

все еще O (N), потенциально быстрее и меньше оперативной памяти, чем моя версия выше.

Ответ 4

как насчет этого:

def delta(a, b):
    rank_a = dict((k, v) for v, k in enumerate(a))
    rank_b = enumerate(b)
    return dict((k, rank_a[k]-i) for i, k in rank_b)

который создает только один диктофон для поиска вещей.

Хорошо, если каждая запись в обоих списках присутствует ровно один раз, то мы знаем, что как только мы посмотрим ключ в коллекции rank_a, нам это больше не нужно. Мы можем удалить его. Кроме того, чтобы сэкономить место, нам не нужно заполнять эту коллекцию до момента, когда нужен конкретный ключ.

class LookupOnce:
    def __init__(self, seq):
        self.cache = {}
        self.seq = iter(seq)
    def get(self, key):
        if key in self.cache:
            value = self.cache[key]
            del self.cache[key]
            return value
        for v,k in self.seq:
            if k == key:
                return v
            self.cache[k] = v
        raise KeyError


def delta(a, b):
    rank_a = LookupOnce(enumerate(a))
    rank_b = enumerate(b)
    result = {}
    for i, k in rank_b:
        result[k] = i - rank_a.get(k)
    return result

Ответ 5

>>> def transform(albums):
...     return dict((album, i) for i, album in enumerate(albums))
... 
>>> def show_diffs(album1, album2):
...     album_dict1, album_dict2  = transform(album1), transform(album2)
...     for k, v in sorted(album_dict1.iteritems()):
...         print k, album_dict2[k] - v
... 
>>> albums_today = ['album1', 'album2', 'album3']
>>> albums_yesterday = ['album2', 'album1', 'album3']
>>> show_diffs(albums_today, albums_yesterday)
album1 1
album2 -1
album3 0

Ответ 6

хорошо, в зависимости от размера ваших списков, существует несколько разных подходов. не зная, насколько велики ваши наборы данных, я бы предположил, что самый простой (возможно, излишне оптимизированный) метод выглядит примерно так:

albums_yesterday_lookup = new HashMap();
differences = new HashMap();
foreach(albums_yesterday as position => album_title)
    albums_yesterday_lookup.put(album_title,position);

foreach(albums_today as position => album_title)
    differences.put(album_title, albums_yesterday_lookup.get(album_title) - position);

который работает как O (N).

Ответ 7

Новый и улучшенный, а не O (n 2): Но все же медленнее, чем два других ответа.

Единственное преимущество этого решения - экономия памяти. Он избегает создания большого дикта и вместо этого хранит только то, что необходимо в то время. Второе решение TokenMacGuy делает это, но это немного быстрее.

def get_deltas_aas(today, yesterday):
    deltas = {}
    for (new_rank, new_album), (old_rank, old_album) in \
            itertools.izip(enumerate(today), enumerate(yesterday)):
        if old_album in deltas:
            #Believe it or not, this is faster than deltas.pop(old_album) + old_rank
            yield (old_album, deltas[old_album] + old_rank)
            del deltas[old_album]    
        else:
            deltas[old_album] = old_rank

        if new_album in deltas:
            yield (new_album, deltas[new_album] - new_rank)
            del deltas[new_album]
        else:
            deltas[new_album] = -new_rank

Здесь приведены некоторые результаты синхронизации для большинства ответов здесь (все из них на Python, если я не пропустил что-то). dict действует. Если кто-то хочет, чтобы я каким-то образом менял свой код, просто пингуйте меня.

get_deltas_token1: 1.08131885529 msecs
get_deltas_gnibbler: 1.06443881989 msecs
get_deltas_tyler: 1.61993408203 msecs
get_deltas_token2: 1.52525019646 msecs
get_deltas_hughdbrown: 3.27240777016 msecs
get_deltas_aas: 1.39379096031 msecs

Код, который я использовал для синхронизации, здесь. Я доволен временными рамками, которые я выбрасывал для него сверху времени. Должен быть полезен в будущем после реорганизации кода для запуска тестов.