ファイルを数える
あなたは大規模なプロジェクトの一員で、チームはタスクに応じてさまざまな種類のファイルを使用しています。Pythonのコードには
.py
ファイル、メモには .txt
ファイル、データには .csv
ファイルを使っています。これらのファイルはさまざまなディレクトリやサブディレクトリに散らばっています。ある日、特定のディレクトリとそのすべてのサブディレクトリ内に存在するPythonファイルが何個あるかを調べるよう依頼されました。そこで、あなたはこの作業を自動化するPythonのスクリプトを書くことにしました。あなたが作成するスクリプトは、ディレクトリのパスとファイル拡張子の2つのパラメータを受け取ります。スクリプトはそのディレクトリとすべてのサブディレクトリを再帰的に検索し、指定された拡張子を持つファイルの数を数えます。
プログラムの入力は2行で構成されます。1行目にディレクトリのパス、2行目にファイル拡張子(ドットなし)が含まれます。
プログラムは指定された拡張子のファイルがディレクトリとそのサブディレクトリ内に何個あるかを調べ、その数を整数で出力します。
入力 | 出力 |
documents/python-projects
py | 2 |
Constraints
Time limit: 2 seconds
Memory limit: 512 MB
Output limit: 1 MB