C++学习篇（续）

网上的一个C++的STL的题目：

总时间限制: 4000ms 内存限制: 65536kB

描述

写一个程序完成以下命令：

new id ——新建一个指定编号为id的序列(id<10000)

add id num——向编号为id的序列加入整数num

merge id1 id2——合并序列id1和id2中的数，并将id2清空

unique id——去掉序列id中重复的元素

out id ——从小到大输出编号为id的序列中的元素，以空格隔开

输入
第一行一个数n，表示有多少个命令( n＜＝２０００００)。以后n行每行一个命令。
输出
按题目要求输出。

样例输入

16
new 1
new 2
add 1 1
add 1 2
add 1 3
add 2 1
add 2 2
add 2 3
add 2 4
out 1
out 2
merge 1 2
out 1
out 2
unique 1
out 1

样例输出

1 2 3
1 2 3 4
1 1 2 2 3 3 4

1 2 3 4

关于做这题的话，一开始就创个id类，用来记录数据。然后将所有换行等一系列的问题都搞定后，（在进行unique的时候得先sort一下，不然会报错。我的就是这样的）。发现会超时。最终发现是STL算法中的find太慢了，效率不高。然后就换了另一种方案，用map去记录，结果就搞定啦。还望高手们多多指教。
代码贴上：

</pre></div><div><pre name="code" class="cpp">//
//  List.cpp
//  9w
//
//  Created by vi on 15/7/6.
//  Copyright (c) 2015年 vi. All rights reserved.
//

#include <iostream>
#include <stdio.h>
#include <list>
#include <string>
#include <algorithm>
#include <map>
void list_9w1();
void map_9w1();

//一开始的错误方案
class data{
public:
int id;
std::list<int> num;
data(int id_):id(id_){}
bool operator==(const data &item) const{
if(id == item.id)
return 1;
else
return 0;
}

void PrintNum(){
for(std::list<int>::iterator p3 = num.begin();p3!=num.end();p3++){
std::cout << *p3 << " ";
}
printf("\n");
}
};
void list_9w1(){
//outstd::cout << "ok" ;
std::list<data> total;
std::list<data>::iterator p,p2;
int time,id,id2,num;
std::string each,op;
std::cin >> time;
while(time>0){

std::cin >> op;

if(op[0] == 'n'){
std::cin >> id;
total.push_back(id);

}
else if(op[0] == 'a'){
std::cin >> id;
std::cin >>num;
p = find(total.begin(),total.end(),id);
p->num.push_back(num);

}
else if(op[0] == 'm'){
std::cin >> id;
std::cin >> id2;
p = find(total.begin(),total.end(),id);
p2 = find(total.begin(),total.end(),id2);
p->num.merge(p2->num);

}
else if(op[0] == 'u'){
std::cin >> id;
p = find(total.begin(),total.end(),id);
p->num.sort();
p->num.unique();
}
else if(op[0] == 'o'){
std::cin >> id;
p = find(total.begin(),total.end(),id);
p->num.sort();
p->PrintNum();
}

time--;
}

}

//正确的方案
void map_9w1(){
std::map<int,std::list<int> > total;
std::map<int,std::list<int> >::iterator p,p2;
int time,id,id2,num;
std::string op;
scanf("%d",&time);
std::list<int> temp;
while(time > 0){

std::cin >> op;

if(op[0] == 'n'){
scanf("%d",&id);
total[id] = temp;
}
else if(op[0] == 'a'){
scanf("%d %d",&id,&num);
p = total.find(id);
if(p != total.end()){
p->second.push_back(num);
}

}
else if(op[0] == 'm'){
scanf("%d %d",&id,&id2);
p = total.find(id);
p2 = total.find(id2);
if(p != total.end()&&p != total.end()){
p->second.merge(p2->second);
}
}
else if(op[0] == 'u'){
scanf("%d",&id);
p = total.find(id);
p->second.sort();
p->second.unique();
}
else if(op[0] == 'o'){
scanf("%d",&id);
p = total.find(id);
p->second.sort();
for(std::list<int>::iterator i = p->second.begin(); i != p->second.end(); i++){
printf("%d ",*i);

}
printf("\n");
}

time--;
}
}

int main(){
map_9w1();
}