算法 - PAT 1078 字符串压缩与解压，第五个测试点过不去？

Question

题目传送门：https://pintia.cn/problem-sets/994805260223102976/exam/proble...

第五个测试点过不去，不知道什么问题：

#include<bits/stdc++.h>using namespace std;//压缩void fun1(string s){    int cnt = 1; //算上自身所以是1    string ss;    //首先看一下相同的字母就count++;,这样count就存储了要压缩的字母的个数    for (int i = 0; i < s.size(); i++)    {        if (s[i] == s[i + 1])cnt++;        else        {            if (cnt == 1)ss += s[i];  //只有一个字符的话不要把数字加上，即1s，而是直接加s,即 s           else ss += to_string(cnt) + s[i]; //要压缩的字母个数+字母           cnt = 1;  //重置         }    }    cout << ss<<endl;    }//解压void fun2(string s){    int index = 0;    string ss;    //解压的话我们需要看数字以及数字后面的字符    while (index < s.size())    {              if(isdigit(s[index])&&isdigit(s[index+1]))        {            int temp= (s[index] - '0') * 10 + (s[index + 1] - '0');            while (temp--)            {                if(index+2<s.size())                ss += s[index + 2];            }            index += 3;        }         else if (isdigit(s[index]))         {             int temp = s[index] - '0';             while (temp--)             {                 if(index+1<s.size())                 ss += s[index + 1];             }             index += 2;         }       else //字母         {           if(index<s.size())           {            ss += s[index];            index++;           }        }          }    cout << ss<<endl;}int main(){    char c; cin >> c; getchar();    string s; getline(cin, s);    if (c == 'C')        fun1(s);    else if (c == 'D')        fun2(s);    return 0;}

修正代码

Answer 1

首先，我们需要明确题目的要求。根据提供的链接，这是一个字符串压缩与解压的问题。压缩时，连续的相同字符会被替换为字符及其连续出现的次数；解压时，需要还原回原始字符串。

现在，我们来看你的代码。

1. 压缩函数 (fun1)

* 你在循环中使用了 `s[i + 1]`，这可能会导致数组越界，因为当 `i` 是最后一个字符时，`s[i + 1]` 是未定义的。* 你的代码逻辑在处理只有一个字符的连续序列时存在问题。当只有一个字符时，你不应该增加 `cnt`，而应该直接添加到结果字符串中。

2. 解压函数 (fun2)

* 你的代码在处理数字时存在问题。例如，对于输入 "3a4b"，你的代码会错误地处理为 "aaabbbb"，而不是正确的 "aaaabbbb"。* 你在处理数字时，没有考虑到数字可能超过两位数的情况。

下面是修正后的代码：

#include<bits/stdc++.h>using namespace std;void fun1(string s) {    string ss;    int cnt = 1;    for (int i = 1; i <= s.size(); i++) {        if (i < s.size() && s[i] == s[i - 1]) {            cnt++;        } else {            if (cnt > 1) ss += to_string(cnt);            ss += s[i - 1];            cnt = 1;        }    }    cout << ss << endl;}void fun2(string s) {    string ss;    int num = 0, i = 0;    while (i < s.size()) {        if (isdigit(s[i])) {            num = num * 10 + (s[i] - '0');            i++;        } else {            while (num--) ss += s[i];            num = 0;            i++;        }    }    cout << ss << endl;}int main() {    char c; cin >> c; getchar();    string s; getline(cin, s);    if (c == 'C') fun1(s);    else if (c == 'D') fun2(s);    return 0;}

这个修正的代码应该能解决你的问题，并通过第五个测试点。