|
14 | 14 | }, |
15 | 15 | { |
16 | 16 | "cell_type": "code", |
17 | | - "execution_count": null, |
18 | | - "metadata": { |
19 | | - "collapsed": true |
20 | | - }, |
| 17 | + "execution_count": 1, |
| 18 | + "metadata": {}, |
21 | 19 | "outputs": [], |
22 | 20 | "source": [ |
23 | 21 | "class prefix_tree:\n", |
|
26 | 24 | " self.root = [{}]\n", |
27 | 25 | " \n", |
28 | 26 | " def add(self, string):\n", |
| 27 | + " if self.check(string):\n", |
| 28 | + " return\n", |
| 29 | + " wrk_dict = self.root\n", |
| 30 | + " for i in string:\n", |
| 31 | + " if i in wrk_dict[0]:\n", |
| 32 | + " wrk_dict = wrk_dict[0][i]\n", |
| 33 | + " else:\n", |
| 34 | + " wrk_dict[0][i] = [{}]\n", |
| 35 | + " wrk_dict = wrk_dict[0][i]\n", |
| 36 | + " wrk_dict.append(True)\n", |
29 | 37 | " #TODO добавить строку\n", |
30 | 38 | " \n", |
31 | 39 | " def check(self, string):\n", |
| 40 | + " wrk_dict = self.root\n", |
| 41 | + " for i in string:\n", |
| 42 | + " if i in wrk_dict[0]:\n", |
| 43 | + " wrk_dict = wrk_dict[0][i]\n", |
| 44 | + " else:\n", |
| 45 | + " return False\n", |
| 46 | + " if len(wrk_dict) != 1:\n", |
| 47 | + " return True\n", |
| 48 | + " return False\n", |
32 | 49 | " #TODO проверить наличие строки\n", |
33 | 50 | " \n", |
34 | 51 | " def check_part(self, string):\n", |
| 52 | + " wrk_dict = self.root\n", |
| 53 | + " for i in string:\n", |
| 54 | + " if i in wrk_dict[0]:\n", |
| 55 | + " wrk_dict = wrk_dict[0][i]\n", |
| 56 | + " else:\n", |
| 57 | + " return False\n", |
| 58 | + " return True\n", |
35 | 59 | " #TODO проверить наличие строки как префикса" |
36 | 60 | ] |
37 | 61 | }, |
38 | 62 | { |
39 | 63 | "cell_type": "code", |
40 | | - "execution_count": null, |
41 | | - "metadata": { |
42 | | - "collapsed": true |
43 | | - }, |
| 64 | + "execution_count": 2, |
| 65 | + "metadata": {}, |
44 | 66 | "outputs": [], |
45 | 67 | "source": [ |
46 | 68 | "pr_tree = prefix_tree()\n", |
|
56 | 78 | "metadata": { |
57 | 79 | "anaconda-cloud": {}, |
58 | 80 | "kernelspec": { |
59 | | - "display_name": "Python [default]", |
| 81 | + "display_name": "Python 3", |
60 | 82 | "language": "python", |
61 | 83 | "name": "python3" |
62 | 84 | }, |
|
70 | 92 | "name": "python", |
71 | 93 | "nbconvert_exporter": "python", |
72 | 94 | "pygments_lexer": "ipython3", |
73 | | - "version": "3.5.2" |
| 95 | + "version": "3.6.5" |
74 | 96 | } |
75 | 97 | }, |
76 | 98 | "nbformat": 4, |
|
0 commit comments