猿辅导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();
}
}
京公网安备 11010502036488号