ZeroJudge 解題筆記：g796. 檔案分類 (Files)

作者：王一哲
日期：2025年3月17日



ZeroJudge 題目連結：g796. 檔案分類 (Files)

解題想法

這題用字典計數比較方便，如果用 Python 可以用 defaultdict 就不需要檢查 key 值是否存在，但是在輸出答案時要先按照 key 值由小到大排序；如果用 C++ 的 map，本來就會按照 key 值由小到大排序，直接輸出就好。

Python 程式碼

使用時間約為 50 ms，記憶體約為 3.7 MB，通過測試。

import sys
from collections import defaultdict

for line in sys.stdin:
    n = int(line)  # 檔案數量
    cnt = defaultdict(int)  # 計數器
    for _ in range(n):  # 執行 n 次
        cnt[int(input().strip()[3:])//10] += 1  # 讀取字串，去掉 \n，取後 3 位轉成 int
    for k, v in sorted(cnt.items()): print(k, v)  # 印出答案

C++ 程式碼

使用時間約為 3 ms，記憶體約為 364 kB，通過測試。

#include <iostream>
#include <string>
#include <map>
using namespace std;

int main() {
    ios::sync_with_stdio(0); cin.tie(0);
    int n;  // 檔案數量
    string s;  // 檔案編號
    while(cin >> n) {
        map<int, int> cnt;  // 計數器
        for(int i=0; i<n; i++) {  // 執行 n 次
            cin >> s;  // 讀取編號
            cnt[stoi(s.substr(3))/10]++;  // s 取後 3 位轉成 int
        }
        for(auto it : cnt) cout << it.first << " " << it.second << "\n";  // 印出答案
    }
    return 0;
}