enhance ThirdLatestDate

Author: melmedany

src/main/java/Date.java

@@ -4,9 +4,9 @@
 
 public class Date {
 
-	int year;
-	int month;
-	int day;
+	Integer year;
+	Integer month;
+	Integer day;
 
 	public Date(String date) {
 		String[] d = date.split("-");

src/main/java/ThirdLatestDate.java

@@ -1,9 +1,5 @@
-import java.text.ParseException;
-import java.text.SimpleDateFormat;
 import java.util.Comparator;
-import java.util.Map;
 import java.util.Set;
-import java.util.TreeMap;
 import java.util.TreeSet;
 
 /**
@@ -18,55 +14,27 @@
 public class ThirdLatestDate {
 
 	public Date solve(Date[] dates) {
-
-		Map<Integer, Set<Date>> sortedDates = new TreeMap<Integer, Set<Date>>(new Comparator<Integer>() {
+		Set<Date> sortedDates = new TreeSet<Date>(new Comparator<Date>() {
 			@Override
-			public int compare(Integer i, Integer j) {
-				return j.compareTo(i);
+			public int compare(Date i, Date j) {
+				return i.year.compareTo(j.year);
 			}
-
-		});
-
-		for (Date date : dates) {
-			if (sortedDates.containsKey(date.year)) {
-				sortedDates.get(date.year).add(date);
-			} else {
-				Set<Date> yearDates = new TreeSet<Date>(new Comparator<Date>() {
-
-					SimpleDateFormat f = new SimpleDateFormat("dd-MM-yyyy");
-
-					public int compare(Date d1, Date d2) {
-
-						java.util.Date date1 = null, date2 = null;
-						try {
-							date1 = f.parse(d1.day + "-" + d1.month + "-" + d1.year);
-							date2 = f.parse(d2.day + "-" + d2.month + "-" + d2.year);
-						} catch (ParseException e) {
-						}
-						return date2.compareTo(date1);
-					}
-				});
-
-				yearDates.add(date);
-				sortedDates.put(date.year, yearDates);
+		}.thenComparing(new Comparator<Date>() {
+			@Override
+			public int compare(Date i, Date j) {
+				return i.month.compareTo(j.month);
 			}
-		}
-
-		Date thirdLatesDate = null;
-		int i = 0;
-		for (Set<Date> sd : sortedDates.values()) {
-			for (Date d : sd) {
-				thirdLatesDate = d;
-				if (i == 2) {
-					return d;
-				} else {
-					i++;
-					continue;
-				}
+		}.thenComparing(new Comparator<Date>() {
+			@Override
+			public int compare(Date i, Date j) {
+				return i.day.compareTo(j.day);
 			}
-		}
+		})));
 
-		return thirdLatesDate;
+		for (Date date : dates) {
+			sortedDates.add(date);
+		}
 
+		return (Date) sortedDates.toArray()[sortedDates.size() - 3];
 	}
 }