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();
}

•