编程之美初赛第二场 题目2 : 字符串压缩

时间限制:8000ms

单点时限:4000ms

内存限制:256MB

描述

你的硬盘上有一个神秘的文件占用了大量空间，你决定将其压缩以节省空间。不幸的是，你还没有安装任何压缩软件，所以你决定自己编写一个压缩程序。你发现这是一个文本文件，包括很多行。每行是一个长度恰好为L的字符串，而且字符串可能有重复。行的顺序并不重要，换言之，打乱顺序之后仍然可以认为文件内容和原来相同。

例如，这个文件的内容可以是这样的：

bar

car

bat

cat

cat

 

经过一段时间观察，你发现同一列的字符往往是相同的，于是你设计了一个简单的压缩框架。首先以某种策略调整行的顺序，然后把所有字符串按照先列后行的顺序变换成单个字符串，例如上面的例子，不调整顺序则直接变换成：

bcbccaaaaarrttt

然后使用游程编码(RLE)的到压缩变换后的字符串：

1b1c1b2c5a2r3t

当然也可以先调换顺序：

car

cat

cat

bat

bar

这样的压缩字符串为：

3c2b5a1r3t1r

比不调整顺序的稍短一些。

现在，你已经得到了两个不同的压缩字符串，你想知道他们解压后的文件是否相同，请写一个程序解决这个问题。

输入

第一行是一个整数T (T <= 30)，表示测试数据组数。

每组测试数据占三行。第一行为整数L，表示原始文件中每一行字符串的长度。第二行和第三行分别是两个压缩字符串，格式如c1 n1 c2 n2 … cMnM,表示字符ci连续出现了ni次。具体格式见样例。输入字符串只含a到z的小写字母，确保压缩字符串合法有效，且不为空。

输出

对每组测试数据，首先输出”Case x: ”，其中x表示测试数据编号。如果两个压缩字符串对应于相同的文件内容，则输出”Yes”，否则输出”No”。

数据范围

小数据：1<=L<=10, 1<=ni<=100,压缩字符串长度不超过10^4

大数据：1<=L<=1000, 1<=ni<=10^9,压缩字符串长度不超过10^6

样例输入

2
3
1b1c1b2c5a2r3t
3c2b5a1r3t1r
2
20a20b10a20b10a
20a20b20a20b

样例输出

Case 1: Yes
Case 2: No

 

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

const int MAX = 10000 + 4;

char str[2][MAX*3];
char input[MAX];
int L;
int lineNum;
int Num[2];
int visit[MAX*3];

int main()
{
    int T;
    int n;
    char ch1, ch2;
    scanf("%d", &T);
    for(int t=1; t<=T; t++) {
        scanf("%d", &L);
        getchar();
        Num[0] = Num[1] = 0;
        for(int i=0; i<2; ++i) {
            scanf("%s", input);
            int cur = 0;
            int length = strlen(input);
            for(int j=0; j<length;  ) {
                n = 0;
                while(input[j] <= '9' && input[j] >= '0') {
                    n = n*10 + (input[j++] - '0');
                }
                Num[i] += n;
                for(int k=1; k<=n; k++) {
                    str[i][cur++] = input[j];
                }
                j++;
            }
           // printf("%s\n", str[i]);
        }
        if(Num[0] != Num[1]) {
            printf("Case %d: No\n", t);
            continue;
        }
        lineNum = Num[0] / L;
        memset(visit, -1, sizeof(visit));
        bool res = true;
        for(int i=0; i<lineNum; ++i) {
            bool flag = false;
            for(int j=0; j<lineNum; ++j) {
                if(visit[j] != -1) {
                    continue;
                }
                int k;
                for(k=0; k<L; ++k) {
                    if(str[0][i+k*lineNum] != str[1][j+k*lineNum]) {
                        break;
                    }
                }
                if(k >= L) {
                    visit[j] = 1;
                    flag = true;
                    break;
                }
            }
            if(!flag) {
                res = false;
                break;
            }
        }
        if(res) {
            printf("Case %d: Yes\n", t);
        }else {
            printf("Case %d: No\n", t);
        }
    }
    return 0;
}

主要是分配数组的大小，压缩字符串不超过10^4， 1<=ni<=100, 那么存储的字符串至少要：

100*(10000 /(3+1)) = 25000