PAT 乙级 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

本文需要注意的是aaa的情况，其他的基本能过

// 1078.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include<iostream>
#include<string>
#include<cstdlib>
#include<stack>

using namespace std;

//将字母的个数置于字符串中
void get_num(int &num,string &str2,const string &str,int &i)
{
    stack<int> s;

    if (num != 1)
    {
        while (num)//将数字分别压入栈中
        {
            s.push(num % 10);
            num /= 10;
        }

        while (!s.empty())
        {
            str2 += (char)(s.top() + 48);
            s.pop();
        }

    }

    str2.append(1, str[i - 1]);
    num = 1;
}

void Zip()
{
    string str, str2;
    getline(cin, str);
    int size = str.size(), num = 1;

    for (int i = 1; i<size; ++i)
    {
        if (str[i] == str[i - 1])
        {
            num++;

            if (i == size - 1)
                get_num(num,str2,str,i);
        }
        else
        {
            get_num(num, str2, str, i);

            if (i == size - 1)
                str2.append(1, str[i]);
        }
    }

    cout << str2 << endl;
}

bool isdigit(char c)
{
    if (c <= '9'&&c >= '0')
        return true;
    else
        return false;
}

void Unzip()
{
    string str, str2;
    getline(cin, str);
    int size = str.size();

    for (int i = 0; i<size; ++i)
    {
        if (isdigit(str[i]))//如果为数字求得该数字
        {
            int num = 0, j;
            for (j = i; isdigit(str[j]); ++j)
            {
                num *= 10;
                num += str[j] - 48;
            }

            str2.append(num, str[j]);
            i = j;        //刷新i
        }
        else
            str2.append(1, str[i]);
    }

    cout << str2 << endl;
}

int main()
{
    char ch = getchar();
    getchar();

    if (ch == 'C')
        Zip();
    else
        Unzip();

    return 0;
}