面试题：TypeScript中函数类型重载的基本实现

Question

请编写一个TypeScript函数，实现根据传入参数类型不同返回不同结果的函数类型重载。函数接收一个参数，当参数为数字时，返回该数字的平方；当参数为字符串时，返回该字符串的长度。

Answer 1

function calculate(input: number): number;
function calculate(input: string): number;
function calculate(input: number | string): number {
    if (typeof input === 'number') {
        return input * input;
    } else {
        return input.length;
    }
}