猿辅导APP需要下发一些宣传文本给学生,工程师们使用了一种字符压缩算法,为简单起见,假设被压缩的字符全部为大写字母序列,A,B,C,D....Z,压缩规则如下:
1.AAAB可以压缩为A3B (单字符压缩不加括号)
2.ABABA可以压缩为(AB)2A (多字符串压缩才加括号)
输入数据保证不会出现冗余括号,且表示重复的数字一定合法且大于1,即不会出现:
1.(A)2B ------- (应为:A2B)
2. ((AB))2C,-----(应为:(AB)2C )
3. (A)B ----- (应为:AB)
4. A1B,(AB)1C,(应为 AB,ABC)
注意:数字可能出现多位数即A11B或者(AB)10C或者A02这种情况。
A11B = AAAAAAAAAAAB
(AB)10C = ABABABABABABABABABABC
A02 = AA
数据分布:
对于60%的数据,括号不会出现嵌套,即不会有 ((AB)2C)2这种结构。
对于80%的数据,括号最多只嵌套一层,即不会有 (((AB)2C)2D)99 这种结构。
对于100%的数据,括号可以嵌套任意层。
解析:关键是对括号的处理,由于要先处理最里层的括号,再处理最外层的括号,符合栈的性质。因此用一个栈来保存字符串开始重复的索引。
import java.util.*; public class Main{ public static void main(String[] args) { Scanner input; int C, i; String[] strs; input = new Scanner(System.in); while (input.hasNext()) { C = input.nextInt(); strs = new String[C]; for(i = 0; i < C; i++){ strs[i] = input.next(); } for(i = 0; i < C; i++){ System.out.println(Solution(strs[i])); } } } private static String Solution(String str) { StringBuilder ans; Stack<Integer> stack; int i, j, l; char c; ans = new StringBuilder(); stack = new Stack<>(); l = str.length(); i = 0; while (i < l) { c = str.charAt(i); if (c == '(') { stack.push(ans.length()); i++; } else if (c >= 'A' && c <= 'Z') { ans.append(c); i++; } else if (c >= '0' && c <= '9') { j = i; while (i < l && str.charAt(i) >= '0' && str.charAt(i) <= '9') { i++; } ans.append(repeat(ans.substring(ans.length() - 1), Integer.parseInt(str.substring(j, i)) - 1)); } else if ( c == ')') { i++; j = i; while (i < l && str.charAt(i) >= '0' && str.charAt(i) <= '9') { i++; } ans.append(repeat(ans.substring(stack.pop()), Integer.parseInt(str.substring(j, i)) - 1)); } } return ans.toString(); } private static String repeat(String str, int n) { int i; StringBuilder ans; ans = new StringBuilder(); for (i = 0; i < n; i++) { ans.append(str); } return ans.toString(); } }