[蓝桥杯]日期问题 JAVA简洁代码

题目描述

小明正在整理一批历史文献。这些历史文献中出现了很多日期。小明知道这些日期都在1960年1月1日至2059年12月31日。令小明头疼的是，这些日期采用的格式非常不统一，有采用年/月/日的，有采用月/日/年的，还有采用日/月/年的。更加麻烦的是，年份也都省略了前两位，使得文献上的一个日期，存在很多可能的日期与其对应。

比如02/03/04，可能是2002年03月04日、2004年02月03日或2004年03月02日。

给出一个文献上的日期，你能帮助小明判断有哪些可能的日期对其对应吗？

输入

一个日期，格式是"AA/BB/CC"。 (0 <= A, B, C <= 9)

输出

输出若干个不相同的日期，每个日期一行，格式是"yyyy-MM-dd"。多个日期按从早到晚排列。

样例输入

02/03/04
           

样例输出

2002-03-04
2004-02-03
2004-03-02
           

后台的数据有点坑，题目说明给出的数字都大于0了，但是如果不加上  day > 0的判断的话还是会出错。

import java.util.*;
public class Main {
	public static TreeSet<Integer> set = new TreeSet<Integer>(); 
	public static void IsLegal(String year,String month,String day) {
		int y = Integer.parseInt(year);
		int m = Integer.parseInt(month);
		int d = Integer.parseInt(day);
		if(y <= 59) y += 2000;
		else y += 1900;
		if( ((y % 4 == 0 && y % 100 != 0) || y % 400 == 0) && m == 2 && d > 0 && d <= 29  ){  //闰年的二月是否合法
			set.add(Integer.parseInt(""+y+month+day));
		}
		if(y % 4 != 0 && m == 2 && d > 0 && d <= 28) {                              //平年的二月是否合法
			set.add(Integer.parseInt(""+y+month+day));
		}
		if((m==1||m==3||m==5||m==7||m==8||m==10||m==12) && d>0 && d<=31)  {         //大月是否合法
                        set.add(Integer.parseInt(""+y+month+day));  
                }  
                if((m==4|| m==6|| m==9|| m==11) && d>0&&d<=30)  {                           //小月是否合法
                        set.add(Integer.parseInt(""+y+month+day));  
                }  
	}
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		String str = sc.next();
		String[] s = str.split("/");
		IsLegal(s[0],s[1],s[2]);
		IsLegal(s[2],s[1],s[0]);
		IsLegal(s[2],s[0],s[1]);
		for(Integer tmp:set) {
			String ans = ""+tmp;
			System.out.println(ans.substring(0, 4)+"-"+ans.substring(4,6)+"-"+ans.substring(6,8));  
		}
	}
}