每日OJ_牛客_美国节日（日期模拟）

_GR2024-08-23 23:17

牛客_美国节日（日期模拟）

解析代码

题目表述很明白，难点在于要求一个月第N个星期W。那么面对这个问题，拆解的思路是，首先，要想找到一个月第N个星期W，一定需要一个参照物，最好的目标当然是这个月的第一天。拿到参照物后，要能得到参照物的星期数，然后就能得到结果了。所以这个题有两个难点：

判断某个月的1号到底是周几，
然后根据这星期数得到这个月第N个星期W。

这两个功能写成函数，即可通过反复调用拿到结果。

cpp 复制代码

#include <iostream>
#include <cstdio>
using namespace std;
bool isLeapYear(int y)
{
    return y % 400 == 0 || (y % 100 != 0 && y % 4 == 0);
}
int DAYS[] = {31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
// 求解 y,m,d 这一年过了多少天
int nDays(int y, int m, int d)
{
    int n = d;
    // m = 10
    // [1, 9]
    // [0, 8]
    for (int i = 0; i < m - 1; i++)
    {
        n += DAYS[i];
    }
    if (m > 2 && isLeapYear(y))
        n++;
    return n;
}
// 传入 y,m,d 计算从基准日期，到带计算日期过了多少天。
// 算出这个天数的 MOD 7 的同余数
int diff(int y, int m, int d)
{
    return (y - 1) + (y - 1) / 4 - (y - 1) / 100 + (y - 1) / 400 + nDays(y, m, d);
}
// 传入 y,m,d，计算周几
int week(int y, int m, int d)
{
    int w = diff(y, m, d) % 7;
    if (w == 0)
        w = 7;
    return w;
}
// 已知当月1日是星期 w，计算第 n 个星期 e 是几号
int m1(int w, int n, int e)
{
    return 1 + (n - 1) * 7 + (7 - w + e) % 7;
}
// 已知6月1日是星期w，计算5月的最后一个星期一
int m2(int w)
{
    int d = (w == 1 ? 7 : w - 1);
    return 32 - d;
}
int main()
{
    int y = 0;
    while (cin >> y)
    {
        printf("%d-01-01\n", y);
        int w = week(y, 1, 1);
        printf("%d-01-%02d\n", y, m1(w, 3, 1));
        w = week(y, 2, 1);
        printf("%d-02-%02d\n", y, m1(w, 3, 1));
        w = week(y, 6, 1);
        printf("%d-05-%02d\n", y, m2(w));
        printf("%d-07-04\n", y);
        w = week(y, 9, 1);
        printf("%d-09-%02d\n", y, m1(w, 1, 1));
        w = week(y, 11, 1);
        printf("%d-11-%02d\n", y, m1(w, 4, 4));
        printf("%d-12-25\n\n", y);
    }
}