class LogSystem {

private static final DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy:MM:dd:HH:mm:ss");

private static final int K = 6;

private TreeMap<String, Integer> treeMap;

public LogSystem() {

Comparator<String> comp = (s1, s2) -> s1.compareTo(s2);

treeMap = new TreeMap<>(comp);

}

public void put(int id, String timestamp) {

treeMap.put(timestamp, id);

}

public List<Integer> retrieve(String s, String e, String gra) {

String start = roundLeft(s, gra);

String end = roundRight(e, gra);

Map<String, Integer> subMap = treeMap.subMap(start, end);

List<Integer> res = new ArrayList<>();

for (Integer id: subMap.values()) {

res.add(id);

}

return res;

}

private String roundLeft(String timestamp, String gra) {

TimeUnit unit = TimeUnit.valueOf(gra);

return roundLeft(timestamp, unit);

}

private String roundLeft(String timestamp, TimeUnit unit) {

String[] times = timestamp.split(":");

for (int k=K-1; k>unit.pos; k--) {

times[k] = "00";

}

return String.join(":", times);

}

private String plus(String timestamp, TimeUnit unit) {

LocalDateTime newTime = LocalDateTime.parse(timestamp, formatter).plus(1, ChronoUnit.valueOf(unit.chrono));

return formatter.format(newTime);

}

private String roundRight(String timestamp, String gra) {

TimeUnit unit = TimeUnit.valueOf(gra);

String newTs = plus(timestamp, unit);

return roundLeft(newTs, unit);

}

}

enum TimeUnit {

Year(0, "YEARS"),

Month(1, "MONTHS"),

Day(2, "DAYS"),

Hour(3, "HOURS"),

Minute(4, "MINUTES"),

Second(5, "SECONDS");

int pos;

String chrono;

TimeUnit(int position, String chrono) {

this.pos = position;

this.chrono = chrono;

}

}

/**

class LogSystem2 {

List<String[]> timestamps = new LinkedList<>();

List<String> units = Arrays.asList("Year", "Month", "Day", "Hour", "Minute", "Second");

int[] indices = new int[]{4,7,10,13,16,19};

public void put(int id, String timestamp) { timestamps.add(new String[]{Integer.toString(id), timestamp}); }

public List<Integer> retrieve(String s, String e, String gra) {

List<Integer> res = new LinkedList<>();

int idx = indices[units.indexOf(gra)];

for (String[] timestamp : timestamps) {

if (timestamp[1].substring(0, idx).compareTo(s.substring(0, idx)) >= 0 &&

timestamp[1].substring(0, idx).compareTo(e.substring(0, idx)) <= 0) res.add(Integer.parseInt(timestamp[0]));

}

return res;

}

}