diff --git a/leetcode_helper.py b/leetcode_helper.py index fa9b007a2cb48c45d8ac490a23bd4340a243aa89..4636d456ab981faffd2e791c8cb78b1b77c2baf3 100644 --- a/leetcode_helper.py +++ b/leetcode_helper.py @@ -3,6 +3,13 @@ import re import shutil import json import uuid +import argparse +import collections +parser = argparse.ArgumentParser() +parser.add_argument("--run", type=str, help="Decide to run which function") +args = parser.parse_args() +helper_function = args.run + def get_files_path(file_dir, filetype='.txt'): """得到文件夹下的所有.txt文件的路径 @@ -175,5 +182,44 @@ def count_tag_class(): print(tags_set) -leetcode_helper_update_md() -leetcode_helper_update_config() \ No newline at end of file +def count_exercises(): + dirs = ['data/2.算法中阶', 'data/3.算法高阶'] + exercises_ids = [] + for dir in dirs: + dirs_ = os.listdir(dir) + algo_floor_dirs = [] + for algo_floor_dir in dirs_: + leetcode_class_dir = os.path.join(dir, algo_floor_dir) + if os.path.isdir(leetcode_class_dir): + algo_floor_dirs.append(leetcode_class_dir) + exercises_dirs = [] + for algo_floor_dir in algo_floor_dirs: + exercises_dirs_ = os.listdir(algo_floor_dir) + for exercises_dir_ in exercises_dirs_: + exercises_dir = os.path.join(algo_floor_dir, exercises_dir_) + if os.path.isdir(exercises_dir): + exercises_dirs.append(exercises_dir) + for exercises_dir in exercises_dirs: + exercises_id = int(exercises_dir.split('/')[-1].split('_')[0]) + exercises_ids.append(exercises_id) + try: + assert len(set(exercises_ids)) == len(exercises_ids) + except: + print(collections.Counter(exercises_ids)) + print('------分割线-------') + dst_exercises_ids = [i for i in range(200)] + lacked_id = set(dst_exercises_ids) - set(exercises_ids) + print(lacked_id) + + + + + +if helper_function == 'count_tag_class': + count_tag_class() +if helper_function == 'count_exercises': + count_exercises() + + +# leetcode_helper_update_md() +# leetcode_helper_update_config() \ No newline at end of file diff --git a/src/__pycache__/tree.cpython-38.pyc b/src/__pycache__/tree.cpython-38.pyc deleted file mode 100644 index b034f05a92591857f96e5847b263a3032e6f682d..0000000000000000000000000000000000000000 Binary files a/src/__pycache__/tree.cpython-38.pyc and /dev/null differ