题目:
You are given several logs that each log contains a unique id and timestamp. Timestamp is a string that has the following format: Year:Month:Day:Hour:Minute:Second, for example, 2017:01:01:23:59:59. All domains are zero-padded decimal numbers.
Design a log storage system to implement the following functions:
void Put(int id, string timestamp): Given a log's unique id and timestamp, store the log in your storage system.
int[] Retrieve(String start, String end, String granularity): Return the id of logs whose timestamps are within the range from start to end. Start and end all have the same format as timestamp. However, granularity means the time level for consideration. For example, start = "2017:01:01:23:59:59", end = "2017:01:02:23:59:59", granularity = "Day", it means that we need to find the logs within the range from Jan. 1st 2017 to Jan. 2nd 2017.
Example 1:
put(1, "2017:01:01:23:59:59");
put(2, "2017:01:01:22:59:59");
put(3, "2016:01:01:00:00:00");
retrieve("2016:01:01:01:01:01","2017:01:01:23:00:00","Year"); // return [1,2,3], because you need to return all logs within 2016 and 2017.
retrieve("2016:01:01:01:01:01","2017:01:01:23:00:00","Hour"); // return [1,2], because you need to return all logs start from 2016:01:01:01 to 2017:01:01:23, where log 3 is left outside the range.
Note:
There will be at most 300 operations of Put or Retrieve.
Year ranges from [2000,2017]. Hour ranges from [00,23].
Output for Retrieve has no order required.
题解:
Inorder to retireve range of timestamp. It needs a linear data structure to store log based on timestamp. It could be list or array, but every time, it needs to retrieve everything again.
Or we could use TreeMap. Thus timestamp is sorted.
When we have the start and end time, revise a little bit from the granularity index.
Use TreeMap subMap to get sub map range from [start, end] timestamp.
And add values to res.
Time Complexity: put, O(logn). n = current count of log. retrieve(logn + m). m = count of logs between start and end.
Space: O(n).
AC Java:
1 classLogSystem {2 private TreeMap>treeMap;3 private HashMapmap;4 private String min = "2000:01:01:00:00:00";5 private String max = "2017:12:31:23:59:59";6
7 publicLogSystem() {8 this.treeMap = new TreeMap<>();9 this.map = new HashMap<>();10 map.put("Year", 4);11 map.put("Month", 7);12 map.put("Day", 10);13 map.put("Hour", 13);14 map.put("Minute", 16);15 map.put("Second", 19);16 }17
18 public void put(intid, String timestamp) {19 treeMap.putIfAbsent(timestamp, new HashSet<>());20 treeMap.get(timestamp).add(id);21 }22
23 public Listretrieve(String s, String e, String gra) {24 int index =map.get(gra);25 String start = s.substring(0, index) +min.substring(index);26 String end = e.substring(0, index) +max.substring(index);27
28 Map> subMap = treeMap.subMap(start, true, end, true);29 List res = new ArrayList<>();30 for(Map.Entry>entry : subMap.entrySet()){31 res.addAll(entry.getValue());32 }33
34 returnres;35 }36 }37
38 /**
39 * Your LogSystem object will be instantiated and called as such:40 * LogSystem obj = new LogSystem();41 * obj.put(id,timestamp);42 * List param_2 = obj.retrieve(s,e,gra);43 */
用list直接存log. 根据granularity取出timestamp的substring直接比较.
Time Complexity: put, O(1). retrieve, O(logs.size()).
Space: O(logs.size()).
AC Java:
1 classLogSystem {2 Listlogs;3 List units = Arrays.asList("Year", "Month", "Day", "Hour", "Minute", "Second");4 int [] indices = new int[]{4,7,10,13,16,19};5
6 publicLogSystem() {7 logs = new LinkedList();8 }9
10 public void put(intid, String timestamp) {11 logs.add(newString[]{Integer.toString(id), timestamp});12 }13
14 public Listretrieve(String s, String e, String gra) {15 List res = new ArrayList();16 int ind =indices[units.indexOf(gra)];17
18 String sSub = s.substring(0, ind);19 String eSub = e.substring(0, ind);20
21 for(String [] log : logs){22 String sub = log[1].substring(0, ind);23 if(sub.compareTo(sSub)>=0 && sub.compareTo(eSub)<=0){24 res.add(Integer.valueOf(log[0]));25 }26 }27 returnres;28 }29 }30
31 /**
32 * Your LogSystem object will be instantiated and called as such:33 * LogSystem obj = new LogSystem();34 * obj.put(id,timestamp);35 * List param_2 = obj.retrieve(s,e,gra);36 */