C++学习（九）：日期类的简单实现

一：头文件

#include<iostream>
using namespace std;
class Date {
	friend ostream& operator<<(ostream& out, const Date& d);
	friend istream& operator>>(istream& in, const Date& d);
public:
	int GetMonthDay(int year, int month) const;
	Date(int year = 1, int month = 1, int day = 1);
	void Print() const;
	bool operator<(const Date& d) const;
	bool operator==(const Date& d) const;
	bool operator<=(const Date& d) const;
	bool operator>(const Date& d) const;
	bool operator>=(const Date& d) const;
	bool operator!=(const Date& d) const;

	Date& operator+=(int day);
	Date operator+(int day) const;
	Date& operator-=(int day);
	Date operator-(int day) const;
	Date& operator++();
	// d1++ -> d1.operator++(0)
	// 加一个int参数，进行占位，跟前置++构成函数重载进行区分
	// 本质后置++调用，编译器进行特殊处理
	Date operator++(int);
	Date& operator--();
	Date operator--(int);

	int operator-(const Date& d) const;
private:
	int _year;
	int _month;
	int _day;

};
ostream& operator<<(ostream& out, const Date& d);
istream& operator>>(istream& in, Date& d);

二：源文件

#include"Date.h"
int Date::GetMonthDay(int year, int month) const
{
	const static int monthArray[13] = { 0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };

	if (month == 2
		&& ((year % 4 == 0 && year % 100 != 0) || (year % 400 == 0)))
	{
		return 29;
	}

	return monthArray[month];
}
Date::Date(int year, int month, int day) {

	_year = year;
	_month = month;
	_day = day;
	if (month < 1 || month>13 || day<1 || day>GetMonthDay(year, month)) {
		cout << "非法日期";
		exit(-1);

	}
}
void Date::Print() const
{
	cout << _year << "年" << _month << "月" << _day << "日" << endl;
}
bool Date::operator<(const Date& d) const
{
	if (_year < d._year)
	{
		return true;
	}
	else if (_year == d._year && _month < d._month)
	{
		return true;
	}
	else if (_year == d._year && _month == d._month && _day < d._day)
	{
		return true;
	}
	else
	{
		return false;
	}
}
bool Date::operator==(const Date& d) const
{
	return _year == d._year
		&& _month == d._month
		&& _day == d._day;
}
bool Date::operator<=(const Date& d) const
{
	return *this < d || *this == d;

}
bool Date:: operator>(const Date& d) const
{
	return !(*this < d);
}
bool Date::operator>=(const Date& d) const
{
	return !(*this < d);
}
bool Date::operator!=(const Date& d) const
{
	return !(*this == d);
}
Date& Date::operator+=(int day) {
	if (day < 0) return *this -= (-day);
	_day += day;
	while (_day > GetMonthDay(_year, _month)) {
		_day -= GetMonthDay(_year, _month);
		_month++;
		if (_month > 12) { _year++; _month = 1; }
	}
	return *this;
}
Date Date:: operator+(int day) const {
	Date tmp(*this);
	tmp += day;
	return tmp;
}
Date& Date::operator-=(int day)
{
	if (day < 0)
	{
		return *this += (-day);
	}

	_day -= day;

	while (_day <= 0)
	{
		--_month;
		if (_month == 0)
		{
			--_year;
			_month = 12;
		}

		_day += GetMonthDay(_year, _month);
	}

	return *this;
}

Date Date::operator-(int day) const
{
	Date tmp(*this);

	tmp -= day;

	return tmp;
}
Date& Date::operator++()
{
	*this += 1;
	return *this;
}

Date Date::operator++(int)
{
	Date tmp(*this);

	*this += 1;

	return tmp;
}


Date& Date::operator--()
{
	*this -= 1;
	return *this;
}

Date Date::operator--(int)
{
	Date tmp(*this);

	*this -= 1;

	return tmp;
}
int Date:: operator-(const Date& d) const {
	Date max = *this;
	Date min = d;
	int flag = 1;
	if (max < min) {
		min = *this;
		max = d;
		flag = -1;
	}
	int n = 0;
	while (min != max) {
		 min++;
		n++;
	}
	return n * flag;
}
ostream& operator<<(ostream& out, const Date& d) {
	out << d._year << " " << d._month << " " << d._day;
	return out;
}
istream& operator>>(istream& in, const Date& d) {
	in >> d._year >> d._month >> d._day;
	return in;
}

然后大家再写个测试源文件测测就行了

感谢你看到这大家共同进步