文本压缩有很多种方法,这里我们只考虑最简单的一种:把由相同字符组成的一个连续的片段用这个字符和片段中含有这个字符的个数来表示。例如 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
思路:分别写两个函数,解决压缩和解压的问题。
压缩函数:
void compress(string s)
{
string ans;
int j;
for (int i = 0; i < s.size(); i = j)
{
for (j = i; j < s.size() && s[i] == s[j]; j++);
if (j - i == 1)
ans += s[i];
else
ans += to_string((long long)j - i) + s[i];
}
cout << ans;
}
让j从i处开始数,当和i不相等时候,j-i即两数之间相差的数,接着让i到j位置,循环即可。但是我刚开始第二个for循环是这样写的:
for (j = i; s[i] == s[j] && j < s.size(); j++);
只和之前的判断条件顺序相反,却导致程序string越界。原因是逻辑判断运算符&&左边的式子已经判断错误时,将不会对&&右边的式再进行判断。所以假设j在字符串中最后一个,如果判断字符相同时,j此时因为还是小于s.size()的,所以j还是会往后走,然后下一次循环判断s[i] == s[j]时就会出错,因为不存在s[j]。而将j < s.size()放在前面就不会出现j出界后还会判断s[i] == s[j]的情况。
解压函数:
void unzip(string s)
{
string ans, num;
int cnt = 1;
for (int i = 0; i < s.size(); i++)
{
if (s[i] >= '0' && s[i] <= '9') num += s[i];
else{
if (num.size() > 0) cnt = stoi(num);
while (cnt--) ans += s[i];
cnt = 1;
num.clear();
}
}
cout << ans;
}
我本来以为字符重复个数在0~9之间,后来发现题目要求“题目保证字符重复个数在整型范围内”,也就是两位数三位数都有可能。然后就想不到办法来提取字符串中的数字。后来参考了柳神的做法,利用string一个一个的吸收每个“字符数字”,当当前字符不是数字之后用stoi函数将吸收的“字符串数字”变成整型数字,之后按题目要求输出。
完整代码:
#include <iostream>
#include <string>
using namespace std;
void compress(string s);
void unzip(string);
int main()
{
char ch;
string s;
cin >> ch;
getchar(); //注意getchar()吸收换行符
getline(cin, s);
if (ch == 'C') compress(s);
else unzip(s);
return 0;
}
void compress(string s)
{
string ans;
int j;
for (int i = 0; i < s.size(); i = j)
{
for (j = i; j < s.size() && s[i] == s[j]; j++);
if (j - i == 1)
ans += s[i];
else
ans += to_string((long long)j - i) + s[i];
}
cout << ans;
}
void unzip(string s)
{
string ans, num;
int cnt = 1;
for (int i = 0; i < s.size(); i++)
{
if (s[i] >= '0' && s[i] <= '9') num += s[i];
else{
if (num.size() > 0) cnt = stoi(num);
while (cnt--) ans += s[i];
cnt = 1;
num.clear();
}
}
cout << ans;
}