递归python函数 - 无法计算边缘情况

时间:2017-01-11 22:16:56

标签: python python-2.7 recursion

我有以下字典:

>>> for key, details in relationships.items():
                print key, details[2]

('INVOICE', 'INVOICE') 1
('INVOICE', 'ORDER2') 0.50000000
('INVOICE', 'ORDER1') 0.01536410
('ORDER1', 'ORDER2') 0.05023163
('INVOICE', 'ORDER4') 0.00573215
('ORDER4', 'ORDER1') 0.08777898
('ORDER4', 'ORDER3') 0.01674388

这将创建以下层次结构:

INVOICE -> ORDER2
        -> ORDER1 -> ORDER2
        -> ORDER4 -> ORDER1 -> ORDER2
                  -> ORDER3

其中每个箭头代表details[2]的值。最终的关系'每个订单的发票需要计算。预期值:

> ORDER1: 0.01586726 (0.0153641 + 0.0877898 x 0.00573215) 
> ORDER2: 0.50079704 (0.5 + 0.05023163 x 0.0153641 + 0.05023163 x 0.0877898 x 0.00573215)
> ORDER3: 0.00009598 (0.01674388 x 0.00573215)
> ORDER4: 0.00573215 (0.00573215)

我对递归函数进行了以下尝试:

for invoice in final_relationships.keys():
     calculate(invoice, invoice, Decimal(1))

def calculate(orig_ord, curr_ord, contribution):
     for rel_ID, rel_details in relationships.items():
          if rel_ID[1] == curr_ord:
               if orig_ord == curr_ord:
                    contribution = Decimal(1)
               if rel_ID[0] != rel_ID[1]:
                    contribution = (contribution * rel_details[2]).quantize(Decimal('0.00000001'), rounding=ROUND_HALF_UP)
                    calculate(orig_ord, rel_ID[0], contribution)
               else:
                    final_relationships[orig_ord] += contribution
                    contribution = Decimal(0)

这为ORDER2正确计算除一个以外的所有方案。

------- ORDER2
1  # rel_ID,               curr_ord,     rel_details[2],    contribution
2  ('ORDER1', 'ORDER2')    ORDER2        0.05023163         1
3  ('INVOICE', 'ORDER1')    ORDER1        0.01536410         0.05023163
4  ('INVOICE', 'INVOICE')  INVOICE       1                  0.00077176
5  # final
6  0.00077176
7  ('ORDER4', 'ORDER1')    ORDER1        0.08777898         0.00077176
8  ('INVOICE', 'ORDER4')   ORDER4        0.00573215         0.00006774
9  ('INVOICE', 'INVOICE')  INVOICE       1                  3.9E-7
10 # final
11 0.00077215
12 ('INVOICE', 'ORDER2')   ORDER2        0.50000000         0.05023163
13 ('INVOICE', 'INVOICE')  INVOICE       1                  0.50000000
14 # final
15 0.50077215

问题在第7行,因为贡献从0.00077176而不是0.05023163开始,因为(' ORDER1',' ORDER2')的迭代不会发生第二次(第2行之后)。这是关系INVOICE -> ORDER4 -> ORDER1 -> ORDER2

如何修复此功能?我尝试重置贡献,如果" orig_ord'没有处理,但无法弄清楚把它放在哪里。如果整件事情都是愚蠢的,只要我完成工作,我就可以重写了。

1 个答案:

答案 0 :(得分:0)

我认为你必须做一棵树因为它看起来像一棵树

INVOICE -> ORDER2
        -> ORDER1 -> ORDER2
        -> ORDER4 -> ORDER1 -> ORDER2
                  -> ORDER3

所以我实施this code来制作正确的树

hierarchi = [('INVOICE', 'ORDER2'),
('INVOICE', 'ORDER1'),
('ORDER1', 'ORDER2'),
('INVOICE', 'ORDER4'),
('ORDER4', 'ORDER1'),
('ORDER4', 'ORDER3')]

def make_based_list(base):
  the_list = []
  i = 0
  for item in hierarchi:
    if(item[0] == base):
      the_list.insert(i,item)
      i+=1
  return the_list

class Node:
  def __init__(self,base):
    self.base = base
    self.children = []
    self.n_of_child = 0

  def get_child(self,base):
    found_child = Node('NOT_FOUND')
    for child in self.children:
      if child.base == base:
        found_child = child

    if found_child.base!='NOT_FOUND':
      return found_child
    else:
      for child in self.children:
        found_child = child.get_child(base)
        if found_child.base!='NOT_FOUND':
          return found_child
    return found_child

  def add_child(self,child_node):
    self.children.insert(self.n_of_child,child_node)
    self.n_of_child+=1

  def update_child(self,base,child_node):
    for child in self.children:
        if child.base == base:
          child = child_node
          return 1
    return 0

  def populate(self,assoc_list):
    for assoc in assoc_list:
      child_node = Node(assoc[1])
      self.add_child(child_node)

  def print_childrens(self):
    for child in self.children:
      child.print_me()

  def print_me(self):
    print("Me:"+self.base)
    if self.n_of_child>0:
      print("Mychilds:")
      self.print_childrens()
    else:
      print("No child")

top_base = 'INVOICE'
top_based_list = make_based_list(top_base)
top_node = Node(top_base)
top_node.populate(top_based_list)

for item in hierarchi:
  if item[0]!=top_base:
    the_child = top_node.get_child(item[1])
    the_parent = top_node.get_child(item[0])
    if the_child.base=='NOT_FOUND':
      the_child = Node(item[1])
    the_parent.add_child(the_child)
    top_node.update_child(item[0],the_parent)

top_node.print_me()

我是python的新手,我知道代码可以更好,但我只想帮助你 现在,我将实施您的计算并更新我的答案

我希望这对你有用