fork download
  1. # your code goes here
  2.  
  3. class Kamus:
  4. def __init__(self):
  5. self.graph = {}
  6.  
  7. def tambah(self, kata, sinonim):
  8. if not kata or not isinstance(sinonim, (list, tuple, set)):
  9. return
  10.  
  11. if kata not in self.graph:
  12. self.graph[kata] = set()
  13.  
  14. for s in sinonim:
  15. if not s or s == kata:
  16. continue
  17.  
  18. self.graph[kata].add(s)
  19.  
  20. if s not in self.graph:
  21. self.graph[s] = set()
  22. self.graph[s].add(kata)
  23.  
  24. def ambilSinonim(self, kata):
  25. if kata not in self.graph:
  26. return None
  27.  
  28. if len(self.graph[kata]) == 0:
  29. return [kata]
  30.  
  31. return list(self.graph[kata])
  32.  
Success #stdin #stdout 0.01s 7092KB
stdin
Standard input is empty
stdout
Standard output is empty