根据条件有效地将两个不均匀列表组合成字典

Question

我有两个元组列表，我想根据条件将e 中的元素映射到s 中的元素。条件是e 中某物的第一个元素需要>= 到s 中的第一个元素，e 中的元素1+2 需要s 中的元素1+2。每个s 元组中的第一个数字是起始位置，第二个是长度。我可以这样做：

e = [('e1',3,3),('e2',6,2),('e3',330,3)]
s = [('s1',0,10),('s2',11,24),('s3',35,35),('s4',320,29)]

for i in e:
    d = {i:j for j in s if i[1]>=j[1] and i[1]+i[2]<=j[1]+j[2]}
    print(d)

输出（我想要的）：

{('e1', 3, 3): ('s1', 0, 10)}
{('e2', 6, 2): ('s1', 0, 10)}
{('e3', 330, 3): ('s4', 320, 29)}

有没有更有效的方法来获得这个结果，理想情况下没有 for 循环（至少不是双循环）？我已经用zip 尝试了一些东西，以及一些类似的东西

list(map(lambda i,j: {i:j} if i[1]>=j[1] and i[1]+i[2]<=j[1]+j[2] else None, e, s))

但它并没有给我我想要的东西。

s 中的元素永远不会重叠。例如，您不会有 ('s1',0,10) 和 ('s2', 5,15)。换句话说，范围 (0, 0+10) 永远不会与 (5,5+15) 或任何其他元组的范围重叠。此外，e 中的所有元组都是唯一的。

Answer 1

s 中的元组不能重叠的约束非常强。特别是，这意味着e 中的每个值最多只能与s 中的一个值匹配（我认为最简单的证明方法是假设s 中的两个不同的、不重叠的元组与e 中的单个元素并导出矛盾）。

由于该属性，任何匹配 e-tuple e1 的 s-tuple s1 都具有以下属性：在s 和 sx[1]<=e1[1] 中的 所有 元组 sx 中，它是具有最大sum(sx[1:]) 的那个，因为如果不是，那么具有足够小的第一个坐标和更大的总和的 s 元组也将是匹配的（我们已经知道这是不可能的）。

该观察结果适用于一个相当简单的算法，我们线性遍历e 和s（已排序），跟踪总和最大的 s 元组。如果与我们正在查看的 e-tuple 相比，该总和足够大，则将该对添加到我们的结果集中。

def pairs(e, s):
  # Might be able to skip or replace with .sort() depending
  # on the rest of your program
  e = sorted(e, key=lambda t: t[1])
  s = sorted(s, key=lambda t: t[1])

  i, max_seen, max_sum, results = 0, None, None, {}
  for ex in e:
    while i < len(s) and (sx:=s[i])[1] <= ex[1]:
      if not max_seen or sum(sx[1:]) > max_sum:
        max_seen, max_sum = sx, sum(sx[1:])
      i += 1
    
    if max_seen and max_sum > sum(ex[1:]):
      results[ex] = s[i]

  return results