注册一亩三分地论坛,查看更多干货!
您需要 登录 才可以下载或查看附件。没有帐号?注册账号 
x
安贤量化
笔试是 python 和东京交易所的数据相关,有 hidden test case
是 202407 左右面的- class Trie: def init(self): self.v = None self.children = {} def setitem(self, key, value): if key.endswith('.'): raise ValueError("e") parts = key.split('.') cur = self for part in parts[:-1]: if part not in cur.children: raise ValueError("e") cur = cur.children[part] if parts[-1] not in cur.children: cur.children[parts[-1]] = Trie() cur.children[parts[-1]].v = value
- def get_v(self, key): parts = key.split('.') cur = self for part in parts: cur = cur.children[part] return cur.v def get_d(self, key): parts = key.split('.') cur = self for part in parts[:-1]: cur = cur.children[part] return cur def getitem(self, key): if key.endswith('.'): return self.get_d(key) return self.get_v(key)
- def repr(self): return repr({k: v.v if v.v is not None else {} for k, v in self.children.items()})
- d = Trie() d["a"] = 1 d["a.b"] = 2 d["a.c"] = 3 print(d["a"]) print(d["a."]) print(d["a."]["b"])
- d["a.b.c"] = 3 print(d["a."]["b."]) print(d["a."]["b.c"]) print(d["a."]["b."]["c"])
复制代码 |