Design Compressed String Iterator

Design and implement a data structure for a compressed string iterator. It should support the following operations: next and hasNext.

The given compressed string will be in the form of each letter followed by a positive integer representing the number of this letter existing in the original uncompressed string.

next() - if the original string still has uncompressed characters, return the next letter; Otherwise return a white space.
hasNext() - Judge whether there is any letter needs to be uncompressed.

Note:
Please remember to RESET your class variables declared in StringIterator, as static/class variables are persisted across multiple test cases. Please see here for more details.

Example:

StringIterator iterator = new StringIterator("L1e2t1C1o1d1e1");

iterator.next(); // return 'L'
iterator.next(); // return 'e'
iterator.next(); // return 'e'
iterator.next(); // return 't'
iterator.next(); // return 'C'
iterator.next(); // return 'o'
iterator.next(); // return 'd'
iterator.hasNext(); // return true
iterator.next(); // return 'e'
iterator.hasNext(); // return false
iterator.next(); // return ' '
 1 public class StringIterator {
 2     int i;
 3     String[] arr;
 4     int[] counts;
 5 
 6     public StringIterator(String str) {
 7         arr = str.split("\\d+");
 8         counts = Arrays.stream(str.substring(1).split("[a-zA-Z]+")).mapToInt(Integer::parseInt).toArray();
 9     }
10 
11     public char next() {
12         if (!hasNext()) {
13             return ' ';
14         }
15         char ch = arr[i].charAt(0);
16         counts[i]--;
17 
18         if (counts[i] == 0) {
19             ++i;
20         }
21         return ch;
22     }
23 
24     public boolean hasNext() {
25         if (i == arr.length) {
26             return false;
27         }
28         return true;
29     }
30 }

 

posted @ 2019-08-12 05:14  北叶青藤  阅读(175)  评论(0编辑  收藏  举报