字符串压缩之C++实现

题目： 输入一个字符串，输出对其压缩过的形式。
    如输入为：aaabbbccc则输出为3a3b3c；如输入为abbc，则输出为a2bc。
    aaabbbccc  ---->  3a3b3c

    abc        ---->  abc

#include <iostream>
#include <string>
using namespace std;

int main()
{
    int cnt = 1;///cnt the number of every char
    int pos = 1;
    char tmp_A,tmp_B;
    string ori_str, sht_str;
    string::size_type ori_len;
    char c_cnt;
    cout<<"input the ori string"<<endl;
    cin>>ori_str;

    ori_len = ori_str.size();
    for (; pos < static_cast<int>(ori_len); ++pos)
    {
        tmp_A = ori_str[pos - 1];
        tmp_B = ori_str[pos];



        if (tmp_A != tmp_B && pos != static_cast<int>(ori_len-1))
        ///没到末尾的时候，AB不同
        {
            if (cnt == 1)
            {
                sht_str.push_back(tmp_A);
            }
            else
            {
                c_cnt = cnt + '0';
                sht_str.push_back(c_cnt);
                sht_str.push_back(tmp_A);
                cnt = 1;
            }
        }
        else if (tmp_A != tmp_B && pos == static_cast<int>(ori_len-1))
        ///在末尾，但是A,B不同
        {
            if (cnt == 1)
            {
                sht_str.push_back(tmp_A);
            }
            else
            {
                sht_str.push_back(cnt);
                sht_str.push_back(tmp_A);
                cnt = 1;
            }
            sht_str.push_back(tmp_B);
        }
        else if (tmp_A == tmp_B && pos != static_cast<int>(ori_len-1))
        ///不在末尾，AB相同
        {
            ++cnt;
        }
        else if (tmp_A == tmp_B && pos == static_cast<int>(ori_len-1))
        ///在末尾，AB相同
        {
            cnt++;
            c_cnt = cnt + '0';
            sht_str.push_back(c_cnt);
            sht_str.push_back(tmp_A);
        }
    }

    cout<<sht_str;
    return 0;
}