ZeroJudge 解題筆記：f818. 物競天擇 (Survival)

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



ZeroJudge 題目連結：f818. 物競天擇 (Survival)

解題想法

這題考排序，需要自訂排序的比較式，因為比較式不像前一題那麼複雜，我是用 lambda function 寫比較式。

Python 程式碼

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

import sys

for line in sys.stdin:
    n = int(line)  # 小獅子數量 n
    hs = list(map(int, input().split()))  # 小獅子身高 h
    ws = list(map(int, input().split()))  # 小獅子體重 w
    hw = [(h, w) for h, w in zip(hs, ws)]  # 將 hs, ws 合併成二維串列，資料為 (h, w)
    hw.sort(key = lambda x : x[0]*x[1])  # 將 hw 依照 h*w 由小到大排序
    print(*hw[0])  # 最小值在首項

C++ 程式碼

因為資料只有兩項，可以用 vector 及 pair 儲存資料，不需要自己定義 struct。使用時間約為 3 ms，記憶體約為 352 kB，通過測試。

#include <iostream>
#include <vector>
#include <utility>
#include <algorithm>
using namespace std;

int main() {
    ios::sync_with_stdio(0); cin.tie(0);
    int n;  // 小獅子數量 n
    while(cin >> n) {
        vector<pair<int, int>> hw (n);  // 小獅子身高、體重
        for(int i=0; i<n; i++) cin >> hw[i].first;  // 讀取身高
        for(int i=0; i<n; i++) cin >> hw[i].second;  // 讀取體重
        sort(hw.begin(), hw.end(), [] (pair<int, int> a, pair<int, int> b) {
            return a.first*a.second < b.first*b.second; } );  // 依照 h*w 由小到大排序
        cout << hw[0].first << " " << hw[0].second << "\n";  // 最小值在首項
    }
    return 0;
}

自己定義 struct，用來儲存小獅子的身高、體重。使用時間約為 3 ms，記憶體約為 356 kB，通過測試。

#include <iostream>
#include <algorithm>
using namespace std;

struct Lion {  // 自定結構體，儲存身高、體重
    int h, w;
};

int main() {
    ios::sync_with_stdio(0); cin.tie(0);
    int n;  // 小獅子數量 n
    while(cin >> n) {
        Lion hw[n];  // 小獅子身高、體重
        for(int i=0; i<n; i++) cin >> hw[i].h;  // 讀取身高
        for(int i=0; i<n; i++) cin >> hw[i].w;  // 讀取體重
        sort(hw, hw+n, [] (Lion a, Lion b) {
            return a.h*a.w < b.h*b.w; } );  // 依照 h*w 由小到大排序
        cout << hw[0].h << " " << hw[0].w << "\n";  // 最小值在首項
    }
    return 0;
}