Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

โปรแกรม Python สร้าง Binomial Tree


เมื่อต้องการสร้าง Binomial Tree ใน Python ใช้วิธี Object-Oriented กำหนดคลาสและแอตทริบิวต์ กำหนดฟังก์ชันภายในคลาสที่ดำเนินการบางอย่าง สร้างอินสแตนซ์ของคลาส และใช้ฟังก์ชันดำเนินการ

ตัวอย่าง

class binomial_tree:
    def __init__(self, key):
        self.key = key
        self.children = []
        self.order = 0

    def add_at_end(self, t):
        self.children.append(t)
        self.order = self.order + 1

my_tree = []
print('Menu')
print('create ')
print('combine  ')
print('exit')
while True:
    option = input('What do you wish like to do? ').split()
    operation = option[0].strip().lower()
    if operation == 'create':
        key = int(option[1])
        b_tree = binomial_tree(key)
        my_tree.append(b_tree)
        print('Binomial tree has been created.')
    elif operation == 'combine':
        index_1 = int(option[1])
        index_2 = int(option[2])
        if my_tree[index_1].order == my_tree[index_2].order:
            my_tree[index_1].add_at_end(my_tree[index_2])
            del my_tree[index_2]
            print('Binomial trees have been combined.')
        else:
            print('Order of trees need to be the same to combine them.')
    elif operation == 'exit':
        print("Exit")
        break
    print('{:>8}{:>12}{:>8}'.format('Index', 'Root key', 'Order'))
    for index, t in enumerate(my_tree):
        print('{:8d}{:12d}{:8d}'.format(index, t.key, t.order))

ผลลัพธ์

Menu
create 
combine  
exit
What do you wish like to do? create 7
Binomial tree has been created.
Index Root key Order
0 7 0
What do you wish like to do? create 11
Binomial tree has been created.
Index Root key Order
0 7 0
1 11 0
What do you wish like to do? create 4
Binomial tree has been created.
Index Root key Order
0 7 0
1 11 0
2 4 0
What do you wish like to do? combine 0 1
Binomial trees have been combined.
Index Root key Order
0 7 1
1 4 0
What do you wish like to do? exit
Exit

คำอธิบาย

  • กำหนดคลาส 'binomial_tree'
  • มีเมธอดเพิ่มองค์ประกอบท้ายของ tree
  • สร้างลิสต์ว่าง
  • ตามตัวเลือก ผู้ใช้เลือกตัวเลือก
  • หากเลือกสร้าง key สร้างอินสแตนซ์ของคลาส และสร้าง binomial tree
  • คำนวณ index, root value และ order
  • หากต้องการรวม indices เลือกตัวเลือกอื่น และระบุ index ของโหนดที่ต้องการรวม
  • รวมข้อมูล และแสดงผล