假設數據結構是這樣的：有多條數據，每條數據都有屬性parent(指向它的所屬上級id)，自身的唯一標識id。
class data
parent
id
當我拿出其中一條數據的時候，用一個方法計算出所有下級數據。就好比中國行政區數據一樣，當我拿到廣東省的時候，下面的市級，縣級都要得到，如果縣級下面還有分級也要一并拿到。
我寫了一些代碼，都太丑陋了，似乎是要用遞歸，求教一下有沒有什么好的思路？
3 回答
12345678_0001
TA貢獻1802條經驗 獲得超5個贊
這個問題可以轉換成N叉樹的遍歷，將某個節點的所有子節點計算出來。
def return_all_children(self):
"""
返回所有的子項
:return:
"""
root = self
if not root:
return []
que = [] # 保存節點的隊列
res = [] # 保存結果的列表
que.append(root) #
while len(que): # 判斷隊列不為空
length = len(que)
sub = [] # 保存每層的節點的值
for i in range(length):
current = que.pop(0) # 出隊列的當前節點
print(current)
sub.append(current)
for child in current.return_children: # 所有子結點入隊列
que.append(child)
res.append(sub) # 把每層的節點的值加入結果列表
return res
動漫人物
TA貢獻1815條經驗 獲得超10個贊
儲存空間夠大的話可以建立一個字典
假如你的數據是個list名字叫CN（中國所有省市縣...）
parent_d = {}
for item in CN:
parent = item['parent']
if parent in parent_d:
parent_d[parent].append(item['id'])
else:
parent_d[parent] = [item['id']]之后遍歷一下
def get_all_children(city_id, parent_d):
if city_id not in parent_d or not city_id:
return []
result = []
temp_parent = [city_id]
while temp:
cur_id = temp_parent.pop(-1)
result += parent_d[cur_id]
temp_parent = parent_d[cur_id] + temp_parent
return result
添加回答
舉報
0/150
提交
取消