PAT 1078 字符串压缩与解压（20）（代码+思路）

1078 字符串压缩与解压（20 分）

文本压缩有很多种方法，这里我们只考虑最简单的一种：把由相同字符组成的一个连续的片段用这个字符和片段中含有这个字符的个数来表示。例如 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

PS：该题要实现两个功能：（争取只遍历一遍字符串）

1、压缩字符：我这里从左到右遍历，比较当前字符跟下一个字符，若相等，计数器+1，直到不等时，输出（当前字符*计数器），计数器置0；

2、解压缩字符：这里依旧是从左到右变量，如果遇到数字，输出（数字*后面的字符），别忘记假指针（int i）要往后移，如果遇到字符，直接输出字符。

注意：解压缩字符的时候遇到的数字不一定只有一位。

另：因为只遍历一遍字符串，我们可以直接在输入的时候就对它进行操作，这样就不需要浪费额外的空间来进行存储了，这么艰巨的任务，就交给你了，哈哈。

提示：

int main() {
	char ch;
	while (1) {
		ch = getchar();
		if (ch == '\n') break;
		/*在这里对ch进行操作*/
	}
	return 0;
}

#include<iostream>
#include<string>
using namespace std;
string compress(string &s) {
	string cs;		//压缩字符串
	int count = 1;
	for (int i = 0; i < s.length(); i++) {
		if (s[i] == s[i + 1])	//与后一个进行比较
			count++;
		else {
			if (count > 1)		//如果该字符长度超过1，要在前加数字
				cs += to_string(count);
			cs += s[i];
			count = 1;
		}
	}
	return cs;
}
string decompress(string s) {
	string des;		//解压字符串
	string n;		//存放数字
	for (int i = 0; i < s.length(); i++) {
		while (isdigit(s[i])) {		//如果是数字就往后找，直到找到非数字（字母，空格）时输出
			n += s[i];
			i++;
			if (!isdigit(s[i]))
				for (int j = 0; j < atoi(n.c_str()) - 1; j++)
					des += s[i];
		}
		n.clear();
		des += s[i];		//如果是一个字符，直接输出
	}
	return des;
}
int main() {
	char ch;
	string str;
	cin >> ch;
	getchar();
	getline(cin, str);
	switch (ch) {
	case 'C':	cout << compress(str) << endl; break;
	case 'D':	cout << decompress(str) << endl; break;
	}
	return 0;
}