牛客网 KY222 打印日期

利用一个数组表示月份及对应的天数，初始每个元素值均为0，按元素下标递增顺序将所给天数分配到各个月份对应的数组位置，最后输出最后一个不为0的数组下标及元素即得到月份及天数。

#include <iostream>
#include <iomanip>
#include <set>
using namespace std;

bool isLeapYear(int year) {			//判断是否为闰年
	if (year % 400 == 0 || year % 4 == 0 && year % 100 != 0) return true;
	else return false;
}

void calculate(int a[15], int& b, int c, int& d) {		//依次传入结果数组，月份，						
	if ((d - c) > 0) {									//当月天数，总天数
		a[b] = c;
		b++;
		d -= c;
	}
	else {
		a[b] = d;
		d = 0;
	}
}

int main()
{	
	int year, day;
	set<int> bigMonth = { 1,3,5,7,8,10,12 };		//31天的月份集合

	while (scanf("%d %d", &year, &day) != EOF) {
		int month = 1;
		int statistical[15] = { 0 };

		while (day > 0) {
			if (bigMonth.find(month) != bigMonth.end()) {		//判断是否为31天的月份
				calculate(statistical, month, 31, day);
			}
			else if (month == 2) {
				if (isLeapYear(year)) {
					calculate(statistical, month, 29, day);
				}
				else {
					calculate(statistical, month, 28, day);
				}
			}
			else {
				calculate(statistical, month, 30, day);
			}
		}
		cout << year << "-" << right << setw(2) << setfill('0') << month << "-" << right << setw(2) << setfill('0') << statistical[month] << endl;
	}
	

}