相关月问题

来源

北京大学医学部计算概论08期末考试题

描述

“相关月”是指那些在一年中月份的第一天星期数相同的月份。例如，九月和十二月是相关的，因为九月一日和十二月一日的星期数总是相同的。两个月份相关，当且仅当两个月份第一天相差的天数能被7整除，也就是说，这两天相差为几个整星期。又如，二月和三月一般都是相关月，因为二月有28天，能被7整除，也恰好为4个星期。而在闰年，一月和二月的相关月与它们在平年的相关月是不同的，因为二月有29天，其后每个月份的第一天星期数都推后了一天。

关于输入

输入的第一行为整数n（n≤200），
其后n行，每行三个整数，依次为一个年份和两个月份，整数之间用一个空格分隔。

关于输出

输出有n行，对应于每个输入的年份和相应两个月份，
如果这两个月份是相关的，则输出YES；
否则，输出NO。

例子输入

5
1994 10 9
1935 12 1
1957 1 9
1917 9 12
1948 1 4

例子输出

NO
NO
NO
YES
YES

#include <stdio.h>
int leapyear(int a) {
	if (a % 4 != 0 || a % 100 == 0 && a % 400 != 0)return 0;
	else return 1;
}
// culculate leap year(if the year is not leap year,return 0;else return 1;
int month[2][12] = { 31,28,31,30,31,30,31,31,30,31,30,31,
					31,29,31,30,31,30,31,31,30,31,30,31 };
//calculate leap February or not
int main() {
	int n = 0;//integer n;
	scanf("%d", &n);
	int j = 0;//to count month
	int year_num, month_1, month_2 = 0;
	int i = 0;
	int temp = 0;
	for (i = 0; i < n; i++) {
		scanf("%d %d %d", &year_num, &month_1, &month_2);
		if (month_1 > month_2) {
			temp = month_1;
			month_1 = month_2;
			month_2 = temp;
		}// swap two months to make a perfect loop
		int day_sum = 0;
		for (j = month_1 - 1; j < month_2 - 1; j++) {
			day_sum += month[leapyear(year_num)][j];
		}
		if (day_sum % 7 == 0)printf("YES\n");
		else printf("NO\n");
	}

	return 0;
}