校验选择的时间是否为连续3个月

写一个函数判断是否是连续的3个月 我传入[ "2022-01", "2022-03" ] 这个是正常的连续3个月 则返回true

如果判断不是连续的3个月 则 取数组里面的第一个数据 往后面加2个月 返回

举例 传入[ "2022-05-01", "2022-06-01" ] 函数返回[ "2022-05-01", "2022-07-01"]

function checkAndAdjustDates(dates) {
    if (dates.length !== 2) {
        return "输入应包含两个日期";
    }

    // 从年月字符串中提取年和月
    const startYear = parseInt(dates[0].split("-")[0]);
    const startMonth = parseInt(dates[0].split("-")[1]);
    const endYear = parseInt(dates[1].split("-")[0]);
    const endMonth = parseInt(dates[1].split("-")[1]);

    // 计算日期差值（月份之差）
    const diff = (endYear - startYear) * 12 + (endMonth - startMonth);

    if (diff === 2) {
        // 如果是连续的3个月，则返回true
        return true;
    } else {
        // 如果不是连续的3个月，调整日期范围
        const newEndYear = startYear + Math.floor((startMonth + 2) / 12);
        const newEndMonth = (startMonth + 2) % 12 || 12;

        // 格式化新的结束日期字符串
        const adjustedDates = [`${startYear}-${startMonth.toString().padStart(2, '0')}`, `${newEndYear}-${newEndMonth.toString().padStart(2, '0')}`];

        return adjustedDates;
    }
}

// 示例用法
const inputDates = ["2022-01", "2022-03"];
const result = checkAndAdjustDates(inputDates);

console.log(result); // 输出 true