面试题：TypeScript 剩余参数在函数重载中的应用

Question

假设有一个函数，它可以接受不同类型和数量的参数。当传入一个字符串参数时，返回这个字符串的长度；当传入多个数字参数时，返回这些数字的平均值。请使用 TypeScript 的函数重载和剩余参数来实现这个功能，并确保类型安全。

Answer 1

function calculate(...args: [string]): number;
function calculate(...args: number[]): number;
function calculate(...args: (string | number)[]): number {
    if (typeof args[0] ==='string') {
        return args[0].length;
    } else {
        return args.reduce((acc, num) => acc + num, 0) / args.length;
    }
}