LeetCode 443. 压缩字符串

LeetCode 443. 压缩字符串

给定一组字符，使用原地算法将其压缩。

压缩后的长度必须始终小于或等于原数组长度。

数组的每个元素应该是长度为1 的字符（不是 int 整数类型）。

在完成原地修改输入数组后，返回数组的新长度。

进阶：
你能否仅使用O(1) 空间解决问题？

示例 1：
输入：
[“a”,“a”,“b”,“b”,“c”,“c”,“c”]
输出：
返回6，输入数组的前6个字符应该是：[“a”,“2”,“b”,“2”,“c”,“3”]
说明：
"aa"被"a2"替代。"bb"被"b2"替代。"ccc"被"c3"替代。

示例 2：
输入：
[“a”]
输出：
返回1，输入数组的前1个字符应该是：[“a”]
说明：
没有任何字符串被替代。

示例 3：
输入：
[“a”,“b”,“b”,“b”,“b”,“b”,“b”,“b”,“b”,“b”,“b”,“b”,“b”]
输出：
返回4，输入数组的前4个字符应该是：[“a”,“b”,“1”,“2”]。
说明：
由于字符"a"不重复，所以不会被压缩。"bbbbbbbbbbbb"被“b12”替代。
注意每个数字在数组中都有它自己的位置。

注意：
所有字符都有一个ASCII值在[35, 126]区间内。
1 <= len(chars) <= 1000。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/string-compression
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

• 解题思路

1.利用一个指针index来指向当前覆盖到的位置的下一位置，用循环遍历，
2.如果当前元素与后一个元素不同，则在index处存储当前元素，index移动到下一位置，如果有多个相同元素，则在index处记录个数，
3.重复2过程直到遍历完成。

• 代码

class Solution {
public:
    int compress(vector<char>& chars) {
        int index=0;
        int count=1;
        for(int i=0;i<chars.size()-1;i++)
        {
           
            if(chars[i]==chars[i+1])
            {
                count++;
                continue;
            }
            else{
                chars[index]=chars[i];
                index++;
                if(count==1)
                {
                  continue;
                }
                else
                {
                    index=to_char(chars,count,index);
                    count=1;
                }
                
            }
            
        }
        chars[index++]=chars[chars.size()-1];
        if(count!=1)
        {
            index=to_char(chars,count,index);
            count=1;
        }
      
        return index;
    }
    int to_char(vector<char>& chars,int count,int index)
    {
         if(count<10)
                    {chars[index++]=count+'0';
                    
                    }
                    else if(count<100)
                    {
                        chars[index++]=count/10 + '0';
                        chars[index++]=count%10 +'0';
                       
                    }
                    else if(count<1000)
                    {
                        chars[index++]=count/100+'0';
                        chars[index++]=(count/10)%10+'0';
                        chars[index++]=count%10+'0';
                       
                    }
                    else
                    {
                        chars[index++]='1';
                        chars[index++]='0';
                        chars[index++]='0';
                        chars[index++]='0';
                        
                        
                    }
        return index;
    }
    
};

• 结果分析

时间复杂度O（n）
空间复杂度O（1）

• 问题分析

1.刚开始判断当前元素与下一元素是否相等时，发生内存溢出。
解决办法，循环条件改为i<chars.size()-1;在循环体外，加入语句

chars[index++]=chars[chars.size()-1];
        if(count!=1)
        {
            index=to_char(chars,count,index);
            count=1;
        }

2.计数器count会存在大于10的情况
解决方法

int to_char(vector<char>& chars,int count,int index)
    {
         if(count<10)
                    {chars[index++]=count+'0';
                    
                    }
                    else if(count<100)
                    {
                        chars[index++]=count/10 + '0';
                        chars[index++]=count%10 +'0';
                       
                    }
                    else if(count<1000)
                    {
                        chars[index++]=count/100+'0';
                        chars[index++]=(count/10)%10+'0';
                        chars[index++]=count%10+'0';
                       
                    }
                    else
                    {
                        chars[index++]='1';
                        chars[index++]='0';
                        chars[index++]='0';
                        chars[index++]='0';
                        
                        
                    }
        return index;
    }
    

因多次使用，写出了函数，方便调用。