王道机试指南NO.3日期差值

题目要求

时间限制：1s 内存限制：32MB

题目描述

有两个日期，求两个日期之间的天数，如果两个日期是连续的我们规定他们之间的天数为两天。

输入

有多组数据，每组数据有两行，分别表示两个日期，形式为YYYYMMDD

输出

每组数据输出一行，即日期差值

样例输入

20110412
20110422

样例输出

11

题目分析

1、求两个日期间的天数差，即求分别以两个特定的日期为界的日期区间的长度。解决这类问题有一个传统的思想——把原区间问题统一到起点确定 的区间问题上。该题上，不妨把问题统一到特定日期与一个原点日期（如0000年1月1日）的天数差，当要求两个特定的日期之间的天数差时，只要将他们与原点日期的天数差相减，既得两个特定日期之间的天数差（必要时要加绝对值）。
2、预处理（用空间换时间）：预处理出所有日期之间的天数差并保存起来，当数据真正输入时，只需用O(1)的时间复杂度将保存的数据读出，稍加处理就得到答案。保存预处理所得数据所需的内存来换取实时处理所需要的时间消耗。
3、日期类问题一定要注意闰年的问题
闰年判断规则：当年数不能被100整除时，若其能被4整除则为闰年，或者其能被400整除时也是闰年。逻辑表达：Year%100 != 0 && Year%4 == 0 || x%400 == 0

代码实现

#include <iostream>
#define ISYEAR(x) x%100 != 0 && x%4 == 0 || x%400 == 0 ? 1 : 0
//定义宏判断是否使闰年，方便计算每月天数

int dayOfMonth[13][2] = {
    0,0,
    31,31,
    28,29,
    31,31,
    30,30,
    31,31,
    30,30,
    31,31,
    31,31,
    30,30,
    31,31,
    30,30,
    31,31
}; //预存每月的天数，注意二月配合宏定义作特殊处理

using namespace std;

class Date{
public:
    int Day;
    int Month;
    int Year;

    void nextDay(){ //计算下一天的日期
        Day++;
        if(Day > dayOfMonth[Month][ISYEAR(Year)]){ //若日期超过了当月最大天数
            Day = 1;
            Month++;
            if(Month > 12){ //月数超过12
                Month = 1;
                Year++;
            }
        }
    }
};

int buf[5001][13][32]; //保存预处理的天数

int Abs(int x){ //求绝对值
    return x < 0 ? -x : x;
}

int main(){
    Date tmp;
    int cnt = 0;
    tmp.Day = 1;
    tmp.Month = 1;
    tmp.Year = 0; //初始化日期类对象0年1月1日

    //预处理
    while(tmp.Year != 5001){ //日期不超过5000年
        buf[tmp.Year][tmp.Month][tmp.Day] = cnt; //将该日与0年1月1日的天数差保存起来
        tmp.nextDay(); //计算下一天的日期
        cnt++;
    }

    int d1, m1, y1;
    int d2, m2, y2;
    int a, b; //接收输入的一串数字
    while(cin >> a){
        y1 = a / 10000;
        m1 = (a / 100) % 100;
        d1 = a % 100;

        cin >> b;
        y2 = b / 10000;
        m2 = (b / 100) % 100;
        d2 = b % 100;

        cout << Abs(buf[y2][m2][d2] - buf[y1][m1][d1]) + 1; //+1是题目要求
    }
    return 0;
}

总结

1、定义一个类，不仅可以用来表示日期，还可以自动计算下一个日期。一方面可以不断计算当前日期的下一个日期，另一方面累加计数器，计算当前日期与原点日期的天数差，并将其保存在内存中（这就是预处理）。计算日期差只需将两个日期与原点的日期差相减，取绝对值，再+1（题目要求：如果两个日期是连续的我们规定他们之间的天数为两天）。
2、若问题为计算某日期是那年的第几天，只需要用该日期与原点日期的天数减去那年元旦与原点日期便可得知该日期是当年的第几天。
3、为判断是否为闰年，定义了一个宏。
4、使用三维数组保存某个特定日期与原点日期的天数差，年月日分别作为该数组的下标。（Hash的基本思想，但不会发生冲突，不需要花力气解决冲突）
5、若用C语言，可以在scanf中用%4d%2d%2d给年、月、日赋值。scanf("%4d%2d%2d",&y1,&m1,&d1);
%d之间插入数字来读取特定位数的数组。
6、buf[5001][13][32]相对比较耗费内存的数组定义为全局变量。若在main函数（或其他函数）中定义该数组，其函数所可以使用的栈空间将不足以提供如此庞大的内存，出现栈溢出，导致程序异常终止。凡是涉及此类需要开辟大量内存空间的情况，我们都必须在函数体外定义，即定义为全局变量；或者在函数中使用malloc等函数动态申请变量空间。

运行结果