編程之美挑戰賽初賽第二場題目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<vector>
#include<iostream>
#include<algorithm>
#include<string>
#include <stdlib.h>
using namespace std;

int main()
{
	int T,num=0;
	cin>>T;
	vector<string> okresult(T);
	for(int i=0;i<T;++i)
	{
		int a,step;
		int sum[2]={0,0};
		vector<vector<string> > result(2);
		vector<string> tep(2);
		vector<string> tep1(2);
		cin>>a;
		cin>>tep[0];
		cin>>tep[1];
		for (int t=0;t<2;++t)
		{
			for (string::iterator st1=tep[t].begin();st1!=tep[t].end();++st1)
			{
				int count=0;
				string ct1;
				while('0'<=*st1&&*st1<='9')
				{
					ct1+=*(st1++);
				}
				count=atoi(ct1.c_str());
				sum[t]+=count;
				for (int j=count;j>0;--j)
				{
					tep1[t]+=*(st1);
				}
			}
		}
		

		if (sum[0]==sum[1])
		{
			step=sum[0]/a;
			string temp;
			for (int t=0;t<2;++t)
			{
				for (int j=0;j<step;j++)
				{
					temp.clear();
					for (int k=j;k<tep1[t].size();k+=step)
					{
						temp+=tep1[t][k];
					}
					//result[t].insert(temp);
					result[t].push_back(temp);
				}
			}
			sort(result[0].begin(),result[0].end());
			sort(result[1].begin(),result[1].end());
			if (result[0]==result[1])
			{
				//cout<<"Case "<<i+1<<": "<<"Yes"<<endl;
				okresult[num++]="Yes";
			}
			else
			{
				//cout<<"Case "<<i+1<<": "<<"No"<<endl;
				okresult[num++]="No";
			}
		}
		else
		{
			//cout<<"Case "<<i+1<<": "<<"No"<<endl;
			okresult[num++]="No";
		}
	}

	for (int j=0;j<T;++j)
	{
		cout<<"Case "<<j+1<<": "<<okresult[j]<<endl;
	}
	system("pause"); 
	return 0;
}