【JAVA】 PAT 1078 字符串压缩与解压

题目链接
文本压缩有很多种方法，这里我们只考虑最简单的一种：把由相同字符组成的一个连续的片段用这个字符和片段中含有这个字符的个数来表示。例如 ccccc 就用 5c 来表示。如果字符没有重复，就原样输出。例如 aba 压缩后仍然是 aba。

解压方法就是反过来，把形如 5c 这样的表示恢复为 ccccc。

本题需要你根据压缩或解压的要求，对给定字符串进行处理。这里我们简单地假设原始字符串是完全由英文字母和空格组成的非空字符串。

输入格式：
输入第一行给出一个字符，如果是 C 就表示下面的字符串需要被压缩；如果是 D 就表示下面的字符串需要被解压。第二行给出需要被压缩或解压的不超过 1000 个字符的字符串，以回车结尾。题目保证字符重复个数在整型范围内，且输出文件不超过 1MB。

输出格式：
根据要求压缩或解压字符串，并在一行中输出结果。

输入样例 1：

C
TTTTThhiiiis isssss a   tesssst CAaaa as

输出样例 1：

5T2h4is i5s a3 te4st CA3a as

输入样例 2：

D
5T2h4is i5s a3 te4st CA3a as10Z

输出样例 2：

TTTTThhiiiis isssss a   tesssst CAaaa asZZZZZZZZZZ

根据题目要求编写两个方法D和C，在main方法里直接调用

char m = bf.readLine().charAt(0);
String s = bf.readLine();
if (m == 'C') {
	C(s, out);
} else {
	D(s, out);
}

压缩方法
首先把第0个字符提出来记为tem，然后从第1个字符开始遍历。如果当前字符与tem相等，则次数count++，同时把tem=当前字符

if (a.charAt(i) == tem) {
	count++;
	tem = a.charAt(i);
} 

如果不相等，首先判断次数是不是出现了一次以上，如果超过了一次，先把次数输出，然后再把tem输出，同时把count记为1。

if (count != 1) {
	out.print(count);
}
	out.print(tem);
	tem = a.charAt(i);
	count = 1;

当遍历到最后一个字符的时候，首先先做上述判断，然后再把count和tem输出

if (i == a.length() - 1) {
	if (count != 1) {
		out.print(count);
	}
	out.print(tem);
}

压缩方法C代码

static void C(String a, PrintWriter out) {
	char tem = a.charAt(0);
	int count = 1;
	for (int i = 1; i < a.length(); i++) {
		if (a.charAt(i) == tem) {
			count++;
			tem = a.charAt(i);
		} else {
			if (count != 1) {
				out.print(count);
			}
			out.print(tem);
			tem = a.charAt(i);
			count = 1;
		}
		if (i == a.length() - 1) {
			if (count != 1) {
				out.print(count);
			}
			out.print(tem);
		}
	}
}

解压方法
首先先把字符串根据非数字切割，split方法的参数为"\D"。注意切分后除了数字还会有空值，处理方式下面再说。再把整个字符串变为字符型数组。index为n的下标。

	String[] n = a.split("\\D");
	char[] s = a.toCharArray();
	int index = 0;

遍历数组s，判断当前字符是不是数字，是数字则继续下一个字符。
不是数字的话则进行如下步骤。
如果是第0个字符，直接输出该字符，并继续下一步循环。

if (i == 0) {
	out.print(s[i]);
	continue;
}

判断前一个字符是不是数字，如果是数字的话进行如下操作。
前文提到存放数字的字符串数组n存在空字符串，所以需要判断遍历n，如果等于空字符串，则把index++。
找到数字后记为t，循环t次输出当前字符
如果前一个字符不是数字，则直接输出该字符。

if (isNum(s[i - 1])) {
	while (index < n.length && n[index].equals("")) {
		index++;
	}
	int t = sti(n[index++]);
	for (int j = 0; j < t; j++) {
		out.print(s[i]);
	}
}else {
	out.print(s[i]);
}

解压方法D代码

static void D(String a, PrintWriter out) {
	String[] n = a.split("\\D");
	char[] s = a.toCharArray();
	int index = 0;
	for (int i = 0; i < s.length; i++) {
		if (!isNum(s[i])) {
			if (i == 0) {
				out.print(s[i]);
				continue;
			}
			if (isNum(s[i - 1])) {
				while (index < n.length && n[index].equals("")) {
					index++;
				}
				int t = sti(n[index++]);
				for (int j = 0; j < t; j++) {
					out.print(s[i]);
				}
			} else {
				out.print(s[i]);
			}
		}
	}
}

D和C方法本来想用一个字符串R来储存答案，然后这两个方法的返回值是String。可是提交上去最后一个测试点会超内存。所以把out传进来，直接输出

完整AC代码

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.util.Map;

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
		char m = bf.readLine().charAt(0);
		String s = bf.readLine();
		if (m == 'C') {
			C(s, out);
		} else {
			D(s, out);
		}
		out.flush();
	}

	static void C(String a, PrintWriter out) {
		char tem = a.charAt(0);
		int count = 1;
		for (int i = 1; i < a.length(); i++) {
			if (a.charAt(i) == tem) {
				count++;
				tem = a.charAt(i);
			} else {
				if (count != 1) {
					out.print(count);
				}
				out.print(tem);
				tem = a.charAt(i);
				count = 1;
			}
			if (i == a.length() - 1) {
				if (count != 1) {
					out.print(count);
				}
				out.print(tem);
			}
		}
	}

	static void D(String a, PrintWriter out) {
		String[] n = a.split("\\D");
		char[] s = a.toCharArray();
		int index = 0;
		for (int i = 0; i < s.length; i++) {
			if (!isNum(s[i])) {
				if (i == 0) {
					out.print(s[i]);
					continue;
				}
				if (isNum(s[i - 1])) {
					while (index < n.length && n[index].equals("")) {
						index++;
					}
					int t = sti(n[index++]);
					for (int j = 0; j < t; j++) {
						out.print(s[i]);
					}
				} else {
					out.print(s[i]);
				}
			}
		}
	}

	static int sti(String a) {
		return Integer.parseInt(a);
	}

	static boolean isNum(char a) {
		return a >= '0' && a <= '9';
	}

}