【劍指Offer系列45】把數組排成最小的數

題目

輸入一個正整數數組，把數組裏所有數字拼接起來排成一個數，打印能拼接出的所有數字中最小的一個。

示例 1:
輸入: [10,2]
輸出: “102”

示例 2:
輸入: [3,30,34,5,9]
輸出: “3033459”

提示:
0 < nums.length <= 100

說明:
輸出結果可能非常大，所以你需要返回一個字符串而不是整數
拼接起來的數字可能會有前導 0，最後結果不需要去掉前導 0

代碼

Python

class Solution:
    def minNumber(self, nums: List[int]) -> str:
        def sort_rule(s1,s2): # 新的排序規則
            if s1+s2 > s2+s1: return 1
            elif s1+s2 < s2+s1: return -1
            else: return 0
            
        strs = [str(num) for num in nums] # 1.轉字符串
        strs.sort(key=functools.cmp_to_key(sort_rule)) # 2.排序，注意制定新的排序規則
        return ''.join(strs) # 3.拼接字符串

C++

class Solution {
    // 新的排序規則
    static bool cmp(const string& s1, const string& s2) {
        return s1+s2<s2+s1;
    }
    
public:
    string minNumber(vector<int>& nums) {
        vector<string> strs;
        string res;
        // 先轉字符串
        for (int i=0; i<nums.size(); i++) {
            strs.push_back(to_string(nums[i]));
        }
        // 排序
        sort(strs.begin(), strs.end(), cmp);
        // 拼接字符串
        for (int j=0; j<strs.size(); j++) {
            res += strs[j];
        }
        return res;
    }
};