mirror of
https://github.com/heqin-zhu/algorithm.git
synced 2024-03-22 13:30:46 +08:00
69 lines
1.7 KiB
Python
69 lines
1.7 KiB
Python
''' mbinary
|
|
#########################################################################
|
|
# File : sort.py
|
|
# Author: mbinary
|
|
# Mail: zhuheqin1@gmail.com
|
|
# Blog: https://mbinary.coding.me
|
|
# Github: https://github.com/mbinary
|
|
# Created Time: 2018-07-05 17:18
|
|
# Description:
|
|
#########################################################################
|
|
'''
|
|
|
|
from functools import total_ordering
|
|
|
|
@total_ordering
|
|
class node:
|
|
def __init__(self,val,left=None,right=None):
|
|
self.val=val
|
|
self.frequency = 1
|
|
self.left=left
|
|
self.right=right
|
|
def __lt__(self,x):
|
|
return self.val<x.val
|
|
def __eq__(self,x):
|
|
return self.val==x.val
|
|
def inc(self):
|
|
self.val+=1
|
|
def dec(self):
|
|
self.val-=1
|
|
def incFreq(self):
|
|
self.frequency +=1
|
|
def decFreq(self):
|
|
self.frequency -=1
|
|
|
|
class binaryTree:
|
|
def __init__(self,reverse = True):
|
|
self.reverse = reverse
|
|
self.data=None
|
|
def cmp(self,n1,n2):
|
|
ret=0
|
|
if n1 < n2: ret=-1
|
|
if n1 > n2: ret= 1
|
|
return ret * -1 if self.reverse else ret
|
|
def addNode(self,nd):
|
|
def _add(prt,chd):
|
|
if self.cmp(prt,chd)==0:
|
|
prt.incFreq()
|
|
return
|
|
if self.cmp(prt,chd)<0:
|
|
|
|
if not isinstance(nd,node):nd=node(nd)
|
|
if not self.root :
|
|
self.root=node(val)
|
|
else:
|
|
if self.root == val:
|
|
self.root.incfreq()
|
|
else:
|
|
cur = self.root
|
|
def build(self,lst):
|
|
dic = {}
|
|
for i in lst:
|
|
if i in dic:
|
|
dic[i].incFreq()
|
|
else:
|
|
dic[i] = node(i)
|
|
self.data =list( dic.values())
|
|
|
|
|