蓝桥杯 java 算法提高 日期计算

算法提高 日期计算

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　已知2011年11月11日是星期五，问YYYY年MM月DD日是星期几？注意考虑闰年的情况。尤其是逢百年不闰，逢400年闰的情况。
输入格式
　　输入只有一行
　　YYYY MM DD
输出格式
　　输出只有一行
　　W
数据规模和约定
　　1599 <= YYYY <= 2999
　　1 <= MM <= 12
　　1 <= DD <= 31，且确保测试样例中YYYY年MM月DD日是一个合理日期
　　1 <= W <= 7，分别代表周一到周日

import java.util.Scanner;
 
public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int y = sc.nextInt();
		int m = sc.nextInt();
		int d = sc.nextInt();
		int mm[] = { 0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
		int sum = 0;
		/** 2018年1月1日为周1 */
		if (y >= 2018) {
			for (int i = 2018; i < y; i++) {
				if (run(i))
					sum += 366;
				else
					sum += 365;
			}
			for (int i = 0; i < m; i++) {
				sum += mm[i];
			}
			sum += d;
			if (m > 2 && run(y))
				sum++;
			if (sum % 7 == 0)
				System.out.println(7);
			else
				System.out.println(sum % 7);
		}
		else {
			for (int i = y + 1; i < 2018; i++) {
				if (run(i))
					sum += 366;
				else
					sum += 365;
			}
			for (int i = m + 1; i <= 12; i++) {
				sum += mm[i];
			}
			sum += mm[m] - d;
			if (m <= 2 && run(y))
				sum++;
			if (sum % 7 == 0)
				System.out.println(7);
			else
				System.out.println(7 - sum % 7);
		}
	}
	private static boolean run(int i) {
		if ((i % 4 == 0 && i % 100 != 0) || (i % 400 == 0))
			return true;
		else
			return false;
	}
}