剑指offer 68-2. 二叉树的最近公共祖先

题目描述

输入一个非负整数数组，把数组里所有数字拼接起来排成一个数，打印能拼接出的所有数字中最小的一个。

示例 1:

输入: [10,2]
输出: "102"

示例 2:

输入: [3,30,34,5,9]
输出: "3033459"

提示:

0 < nums.length <= 100

说明:

输出结果可能非常大，所以你需要返回一个字符串而不是整数
拼接起来的数字可能会有前导 0，最后结果不需要去掉前导 0

题解

(排序,自定义比较器) O(nlogn)

这道题目的巧妙之处是可以通过自定义排序方式将数组排序得到结果。

自定义排序方式：数字类型的 a < b 当前仅当字符串类型的 a + b < b + a，即 ab 排列组成的数字小于 ba 排列组成的数字 ab < ba，排序之后依次遍历数组的结果就是最小的数。

时间复杂度

时间复杂度为排序所需的时间 O(nlogn)

空间复杂度

O(1)

C++ 代码

class Solution {
public:
    static bool cmp(int& a, int& b) {
        string as = to_string(a), bs = to_string(b);
        return as + bs < bs + as;
    }
    
    string minNumber(vector<int>& nums) {
        sort(nums.begin(), nums.end(), cmp);
        string res;
        for (auto x : nums) res += to_string(x);
        return res;
    }
};

Java 代码

class Solution {
    public String minNumber(int[] nums) {
        String[] strNums = new String[nums.length];
        for (int i = 0; i < nums.length; i ++ ) {
            strNums[i] = String.valueOf(nums[i]);
        }
        
        Arrays.sort(strNums, (a, b) -> (a + b).compareTo(b + a));
        
        StringBuilder res = new StringBuilder();
        for (String str : strNums) {
            res.append(str);
        }
        
        return res.toString();
    }
}

Python 代码

class Solution:
    def minNumber(self, nums: List[int]) -> str:
        key = cmp_to_key(lambda x, y: int(x + y) - int(y + x))
        nums = sorted(map(str, nums), key=key)
        return "".join(nums)

本文由读者提供，Github地址：https://github.com/tonngw

点击面试手册，获取本站面试手册PDF完整版

上一篇：剑指offer 68-1. 二叉搜索树的最近公共祖先