面试题：TypeScript函数重载中的类型推断

Question

有一个函数`printValue`，希望实现它的重载，第一个重载版本接受一个数字参数并打印它的平方，第二个重载版本接受一个字符串参数并打印它的长度。同时要求在调用该函数时，TypeScript能根据传入参数正确推断出返回值类型。请写出完整代码。

Answer 1

function printValue(num: number): void;
function printValue(str: string): void;
function printValue(arg: number | string): void {
    if (typeof arg === 'number') {
        console.log(arg * arg);
    } else {
        console.log(arg.length);
    }
}